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

Function _convert_reshape

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

Source from the content-addressed store, hash-verified

1748
1749@RegisterPFor("Reshape")
1750def _convert_reshape(pfor_input):
1751 t = pfor_input.stacked_input(0)
1752 shape = pfor_input.unstacked_input(1)
1753 new_shape = array_ops.concat([pfor_input.pfor.loop_len_vector, shape], axis=0)
1754 return wrap(array_ops.reshape(t, new_shape), True)
1755
1756
1757@RegisterPFor("BroadcastTo")

Callers

nothing calls this directly

Calls 5

stacked_inputMethod · 0.80
unstacked_inputMethod · 0.80
reshapeMethod · 0.80
wrapFunction · 0.70
concatMethod · 0.45

Tested by

no test coverage detected