| /external/tensorflow/tensorflow/lite/kernels/ |
| D | gather_nd.cc | 97 TfLiteStatus GatherNd(const TfLiteTensor* params, const TfLiteTensor* indices, in GatherNd() function
99 reference_ops::GatherNd( in GatherNd()
111 return GatherNd<float, IndicesT>(params, indices, output); in EvalGatherNd()
113 return GatherNd<uint8_t, IndicesT>(params, indices, output); in EvalGatherNd()
115 return GatherNd<int8_t, IndicesT>(params, indices, output); in EvalGatherNd()
117 return GatherNd<int32_t, IndicesT>(params, indices, output); in EvalGatherNd()
119 return GatherNd<int64_t, IndicesT>(params, indices, output); in EvalGatherNd()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gather_nd_op_test.cc | 44 class Node* GatherNd(Graph* g, class Node* in0, class Node* in1) { in GatherNd() function
86 static Graph* GatherNd(int dim) { in GatherNd() function
104 test::graph::GatherNd(g, test::graph::Constant(g, params), in GatherNd()
115 test::Benchmark(#DEVICE, GatherNd<INDEX>(dim)).Run(iters); \
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_GatherNd.pbtxt | 2 graph_op_name: "GatherNd"
4 name: "GatherNd"
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_GatherNd.pbtxt | 2 graph_op_name: "GatherNd"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_GatherNd.pbtxt | 2 graph_op_name: "GatherNd"
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad_test.cc | 157 auto y = GatherNd(scope_, x, indices); in TEST_F()
165 auto y = GatherNd(scope_, x, indices); in TEST_F()
|
| D | array_grad.cc | 273 grad_outputs->push_back(GatherNd(scope, grad_inputs[0], indices)); in ScatterNdGrad()
285 grad_outputs->push_back(GatherNd(scope, grad_inputs[0], indices)); in ScatterNdNonAliasingAddGrad()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
| D | cpu_supported_ops.md | 94 `GatherNd` | `Tindices={int32,int64}`<br>`Tparams={bool,complex64,double…
|
| D | gpu_supported_ops.md | 94 `GatherNd` | `Tindices={int32,int64}`<br>`Tparams={bool,complex64,double…
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 549 class GatherNd class
2370 ops.push_back(MakeUnique<GatherNd>(::tflite::BuiltinOperator_GATHER_ND, in BuildOperatorList()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 391 HANDLE_OPERATORTYPENAME_CASE(GatherNd) in OperatorTypeName()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 2973 inline void GatherNd(const RuntimeShape& params_shape, in GatherNd() function
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 1376 name: "GatherNd"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 1376 name: "GatherNd"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 2264 func GatherNd(scope *Scope, params tf.Output, indices tf.Output) (output tf.Output) { func
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 13185 name: "GatherNd"
|
| D | ops_history.v2.pbtxt | 28417 name: "GatherNd"
|
| D | ops_history.v1.pbtxt | 28788 name: "GatherNd"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 14206 name: "GatherNd"
|