HomeSort by relevance Sort by last modified time
    Searched refs:pool_params (Results 1 - 4 of 4) sorted by null

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

Completed in 500 milliseconds