HomeSort by relevance Sort by last modified time
    Searched refs:Dim_C (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_conv_ops.h 142 mkldnn_sizes[MklDnnDims::Dim_C] = input_depth;
457 mkldnn_sizes[MklDnnDims::Dim_C] = out_depth;
mkl_fused_batch_norm_op.cc 574 depth_ = dnn_shape_src.DimSize(MklDnnDims::Dim_C);
    [all...]
  /external/tensorflow/tensorflow/core/util/
mkl_util.h 94 Dim_C = 1,
531 return MklDnnDims::Dim_C;
695 data_.map_[GetTensorDimIndex<2>(data_format, 'C')] = MklDnnDims::Dim_C;
714 return TfDimIdx(d) == MklDnnDims::Dim_C;
    [all...]

Completed in 516 milliseconds