Home
last modified time | relevance | path

Searched refs:take_along_axis (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tests/
Dsort_ops_test.py149 np.take_along_axis(inp, sorted_indices, axis=dimension)
191 np.take_along_axis(inp, sorted_indices, axis=0) for inp in inputs
/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_array_ops.py1400 def take_along_axis(arr, indices, axis): # pylint: disable=missing-docstring function
1405 return take_along_axis(arr.ravel(), indices, 0)
/external/tensorflow/third_party/py/numpy/tf_numpy_api/
Dtensorflow.experimental.numpy.pbtxt836 name: "take_along_axis"