Searched refs:Dim_C (Results 1 – 3 of 3) sorted by relevance
142 mkldnn_sizes[MklDnnDims::Dim_C] = input_depth; in GetInputSizeInMklOrder()457 mkldnn_sizes[MklDnnDims::Dim_C] = out_depth;
574 depth_ = dnn_shape_src.DimSize(MklDnnDims::Dim_C); in Compute()902 depth_ = dnn_shape_src.DimSize(MklDnnDims::Dim_C); in Compute()904 depth_ = dnn_shape_diff_dst.DimSize(MklDnnDims::Dim_C); in Compute()
94 Dim_C = 1, enumerator531 return MklDnnDims::Dim_C; in GetMklDnnTensorDimIndex()695 data_.map_[GetTensorDimIndex<2>(data_format, 'C')] = MklDnnDims::Dim_C; in SetTfDimOrder()714 return TfDimIdx(d) == MklDnnDims::Dim_C; in IsMklChannelDim()