| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | gather_nd.cc | 106 TfLiteStatus GatherNd(const TfLiteTensor* params, const TfLiteTensor* indices, in GatherNd() function
108 return reference_ops::GatherNd( in GatherNd()
139 status = GatherNd<float, IndicesT>(params, indices, output); in EvalGatherNd()
142 status = GatherNd<uint8_t, IndicesT>(params, indices, output); in EvalGatherNd()
145 status = GatherNd<int8_t, IndicesT>(params, indices, output); in EvalGatherNd()
148 status = GatherNd<int16_t, IndicesT>(params, indices, output); in EvalGatherNd()
151 status = GatherNd<int32_t, IndicesT>(params, indices, output); in EvalGatherNd()
154 status = GatherNd<int64_t, IndicesT>(params, indices, output); in EvalGatherNd()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A 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), \
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/pad/ |
| H A D | ops_defs.py | 55 left_padding_size = tf.raw_ops.GatherNd(params=paddings, indices=[i, 0])
56 right_padding_size = tf.raw_ops.GatherNd(params=paddings, indices=[i, 1])
103 left_padding_size = tf.raw_ops.GatherNd(params=paddings, indices=[i, 0])
104 right_padding_size = tf.raw_ops.GatherNd(params=paddings, indices=[i, 1])
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/api_def/java_api/ |
| H A D | api_def_GatherNd.pbtxt | 2 graph_op_name: "GatherNd"
4 name: "GatherNd"
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| H A D | GatherNd.pbtxt | 2 name: "GatherNd"
31 name: "GatherNd"
|
| /aosp_15_r20/external/tensorflow/tensorflow/security/advisory/ |
| H A D | tfsa-2021-076.md | 1 ## TFSA-2021-076: Division by zero in TFLite's implementation of `GatherNd`
7 The reference implementation of the `GatherNd` TFLite operator is [vulnerable to
|
| H A D | tfsa-2022-087.md | 7 The [`GatherNd`](https://github.com/tensorflow/tensorflow/blob/f463040eb3997e42e60a2ffc6dc72de7ef11…
|
| H A D | tfsa-2022-143.md | 7 The [`GatherNd`](https://github.com/tensorflow/tflite-micro/blob/1bc98621180a350eb4e8d3318ea8e228c7…
|
| H A D | tfsa-2021-161.md | 7 TFLite's [`GatherNd`
|
| /aosp_15_r20/external/armnn/src/armnn/layers/ |
| H A D | GatherNdLayer.cpp | 17 : Layer(2, 1, LayerType::GatherNd, name) in GatherNdLayer()
26 return factory.CreateWorkload(LayerType::GatherNd, descriptor, PrepInfoAndDesc(descriptor)); in CreateWorkload()
|
| /aosp_15_r20/external/armnn/docs/ |
| H A D | 05_02_deserializer_serializer.dox | 45 - GatherNd
137 - GatherNd
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/api_def/python_api/ |
| H A D | api_def_GatherNd.pbtxt | 2 graph_op_name: "GatherNd"
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| H A D | GatherNd.pbtxt | 2 name: "GatherNd"
|
| /aosp_15_r20/external/armnn/src/armnnDeserializer/test/ |
| H A D | DeserializeGatherNd.cpp | 79 layerType: "GatherNd", in GatherNdFixture()
|
| /aosp_15_r20/external/armnn/src/armnn/ |
| H A D | LayersFwd.hpp | 138 DECLARE_LAYER(GatherNd)
|
| /aosp_15_r20/external/armnn/delegate/classic/ |
| H A D | CMakeLists.txt | 25 src/GatherNd.hpp
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/ |
| H A D | array_grad_test.cc | 165 auto y = GatherNd(scope_, x, indices); in TEST_F()
173 auto y = GatherNd(scope_, x, indices); in TEST_F()
|
| H A D | array_grad.cc | 296 grad_outputs->push_back(GatherNd(scope, grad_inputs[0], indices)); in ScatterNdGrad()
308 grad_outputs->push_back(GatherNd(scope, grad_inputs[0], indices)); in ScatterNdNonAliasingAddGrad()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/ |
| H A D | api_def_GatherNd.pbtxt | 2 graph_op_name: "GatherNd"
|
| /aosp_15_r20/external/armnn/include/armnn/ |
| H A D | Types.hpp | 471 X(GatherNd) \
|
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
| H A D | GatherNdTestImpl.cpp | 45 …:unique_ptr<armnn::IWorkload> workload = workloadFactory.CreateWorkload(armnn::LayerType::GatherNd, in GatherNdTestImpl()
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
| H A D | spmd_expanders.cc | 267 REGISTER_SPMD(GatherNd, TF::GatherNdOp, GatherNdSPMDExpander);
|
| /aosp_15_r20/external/armnn/src/armnnSerializer/ |
| H A D | ArmnnSchema.fbs | 184 GatherNd = 67,
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
| H A D | gpu_supported_ops.md | 94 `GatherNd` | `Tindices={int32,int64}`<br>`Tparams={bool,complex64,double…
|
| H A D | cpu_supported_ops.md | 94 `GatherNd` | `Tindices={int32,int64}`<br>`Tparams={bool,complex64,double…
|