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

Function _convert_expanddims

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

Source from the content-addressed store, hash-verified

1775
1776@RegisterPFor("ExpandDims")
1777def _convert_expanddims(pfor_input):
1778 t = pfor_input.stacked_input(0)
1779 dim = pfor_input.unstacked_input(1)
1780 dim += math_ops.cast(dim >= 0, dtypes.int32)
1781 return wrap(array_ops.expand_dims(t, axis=dim), True)
1782
1783
1784@RegisterPForWithArgs("LowerBound", gen_array_ops.lower_bound)

Callers

nothing calls this directly

Calls 5

stacked_inputMethod · 0.80
unstacked_inputMethod · 0.80
wrapFunction · 0.70
castMethod · 0.45
expand_dimsMethod · 0.45

Tested by

no test coverage detected