Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/toco/
Dimport_tensorflow.cc1024 auto* maxpool = new MaxPoolOperator; in ConvertMaxPoolOperator() local
1025 maxpool->inputs.push_back(input_name); in ConvertMaxPoolOperator()
1026 maxpool->outputs.push_back(node.name()); in ConvertMaxPoolOperator()
1031 maxpool->stride_height = strides.i(1); in ConvertMaxPoolOperator()
1032 maxpool->stride_width = strides.i(2); in ConvertMaxPoolOperator()
1037 maxpool->kheight = ksize.i(1); in ConvertMaxPoolOperator()
1038 maxpool->kwidth = ksize.i(2); in ConvertMaxPoolOperator()
1041 maxpool->padding.type = PaddingType::kSame; in ConvertMaxPoolOperator()
1043 maxpool->padding.type = PaddingType::kValid; in ConvertMaxPoolOperator()
1047 model->operators.emplace_back(maxpool); in ConvertMaxPoolOperator()
Dtooling_util.cc1933 const auto& maxpool = *static_cast<const MaxPoolOperator*>(&op); in EstimateArithmeticOpsCount() local
1939 maxpool.kheight * maxpool.kwidth; in EstimateArithmeticOpsCount()
1953 const auto* maxpool = static_cast<const MaxPoolOperator*>(&op); in EstimateArithmeticOpsCount() local
1960 const int64 cost_per_val = 2 * maxpool->kheight * maxpool->kwidth + 32; in EstimateArithmeticOpsCount()