Home
last modified time | relevance | path

Searched refs:pool_params (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_pooling_ops_common.h499 MklPoolParameters* pool_params, in InitMklPoolParameters() argument
503 pool_params->Init(context, this->ksize_, this->stride_, this->padding_, in InitMklPoolParameters()
506 pool_params->Init(context, this->ksize_, this->stride_, this->padding_, in InitMklPoolParameters()
511 void PoolParamsToDims(const MklPoolParameters* pool_params, in PoolParamsToDims() argument
518 memory::dims({pool_params->window_rows, pool_params->window_cols}); in PoolParamsToDims()
520 memory::dims({pool_params->row_stride, pool_params->col_stride}); in PoolParamsToDims()
521 *padding_left = memory::dims({static_cast<int>(pool_params->pad_top), in PoolParamsToDims()
522 static_cast<int>(pool_params->pad_left)}); in PoolParamsToDims()
523 *padding_right = memory::dims({static_cast<int>(pool_params->pad_bottom), in PoolParamsToDims()
524 static_cast<int>(pool_params->pad_right)}); in PoolParamsToDims()
[all …]
Dmkl_avgpooling_op.cc63 MklPoolParameters pool_params; in Compute() local
68 this->InitMklPoolParameters(context, &pool_params, dnn_shape_input, in Compute()
74 this->GetOutputDims(pool_params, &output_dims_mkl_order); in Compute()
79 this->AllocateEmptyOutputTensor(context, kOutputIndex, &pool_params, in Compute()
86 this->PoolParamsToDims(&pool_params, &filter_dims, &strides, in Compute()
193 MklPoolParameters pool_params; in Compute() local
201 this->InitMklPoolParameters(context, &pool_params, orig_input_mkl_shape, in Compute()
205 this->PoolParamsToDims(&pool_params, &filter_dims, &strides, in Compute()
224 this->GetOutputDims(pool_params, &output_dims_mkl_order); in Compute()
Dmkl_maxpooling_op.cc68 MklPoolParameters pool_params; in Compute() local
73 this->InitMklPoolParameters(context, &pool_params, dnn_shape_input, in Compute()
82 this->GetOutputDims(pool_params, &output_dims_mkl_order); in Compute()
87 this->AllocateEmptyOutputTensor(context, kOutputIndex, &pool_params, in Compute()
99 &pool_params, output_dims_mkl_order, in Compute()
127 this->PoolParamsToDims(&pool_params, &filter_dims, &strides, in Compute()
253 MklPoolParameters pool_params; in Compute() local
257 this->InitMklPoolParameters(context, &pool_params, orig_input_mkl_shape, in Compute()
261 this->PoolParamsToDims(&pool_params, &filter_dims, &strides, in Compute()
281 this->GetOutputDims(pool_params, &output_dims_mkl_order); in Compute()
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dperception_ops_utils.cc107 TfLitePoolParams pool_params; in CreateCustomOptions() local
113 pool_params.filter_height = pool_size[0]; in CreateCustomOptions()
114 pool_params.filter_width = pool_size[1]; in CreateCustomOptions()
121 pool_params.stride_height = strides[0]; in CreateCustomOptions()
122 pool_params.stride_width = strides[1]; in CreateCustomOptions()
131 pool_params.padding = kTfLitePaddingValid; in CreateCustomOptions()
133 pool_params.padding = kTfLitePaddingSame; in CreateCustomOptions()
139 pool_params.activation = kTfLiteActNone; in CreateCustomOptions()
140 pool_params.computed.padding = TfLitePaddingValues{0, 0, 0, 0}; in CreateCustomOptions()
142 custom_option_buffer.assign(reinterpret_cast<char*>(&pool_params), in CreateCustomOptions()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dpool_2d_builder.cc38 const TfLitePoolParams* pool_params = in PopulateSubGraph() local
42 if (pool_params->padding == kTfLitePaddingSame) { in PopulateSubGraph()
44 } else if (pool_params->padding == kTfLitePaddingValid) { in PopulateSubGraph()
50 filter_shape_ = {1, pool_params->filter_height, pool_params->filter_width, 1}; in PopulateSubGraph()
55 stride_shape_ = {1, pool_params->stride_height, pool_params->stride_width, 1}; in PopulateSubGraph()
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dpooling.cc138 cmsis_nn_pool_params pool_params; in AverageEvalQuantized() local
139 pool_params.stride.h = params->stride_height; in AverageEvalQuantized()
140 pool_params.stride.w = params->stride_width; in AverageEvalQuantized()
141 pool_params.padding.h = data.padding.height; in AverageEvalQuantized()
142 pool_params.padding.w = data.padding.width; in AverageEvalQuantized()
143 pool_params.activation.min = data.activation_min; in AverageEvalQuantized()
144 pool_params.activation.max = data.activation_max; in AverageEvalQuantized()
160 arm_avgpool_s8(&ctx, &pool_params, &input_dims, in AverageEvalQuantized()
228 cmsis_nn_pool_params pool_params; in MaxEvalInt8() local
229 pool_params.stride.h = params->stride_height; in MaxEvalInt8()
[all …]
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc835 const TfLitePoolParams* pool_params = in VisitNode() local
839 node, context->tensors, pool_params, in VisitNode()
914 const TfLitePoolParams* pool_params = in VisitNode() local
918 context->tensors, pool_params, in VisitNode()
1017 TfLitePoolParams pool_params = {kTfLitePaddingUnknown}; in VisitNode() local
1018 std::memcpy(&pool_params, node->custom_initial_data, in VisitNode()
1023 &pool_params, xnnpack_tensors); in VisitNode()
1025 TfLitePoolParams pool_params = {kTfLitePaddingUnknown}; in VisitNode() local
1026 std::memcpy(&pool_params, node->custom_initial_data, in VisitNode()
1031 &pool_params, xnnpack_tensors); in VisitNode()
[all …]
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dutils.cc237 const TfLitePoolParams* pool_params = in IsNodeSupportedByHexagon() local
239 return pool_params->activation == kTfLiteActNone; in IsNodeSupportedByHexagon()
244 const TfLitePoolParams* pool_params = in IsNodeSupportedByHexagon() local
247 pool_params->activation == kTfLiteActNone); in IsNodeSupportedByHexagon()
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data_test.cc31 TfLitePoolParams pool_params; in TEST() local
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_export.cc408 TfLitePoolParams pool_params; in GetTflitePoolParams() local
409 pool_params.stride_height = op.stride_h().getSExtValue(); in GetTflitePoolParams()
410 pool_params.stride_width = op.stride_w().getSExtValue(); in GetTflitePoolParams()
411 pool_params.filter_height = op.filter_h().getSExtValue(); in GetTflitePoolParams()
412 pool_params.filter_width = op.filter_w().getSExtValue(); in GetTflitePoolParams()
415 pool_params.padding = *padding; in GetTflitePoolParams()
416 pool_params.activation = kTfLiteActNone; in GetTflitePoolParams()
417 pool_params.computed.padding = TfLitePaddingValues{0, 0, 0, 0}; in GetTflitePoolParams()
418 return pool_params; in GetTflitePoolParams()