Searched defs:gather_nd (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_gather_ops.py | 330 def gather_nd(params, indices, batch_dims=0, name=None): function
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | gather_nd.cc | 27 namespace gather_nd { namespace
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | autocast_variable.py | 130 def gather_nd(self, indices, name=None): member in AutoCastVariable
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | resource_variable_ops.py | 719 def gather_nd(self, indices, name=None): member in BaseResourceVariable
|
| D | array_ops.py | 5196 def gather_nd(params, indices, name=None, batch_dims=0): function
|
| D | variables.py | 947 def gather_nd(self, indices, name=None): member in Variable
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantize_model_test.cc | 1817 const auto& gather_nd = subgraph->operators[0]; in TEST_P() local
|