Searched refs:_expand_dims_0d_aware (Results 1 – 2 of 2) sorted by relevance
/dports/misc/tvm/incubator-tvm-0.6.1/nnvm/python/nnvm/frontend/ |
H A D | tensorflow.py | 352 return _expand_dims_0d_aware(inputs[0], attr, axis=axis.asnumpy()[0]) 420 inputs_reshaped = [_expand_dims_0d_aware(i, attr, axis=axis, num_newaxis=1) for i in inputs] 897 def _expand_dims_0d_aware(data, attr, axis, num_newaxis=1): function
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/nnvm/python/nnvm/frontend/ |
H A D | tensorflow.py | 352 return _expand_dims_0d_aware(inputs[0], attr, axis=axis.asnumpy()[0]) 420 inputs_reshaped = [_expand_dims_0d_aware(i, attr, axis=axis, num_newaxis=1) for i in inputs] 897 def _expand_dims_0d_aware(data, attr, axis, num_newaxis=1): function
|