Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/lite/toco/
Dmodel.h117 kTopK_V2, enumerator
1409 TopKV2Operator() : Operator(OperatorType::kTopK_V2) {}
Dexport_tensorflow.cc1743 } else if (src_op.type == OperatorType::kTopK_V2) { in ConvertOperator()
/external/tensorflow/tensorflow/contrib/lite/toco/tflite/
Doperator_test.cc386 GetOperator("TOPK_V2", OperatorType::kTopK_V2), op); in TEST_F()
Doperator.cc827 new TopK_V2(::tflite::BuiltinOperator_TOPK_V2, OperatorType::kTopK_V2)); in BuildOperatorList()
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc1373 case OperatorType::kTopK_V2: in Run()