| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_input_conversion_op.cc | 144 const Tensor* mkl_tensor; variable
151 mkl_tensor = &input_tensor_0;
158 mkl_tensor = &input_tensor_1;
203 mkl_tensor->shape(), mkl_output_mkl_shape);
377 const Tensor* mkl_tensor; variable
384 mkl_tensor = &input_tensor_0;
391 mkl_tensor = &input_tensor_1;
436 mkl_tensor->shape(), mkl_output_mkl_shape);
|
| mkl_concat_op.cc | 517 Tensor* mkl_tensor = nullptr; local
524 mkl_tensor_tf_shape, &mkl_tensor); local
526 mkl_tensor->flat<uint8>().data(),
527 mkl_tensor->flat<uint8>().size() * sizeof(uint8));
547 Tensor* mkl_tensor = nullptr; local
554 mkl_tensor_tf_shape, &mkl_tensor); local
556 mkl_tensor->flat<uint8>().data(),
557 mkl_tensor->flat<uint8>().size() * sizeof(uint8));
[all...] |
| /external/tensorflow/tensorflow/core/util/ |
| mkl_util.h | 679 inline Tensor ConvertMklToTF(OpKernelContext* context, const Tensor& mkl_tensor,
693 void* input_buffer = const_cast<T*>(mkl_tensor.flat<T>().data());
696 if (mkl_tensor.NumElements() != 0) {
704 inline Tensor ConvertMklToTF(OpKernelContext* context, const Tensor& mkl_tensor,
[all...] |
