Searched defs:gather_nd (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | gather_nd.cc | 28 namespace gather_nd { namespace
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_gather_ops.py | 339 def gather_nd(params: ragged_tensor.RaggedOrDense, function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | values_v2.py | 219 def gather_nd(self, indices, name=None): member in DistributedVariable
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| H A D | autocast_variable.py | 125 def gather_nd(self, indices, name=None): member in AutoCastVariable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/ |
| H A D | quantize_model_test.cc | 1434 const auto& gather_nd = subgraph->operators[0]; in TEST_P() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | resource_variable_ops.py | 766 def gather_nd(self, indices, name=None): member in BaseResourceVariable
|
| H A D | array_ops.py | 5473 def gather_nd(params, indices, name=None, batch_dims=0): function
|
| H A D | variables.py | 948 def gather_nd(self, indices, name=None): member in Variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/ |
| H A D | quantize_model_test.cc | 1944 const auto& gather_nd = subgraph->operators[0]; in TEST_P() local
|