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

  /external/tensorflow/tensorflow/core/kernels/
mkl_conv_ops.h 121 mkldnn_sizes[MklDnnDims::Dim_H] = input_rows;
175 mkldnn_sizes[MklDnnDims::Dim_H] = filter_rows;
260 mkldnn_sizes[MklDnnDims::Dim_H] = static_cast<int>(out_rows);
mkl_conv_grad_filter_ops.cc 583 TensorShape output_tf_shape({output_dims_mkl_order[MklDnnDims::Dim_H],
  /external/tensorflow/tensorflow/core/util/
mkl_util.h 62 Dim_H = 2,
440 return MklDnnDims::Dim_H;
571 data_.map_[GetTensorDimIndex<2>(data_format, 'H')] = MklDnnDims::Dim_H;
605 return TfDimIdx(d) == MklDnnDims::Dim_H;
    [all...]

Completed in 188 milliseconds