Searched refs:is_mkl_tensor_ (Results 1 – 1 of 1) sorted by relevance
251 bool is_mkl_tensor_ = false; member313 inline const bool IsMklTensor() const { return data_.is_mkl_tensor_; } in IsMklTensor()315 data_.is_mkl_tensor_ = is_mkl_tensor; in SetMklTensor()378 if (data_.is_mkl_tensor_) { in GetSizesAsMklDnnDims()385 CHECK_EQ(data_.is_mkl_tensor_, true); in GetSizesAsMklDnnDims()398 CHECK_EQ(data_.is_mkl_tensor_, true); in GetTfShape()590 CHECK(buf_size >= sizeof(data_.is_mkl_tensor_)) in DeSerializeMklDnnShape()