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

  /external/tensorflow/tensorflow/core/kernels/
mkl_concat_op.cc 688 bool is_mkl_tensor = input_shapes[k].IsMklTensor(); local
692 src_dims = (is_mkl_tensor)
698 is_mkl_tensor ? input_shapes[k].GetMklLayout() :
    [all...]
  /external/tensorflow/tensorflow/core/util/
mkl_util.h 419 inline void SetMklTensor(bool is_mkl_tensor) {
420 data_.is_mkl_tensor_ = is_mkl_tensor;
648 const bool is_mkl_tensor = *reinterpret_cast<const bool*>(buf); local
649 if (is_mkl_tensor) { // If it is an MKL Tensor then read the rest
    [all...]

Completed in 62 milliseconds