Searched refs:GetOutputDims (Results 1 - 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ | ||
H A D | mkl_pooling_ops_common.h | 144 void GetOutputDims(const MklPoolParameters& mkl_pool_params, function in class:tensorflow::MklPoolingOpBase 355 this->GetOutputDims(pool_params, &output_dims_mkl_order); |
H A D | mkl_conv_ops.h | 476 auto bwd_output_dims = GetOutputDims(fwd_input_dims, fwd_filter_dims); 528 virtual const memory::dims& GetOutputDims( |
H A D | mkl_conv_grad_input_ops.cc | 407 const memory::dims& GetOutputDims(const memory::dims& fwd_input_dims, function in class:tensorflow::MklConv2DCustomBackpropInputOp |
H A D | mkl_conv_grad_filter_ops.cc | 480 const memory::dims& GetOutputDims(const memory::dims& fwd_input_dims, function in class:tensorflow::MklConv2DCustomBackpropFilterOp |
H A D | mkl_avgpooling_op.cc | 466 this->GetOutputDims(pool_params, &output_dims_mkl_order); |
H A D | mkl_maxpooling_op.cc | 534 this->GetOutputDims(pool_params, &output_dims_mkl_order); |
Completed in 180 milliseconds