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

  /external/tensorflow/tensorflow/core/util/
mkl_util.h 168 bool IsMklChannelDim(int d) const { return tf_dim_idx(d) == MklDims::C; }
184 IsMklChannelDim(GetTensorDimIndex<2>(data_format, 'C')) &&
194 IsMklChannelDim(GetTensorDimIndex<2>(data_format, 'C')) &&
589 inline bool IsMklChannelDim(int d) const {
613 IsMklChannelDim(GetTensorDimIndex<2>(data_format, 'C')) &&
623 IsMklChannelDim(GetTensorDimIndex<2>(data_format, 'C')) &&
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
mkl_lrn_op.cc 779 } else if (!src_dnn_shape.IsMklChannelDim(src_dnn_shape.GetDimension() -
    [all...]
mkl_concat_op.cc 300 if (!s.IsMklChannelDim(concat_dim)) {
    [all...]

Completed in 206 milliseconds