Searched refs:kGatherNd (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | op_version.cc | 87 {{OperatorType::kGatherNd, 1}, "1.14.0"}, in GetMinimumRuntimeVersionForModel()
88 {{OperatorType::kGatherNd, 2}, kPendingReleaseOpVersion}, in GetMinimumRuntimeVersionForModel()
|
| D | operator_test.cc | 248 GetOperator("GATHER_ND", OperatorType::kGatherNd), op); in TEST_F()
|
| D | operator.cc | 1868 OperatorType::kGatherNd)); in BuildOperatorList()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model.h | 173 kGatherNd, enumerator
1810 GatherNdOperator() : Operator(OperatorType::kGatherNd) {}
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 80 constexpr char kGatherNd[] = "GatherNd"; variable
417 device_cost_impl_.emplace(kGatherNd, in OpLevelCostEstimator()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 2203 case OperatorType::kGatherNd: in Run()
|