MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / _convert_concatv2

Function _convert_concatv2

tensorflow/python/ops/parallel_for/pfor.py:2027–2034  ·  view source on GitHub ↗
(pfor_input)

Source from the content-addressed store, hash-verified

2025
2026@RegisterPFor("ConcatV2")
2027def _convert_concatv2(pfor_input):
2028 n = pfor_input.num_inputs
2029 pfor_input.stack_inputs(stack_indices=range(n - 1))
2030 axis = pfor_input.unstacked_input(n - 1)
2031 axis += math_ops.cast(axis >= 0, axis.dtype)
2032 return wrap(
2033 array_ops.concat([x.t for x in pfor_input.inputs[:n - 1]], axis=axis),
2034 True)
2035
2036
2037@RegisterPFor("StridedSlice")

Callers

nothing calls this directly

Calls 6

stack_inputsMethod · 0.80
unstacked_inputMethod · 0.80
wrapFunction · 0.70
rangeFunction · 0.50
castMethod · 0.45
concatMethod · 0.45

Tested by

no test coverage detected