Home
last modified time | relevance | path

Searched refs:kTopK_V2 (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_array_data_types.cc150 case OperatorType::kTopK_V2: { in Run()
Dquantize.cc66 type == OperatorType::kPack || type == OperatorType::kTopK_V2 || in SupportsQuantization()
Dpropagate_fixed_sizes.cc2123 case OperatorType::kTopK_V2: in Run()
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h134 kTopK_V2, enumerator
1867 TopKV2Operator() : Operator(OperatorType::kTopK_V2) {}
Dexport_tensorflow.cc2280 } else if (src_op.type == OperatorType::kTopK_V2) { in ConvertOperator()
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator_test.cc479 GetOperator("TOPK_V2", OperatorType::kTopK_V2), op); in TEST_F()
Doperator.cc2426 OperatorType::kTopK_V2)); in BuildOperatorList()