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

  /external/tensorflow/tensorflow/core/kernels/
mkl_pooling_ops_common.h 144 void GetOutputDims(const MklPoolParameters& mkl_pool_params,
355 this->GetOutputDims(pool_params, &output_dims_mkl_order);
mkl_conv_ops.h 476 auto bwd_output_dims = GetOutputDims(fwd_input_dims, fwd_filter_dims);
528 virtual const memory::dims& GetOutputDims(
mkl_conv_grad_filter_ops.cc 480 const memory::dims& GetOutputDims(const memory::dims& fwd_input_dims,
mkl_conv_grad_input_ops.cc 407 const memory::dims& GetOutputDims(const memory::dims& fwd_input_dims,
mkl_avgpooling_op.cc 466 this->GetOutputDims(pool_params, &output_dims_mkl_order);
mkl_maxpooling_op.cc 534 this->GetOutputDims(pool_params, &output_dims_mkl_order);
    [all...]

Completed in 168 milliseconds