Searched refs:Dim_W (Results 1 – 3 of 3) sorted by relevance
150 mkldnn_sizes[MklDnnDims::Dim_W] = input_cols; in GetInputSizeInMklOrder()247 mkldnn_sizes[MklDnnDims::Dim_W] = filter_cols; in GetFilterSizeInMklOrder()468 mkldnn_sizes[MklDnnDims::Dim_W] = static_cast<int>(out_cols);
490 diff_filter_dims[MklDnnDims::Dim_W], in Compute()690 output_dims_mkl_order[MklDnnDims::Dim_W], in AllocateOutputTensor()
74 Dim_W = 3, enumerator344 return MklDnnDims::Dim_W; in GetMklDnnTensorDimIndex()512 data_.map_[GetTensorDimIndex<2>(data_format, 'W')] = MklDnnDims::Dim_W; in SetTfDimOrder()545 return TfDimIdx(d) == MklDnnDims::Dim_W; in IsMklWidthDim()