| /external/tensorflow/tensorflow/lite/kernels/ |
| D | gather_nd.cc | 105 TfLiteStatus GatherNd(const TfLiteTensor* params, const TfLiteTensor* indices, in GatherNd() function
107 reference_ops::GatherNd( in GatherNd()
128 return GatherNd<float, IndicesT>(params, indices, output); in EvalGatherNd()
130 return GatherNd<uint8_t, IndicesT>(params, indices, output); in EvalGatherNd()
132 return GatherNd<int8_t, IndicesT>(params, indices, output); in EvalGatherNd()
134 return GatherNd<int16_t, IndicesT>(params, indices, output); in EvalGatherNd()
136 return GatherNd<int32_t, IndicesT>(params, indices, output); in EvalGatherNd()
138 return GatherNd<int64_t, IndicesT>(params, indices, output); in EvalGatherNd()
|
| /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/kernels/ |
| D | gather_nd_op_test.cc | 44 class Node* GatherNd(Graph* g, class Node* in0, class Node* in1) { in GatherNd() function
112 static Graph* GatherNd(int dim) { in GatherNd() function
130 test::graph::GatherNd(g, test::graph::Constant(g, params), in GatherNd()
139 test::Benchmark(#DEVICE, GatherNd<INDEX>(dim), \
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/pad/ |
| D | ops_defs.py | 59 left_padding_size = tf.raw_ops.GatherNd(params=paddings, indices=[i, 0])
60 right_padding_size = tf.raw_ops.GatherNd(params=paddings, indices=[i, 1])
107 left_padding_size = tf.raw_ops.GatherNd(params=paddings, indices=[i, 0])
108 right_padding_size = tf.raw_ops.GatherNd(params=paddings, indices=[i, 1])
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_GatherNd.pbtxt | 2 graph_op_name: "GatherNd"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | GatherNd.pbtxt | 2 name: "GatherNd"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | GatherNd.pbtxt | 2 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 | 282 grad_outputs->push_back(GatherNd(scope, grad_inputs[0], indices)); in ScatterNdGrad()
294 grad_outputs->push_back(GatherNd(scope, grad_inputs[0], indices)); in ScatterNdNonAliasingAddGrad()
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | op_select_allowlist.md | 200 * `raw_ops.GatherNd`
|
| /external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
| D | gpu_supported_ops.md | 94 `GatherNd` | `Tindices={int32,int64}`<br>`Tparams={bool,complex64,double…
|
| D | cpu_supported_ops.md | 94 `GatherNd` | `Tindices={int32,int64}`<br>`Tparams={bool,complex64,double…
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 515 class GatherNd class
1867 ops.push_back(MakeUnique<GatherNd>(::tflite::BuiltinOperator_GATHER_ND, in BuildOperatorList()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 1094 inline void GatherNd(const RuntimeShape& params_shape, in GatherNd() function
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 412 HANDLE_OPERATORTYPENAME_CASE(GatherNd) in OperatorTypeName()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | legalize-tf.mlir | 412 %0 = "tf.GatherNd"(%arg0, %arg1) : (tensor<3x2x2xf32>, tensor<2xi32>) -> tensor<2xf32>
420 %0 = "tf.GatherNd"(%arg0, %arg1) : (tensor<4x3x2xf32>, tensor<2x2xi32>) -> tensor<2x2xf32>
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | legalize_hlo.mlir | 1892 // CHECK: %[[VAL_0:.*]] = "tf.GatherNd"(%[[ARG_0]], %[[ARG_1]]) : {{.*}} -> tensor<192x2…
1905 // CHECK: %[[VAL_4:.*]] = "tf.GatherNd"(%[[VAL_0]], %[[VAL_3]]) : {{.*}} -> tensor<4x64x1…
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 1842 ### tf.GatherNd
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 1760 name: "GatherNd"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 1760 name: "GatherNd"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 13185 name: "GatherNd"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 2200 func GatherNd(scope *Scope, params tf.Output, indices tf.Output) (output tf.Output) { func
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 4850 def TF_GatherNdOp : TF_Op<"GatherNd", [NoSideEffect]> {
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 18210 name: "GatherNd"
|