Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmkl_pooling_ops_common.h509 MklPoolParameters* pool_params, in InitMklPoolParameters() argument
513 pool_params->Init(context, this->ksize_, this->stride_, this->padding_, in InitMklPoolParameters()
516 pool_params->Init(context, this->ksize_, this->stride_, this->padding_, in InitMklPoolParameters()
521 void PoolParamsToDims(const MklPoolParameters* pool_params, in PoolParamsToDims() argument
528 memory::dims({pool_params->window_rows, pool_params->window_cols}); in PoolParamsToDims()
530 memory::dims({pool_params->row_stride, pool_params->col_stride}); in PoolParamsToDims()
531 *padding_left = memory::dims({static_cast<int>(pool_params->pad_top), in PoolParamsToDims()
532 static_cast<int>(pool_params->pad_left)}); in PoolParamsToDims()
533 *padding_right = memory::dims({static_cast<int>(pool_params->pad_bottom), in PoolParamsToDims()
534 static_cast<int>(pool_params->pad_right)}); in PoolParamsToDims()
[all …]
Dmkl_avgpooling_op.cc79 MklPoolParameters pool_params; in Compute() local
81 pool_params.Init(context, ksize_, stride_, padding_, data_format_, in Compute()
84 pool_params.Init(context, ksize_, stride_, padding_, data_format_, in Compute()
89 ExtractMklOpParams(context, data_format_, pool_params, &mkl_context.params); in Compute()
257 MklPoolParameters pool_params; in Compute() local
258 pool_params.Init(context, ksize_, stride_, padding_, data_format_, in Compute()
267 ExtractMklOpParams(context, data_format_, pool_params, &mkl_context.params); in Compute()
455 MklPoolParameters pool_params; in Compute() local
460 this->InitMklPoolParameters(context, &pool_params, dnn_shape_input, in Compute()
467 this->GetOutputDims(pool_params, &output_dims_mkl_order); in Compute()
[all …]
Dmkl_maxpooling_op.cc84 MklPoolParameters pool_params; in Compute() local
86 pool_params.Init(context, ksize_, stride_, padding_, data_format_, in Compute()
89 context, (pool_params.depth_window == 1), in Compute()
93 pool_params.Init(context, ksize_, stride_, padding_, data_format_, in Compute()
99 ExtractMklOpParams(context, data_format_, pool_params, &mkl_context.params); in Compute()
251 MklPoolParameters pool_params; in Compute() local
253 pool_params.Init(context, ksize_, stride_, padding_, data_format_, in Compute()
256 context, (pool_params.depth_window == 1), in Compute()
260 pool_params.Init(context, ksize_, stride_, padding_, data_format_, in Compute()
265 ExtractMklOpParams(context, data_format_, pool_params, &mkl_context.params); in Compute()
[all …]
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc198 if (auto* pool_params = op->builtin_options_as_Pool2DOptions()) { in ParseOpData() local
199 params->padding = parse_padding(pool_params->padding()); in ParseOpData()
200 params->stride_width = pool_params->stride_w(); in ParseOpData()
201 params->stride_height = pool_params->stride_h(); in ParseOpData()
202 params->filter_width = pool_params->filter_width(); in ParseOpData()
203 params->filter_height = pool_params->filter_height(); in ParseOpData()
205 parse_activation(pool_params->fused_activation_function()); in ParseOpData()
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data_test.cc31 TfLitePoolParams pool_params; in TEST() local