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

  /external/tensorflow/tensorflow/core/kernels/
mkl_pooling_ops_common.h 332 memory::dims* original_input_dims_nchw, MklPoolParameters* pool_params,
334 CHECK_NOTNULL(original_input_dims_nchw);
339 *original_input_dims_nchw =
347 : memory::desc(*original_input_dims_nchw, MklDnnType<T>(),
mkl_avgpooling_op.cc 575 memory::dims output_dims_mkl_order, original_input_dims_nchw; variable
579 &original_input_dims_nchw, &pool_params, &original_input_shape);
596 original_input_dims_nchw,
627 context, pool_bkwd_prim_desc, original_input_dims_nchw,

Completed in 102 milliseconds