Searched refs:Dim3d_W (Results 1 – 4 of 4) sorted by relevance
175 mkldnn_sizes[MklDnnDims3D::Dim3d_W] = input_cols; in GetInputSizeInMklOrder()276 mkldnn_sizes[MklDnnDims3D::Dim3d_W] = filter_cols; in GetFilterSizeInMklOrder()476 mkldnn_sizes[MklDnnDims3D::Dim3d_W] = static_cast<int>(out_cols);
523 diff_filter_dims[MklDnnDims3D::Dim3d_W], in Compute()
634 mkldnn_sizes[MklDnnDims3D::Dim3d_W] = input_md.data.dims[4]; in ConfigureInput()
87 Dim3d_W = 4, enumerator362 return MklDnnDims3D::Dim3d_W; in GetMklDnnTensor3DDimIndex()504 MklDnnDims3D::Dim3d_W; in SetTfDimOrder()