| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_avgpooling_op.cc | 79 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()
457 MklPoolParameters pool_params; in Compute() local
459 this->ConfigureInput(context, dnn_shape_input, input_tensor, &pool_params, in Compute()
466 this->GetOutputDims(pool_params, &output_dims_mkl_order); in Compute()
[all …]
|
| D | mkl_maxpooling_op.cc | 83 MklPoolParameters pool_params; in Compute() local
85 pool_params.Init(context, ksize_, stride_, padding_, data_format_, in Compute()
88 context, (pool_params.depth_window == 1), in Compute()
92 pool_params.Init(context, ksize_, stride_, padding_, data_format_, in Compute()
98 ExtractMklOpParams(context, data_format_, pool_params, &mkl_context.params); in Compute()
249 MklPoolParameters pool_params; in Compute() local
251 pool_params.Init(context, ksize_, stride_, padding_, data_format_, in Compute()
254 context, (pool_params.depth_window == 1), in Compute()
258 pool_params.Init(context, ksize_, stride_, padding_, data_format_, in Compute()
263 ExtractMklOpParams(context, data_format_, pool_params, &mkl_context.params); in Compute()
[all …]
|
| D | mkl_pooling_ops_common.h | 154 MklPoolParameters* pool_params, in InitMklPoolParameters() argument
158 pool_params->Init(context, this->ksize_, this->stride_, this->padding_, in InitMklPoolParameters()
161 pool_params->Init(context, this->ksize_, this->stride_, this->padding_, in InitMklPoolParameters()
197 MklPoolParameters* pool_params, in ConfigureInput() argument
199 CHECK_NOTNULL(pool_params); in ConfigureInput()
209 this->InitMklPoolParameters(context, pool_params, input_mkl_shape, in ConfigureInput()
332 memory::dims* original_input_dims_nchw, MklPoolParameters* pool_params, in ConfigureOriginalInput() argument
335 CHECK_NOTNULL(pool_params); in ConfigureOriginalInput()
336 this->InitMklPoolParameters(context, pool_params, original_input_mkl_shape, in ConfigureOriginalInput()
352 const MklPoolParameters& pool_params, in ConfigureOriginalOutput() argument
[all …]
|
| /external/tensorflow/tensorflow/contrib/lite/ |
| D | model.cc | 297 if (auto* pool_params = op->builtin_options_as_Pool2DOptions()) { in ParseOpData() local
298 params->padding = parse_padding(pool_params->padding()); in ParseOpData()
299 params->stride_width = pool_params->stride_w(); in ParseOpData()
300 params->stride_height = pool_params->stride_h(); in ParseOpData()
301 params->filter_width = pool_params->filter_width(); in ParseOpData()
302 params->filter_height = pool_params->filter_height(); in ParseOpData()
304 parse_activation(pool_params->fused_activation_function()); in ParseOpData()
|