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

  /external/tensorflow/tensorflow/core/kernels/
mkl_conv_grad_input_ops.cc 422 auto bwd_diff_src_dims = GetOutputDims(fwd_src_dims, fwd_filter_dims);
522 const memory::dims& GetOutputDims(const memory::dims& fwd_input_dims,
mkl_avgpooling_op.cc 467 this->GetOutputDims(pool_params, &output_dims_mkl_order);
619 this->GetOutputDims(pool_params, &output_dims_mkl_order);
mkl_conv_grad_filter_ops.cc 481 auto bwd_output_dims = GetOutputDims(fwd_src_dims, fwd_filter_dims);
649 const memory::dims& GetOutputDims(const memory::dims& fwd_input_dims,
mkl_maxpooling_op.cc 537 this->GetOutputDims(pool_params, &output_dims_mkl_order);
731 this->GetOutputDims(pool_params, &output_dims_mkl_order);
    [all...]
mkl_pooling_ops_common.h 490 void GetOutputDims(const MklPoolParameters& mkl_pool_params,

Completed in 192 milliseconds