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

  /external/tensorflow/tensorflow/core/util/
mkl_util.h 483 shape[idx] = data_.sizes_[TfDimIdx(idx)];
582 inline size_t TfDimIdx(int index) const { return data_.map_[index]; }
584 return data_.sizes_[TfDimIdx(index)];
590 return TfDimIdx(d) == MklDnnDims::Dim_C;
595 return TfDimIdx(d) == MklDnnDims::Dim_N;
600 return TfDimIdx(d) == MklDnnDims::Dim_W;
605 return TfDimIdx(d) == MklDnnDims::Dim_H;
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
mkl_concat_op.cc 743 if (are_all_mkl_inputs) concat_dim = input_shapes[0].TfDimIdx(concat_dim);
    [all...]

Completed in 359 milliseconds