Home
last modified time | relevance | path

Searched refs:avgpool (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/
Dimport_tensorflow.cc1063 auto* avgpool = new AveragePoolOperator; in ConvertAvgPoolOperator() local
1064 avgpool->inputs.push_back(input_name); in ConvertAvgPoolOperator()
1065 avgpool->outputs.push_back(node.name()); in ConvertAvgPoolOperator()
1070 avgpool->stride_height = strides.i(1); in ConvertAvgPoolOperator()
1071 avgpool->stride_width = strides.i(2); in ConvertAvgPoolOperator()
1076 avgpool->kheight = ksize.i(1); in ConvertAvgPoolOperator()
1077 avgpool->kwidth = ksize.i(2); in ConvertAvgPoolOperator()
1080 avgpool->padding.type = PaddingType::kSame; in ConvertAvgPoolOperator()
1082 avgpool->padding.type = PaddingType::kValid; in ConvertAvgPoolOperator()
1086 model->operators.emplace_back(avgpool); in ConvertAvgPoolOperator()
Dtooling_util.cc1943 const auto& avgpool = *static_cast<const AveragePoolOperator*>(&op); in EstimateArithmeticOpsCount() local
1949 avgpool.kheight * avgpool.kwidth; in EstimateArithmeticOpsCount()