Searched refs:Dim_C (Results 1 – 3 of 3) sorted by relevance
148 mkldnn_sizes[MklDnnDims::Dim_C] = input_depth; in GetInputSizeInMklOrder()466 mkldnn_sizes[MklDnnDims::Dim_C] = out_depth;
733 depth_ = dnn_shape_src.DimSize(MklDnnDims::Dim_C); in Compute()1134 depth_ = dnn_shape_src.DimSize(MklDnnDims::Dim_C); in Compute()1136 depth_ = dnn_shape_diff_dst.DimSize(MklDnnDims::Dim_C); in Compute()
72 Dim_C = 1, enumerator340 return MklDnnDims::Dim_C; in GetMklDnnTensorDimIndex()454 data_.map_[1] = MklDnnDims::Dim_C; in SetTfLayout()514 data_.map_[GetTensorDimIndex<2>(data_format, 'C')] = MklDnnDims::Dim_C; in SetTfDimOrder()533 return TfDimIdx(d) == MklDnnDims::Dim_C; in IsMklChannelDim()