Searched refs:kTopK_V2 (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | op_version.cc | 164 {{OperatorType::kTopK_V2, 1}, "1.7.0"}, in GetMinimumRuntimeVersionForModel()
165 {{OperatorType::kTopK_V2, 2}, "1.14.0"}, in GetMinimumRuntimeVersionForModel()
|
| D | operator_test.cc | 505 GetOperator("TOPK_V2", OperatorType::kTopK_V2), op); in TEST_F()
|
| D | operator.cc | 1925 OperatorType::kTopK_V2)); in BuildOperatorList()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_array_data_types.cc | 155 case OperatorType::kTopK_V2: { in Run()
|
| D | quantize.cc | 106 OperatorType::kTopK_V2, in SupportsQuantization()
|
| D | propagate_fixed_sizes.cc | 2206 case OperatorType::kTopK_V2: in Run()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model.h | 137 kTopK_V2, enumerator
1933 TopKV2Operator() : Operator(OperatorType::kTopK_V2) {}
|
| D | export_tensorflow.cc | 2321 } else if (src_op.type == OperatorType::kTopK_V2) { in ConvertOperator()
|