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

Function _convert_pack

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

Source from the content-addressed store, hash-verified

1882
1883@RegisterPFor("Pack")
1884def _convert_pack(pfor_input):
1885 pfor_input.stack_inputs()
1886 axis = pfor_input.get_attr("axis")
1887 if axis >= 0:
1888 axis += 1
1889 return wrap(
1890 array_ops.stack([x.t for x in pfor_input.inputs], axis=axis), True)
1891
1892
1893@RegisterPFor("Unpack")

Callers

nothing calls this directly

Calls 4

stack_inputsMethod · 0.80
wrapFunction · 0.70
get_attrMethod · 0.45
stackMethod · 0.45

Tested by

no test coverage detected