Home
last modified time | relevance | path

Searched refs:Dim_C (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_conv_ops.h142 mkldnn_sizes[MklDnnDims::Dim_C] = input_depth; in GetInputSizeInMklOrder()
457 mkldnn_sizes[MklDnnDims::Dim_C] = out_depth;
Dmkl_fused_batch_norm_op.cc574 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()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h94 Dim_C = 1, enumerator
531 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()