Searched refs:batch_gather_nd (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | array_ops_test.py | 1938 out = array_ops.batch_gather_nd(
1965 batch_gather_nd_result = array_ops.batch_gather_nd(
1996 batch_gather_nd_result = array_ops.batch_gather_nd(
2051 array_ops.batch_gather_nd(
2057 array_ops.batch_gather_nd(params=params, indices=indices, batch_dims=-1)
2060 array_ops.batch_gather_nd(params=params, indices=indices, batch_dims=4)
2091 out = array_ops.batch_gather_nd(
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 5354 return batch_gather_nd(params, indices, batch_dims=batch_dims, name=name)
5366 def batch_gather_nd(params, indices, batch_dims, name=None): function
|