Home
last modified time | relevance | path

Searched refs:Dim3d_W (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_conv_ops.h169 mkldnn_sizes[MklDnnDims3D::Dim3d_W] = input_cols; in GetInputSizeInMklOrder()
270 mkldnn_sizes[MklDnnDims3D::Dim3d_W] = filter_cols; in GetFilterSizeInMklOrder()
467 mkldnn_sizes[MklDnnDims3D::Dim3d_W] = static_cast<int>(out_cols);
Dmkl_conv_grad_filter_ops.cc524 bwd_output_dims[MklDnnDims3D::Dim3d_W], in Compute()
Dmkl_pooling_ops_common.h632 mkldnn_sizes[MklDnnDims3D::Dim3d_W] = input_md.data.dims[4]; in ConfigureInput()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h109 Dim3d_W = 4, enumerator
553 return MklDnnDims3D::Dim3d_W; in GetMklDnnTensor3DDimIndex()
685 MklDnnDims3D::Dim3d_W; in SetTfDimOrder()