Searched refs:Dim_W (Results 1 – 3 of 3) sorted by relevance
144 mkldnn_sizes[MklDnnDims::Dim_W] = input_cols; in GetInputSizeInMklOrder()241 mkldnn_sizes[MklDnnDims::Dim_W] = filter_cols; in GetFilterSizeInMklOrder()459 mkldnn_sizes[MklDnnDims::Dim_W] = static_cast<int>(out_cols);
492 bwd_output_dims[MklDnnDims::Dim_W], in Compute()682 output_dims_mkl_order[MklDnnDims::Dim_W], in AllocateOutputTensor()
96 Dim_W = 3, enumerator535 return MklDnnDims::Dim_W; in GetMklDnnTensorDimIndex()693 data_.map_[GetTensorDimIndex<2>(data_format, 'W')] = MklDnnDims::Dim_W; in SetTfDimOrder()724 return TfDimIdx(d) == MklDnnDims::Dim_W; in IsMklWidthDim()