| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_relu_op.cc | 460 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute() local
464 if (src_tensor.dims() == 0) { in Compute()
477 src_dims = TFShapeToMklDnnDims(src_tensor.shape()); in Compute()
490 const T* src_data = src_tensor.flat<T>().data(); in Compute()
493 src.SetUsrMem(src_md, &src_tensor); in Compute()
518 tf_shape_dst = src_tensor.shape(); in Compute()
571 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute() local
579 int src_dims_size = src_tensor.dims(); in Compute()
590 src_dims = TFShapeToMklDnnDims(src_tensor.shape()); in Compute()
615 src_dims = (src_tensor.dims() == 4) in Compute()
[all …]
|
| D | debug_ops.h | 62 const Tensor& src_tensor = context->input(0); in Compute() local
64 if (src_tensor.IsInitialized() && in Compute()
65 DataTypeCanUseMemcpy(src_tensor.dtype()) && in Compute()
69 OP_REQUIRES_OK(context, context->allocate_output(0, src_tensor.shape(), in Compute()
83 device, device_ctxt, &src_tensor, copied_tensor, in Compute()
88 *copied_tensor = tensor::DeepCopy(src_tensor); in Compute()
97 SYCLmemcpy(context->eigen_sycl_device(), src_tensor, copied_tensor); in Compute()
99 *copied_tensor = tensor::DeepCopy(src_tensor); in Compute()
102 *copied_tensor = tensor::DeepCopy(src_tensor); in Compute()
107 context->set_output(0, src_tensor); in Compute()
|
| D | mkl_fused_batch_norm_op.cc | 518 const Tensor& src_tensor = MklGetInput(context, kSrcIndex); in Compute() local
532 src_tensor.shape().DebugString())); in Compute()
534 tf_shape_src = src_tensor.shape(); in Compute()
535 OP_REQUIRES(context, src_tensor.dims() == 4, in Compute()
537 src_tensor.shape().DebugString())); in Compute()
613 : TFShapeToMklDnnDimsInNCHW(src_tensor.shape(), tensor_format_); in Compute()
645 const T* src_data = src_tensor.flat<T>().data(); in Compute()
647 src.SetUsrMem(src_md, &src_tensor); in Compute()
666 : src_tensor.shape().dims(); in Compute()
842 const Tensor& src_tensor = MklGetInput(context, kSrcIndex); in Compute() local
[all …]
|
| D | mkl_softmax_op.cc | 52 const Tensor& src_tensor = MklGetInput(context, src_idx); in Compute() local
61 : src_tensor.shape(); in Compute()
127 src.SetUsrMem(src_md, &src_tensor); in Compute()
|
| D | mkl_lrn_op.cc | 95 const Tensor& src_tensor = MklGetInput(context, kIdxInput); in Compute() local
107 MklDefaultToEigen(context, src_tensor); in Compute()
112 ConvertMklToTF<T>(context, src_tensor, src_dnn_shape); in Compute()
124 TensorShape tf_output_shape = src_tensor.shape(); in Compute()
133 src_dnn_data.SetUsrMem(src_md, &src_tensor); in Compute()
289 const Tensor& src_tensor = MklGetInput(context, kIdxInput); in SanityCheckInputs() local
296 FastBoundsCheck(src_tensor.NumElements(), in SanityCheckInputs()
300 OP_REQUIRES(context, src_tensor.dims() == 4, in SanityCheckInputs()
303 FastBoundsCheck(src_tensor.NumElements(), in SanityCheckInputs()
|
| D | mkl_conv_grad_filter_ops.cc | 380 const Tensor& src_tensor = MklGetInput(context, kInputIdx); in Compute() local
396 TensorShape src_tf_shape = MakeInputTfShape(context, src_tensor); in Compute()
540 src.SetUsrMem(fwd_src_md, &src_tensor); in Compute()
544 src_data = static_cast<T*>(const_cast<T*>(src_tensor.flat<T>().data())); in Compute()
|
| D | mkl_conv_grad_input_ops.cc | 317 const Tensor& src_tensor = MklGetInput(context, kInputIdx); in Compute() local
333 TensorShape src_tf_shape = MakeInputTfShape(context, src_tensor); in Compute()
|
| D | mkl_conv_ops.cc | 923 const Tensor& src_tensor = MklGetInput(context, kInputIndex_Src); in Compute() local
1033 src.SetUsrMem(src_md, &src_tensor); in Compute()
1102 src.SetUsrMem(src_md, &src_tensor); in Compute()
1107 const_cast<Tinput*>(src_tensor.flat<Tinput>().data())); in Compute()
|
| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| D | sycl_util.h | 34 Tensor const& src_tensor, Tensor* dst_tensor) { in SYCLmemcpy() argument
35 const size_t size = src_tensor.TotalBytes(); in SYCLmemcpy()
37 void const* src_ptr = GetBase(&src_tensor); in SYCLmemcpy()
41 switch (src_tensor.dtype()) { in SYCLmemcpy()
73 LOG(FATAL) << "Unknown data type " << src_tensor.dtype(); in SYCLmemcpy()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | while.cc | 40 const TfLiteTensor* src_tensor = in CopyTensorsShapeAndType() local
42 std::vector<int> dims(src_tensor->dims->data, in CopyTensorsShapeAndType()
43 src_tensor->dims->data + src_tensor->dims->size); in CopyTensorsShapeAndType()
46 dst_tensor->type = src_tensor->type; in CopyTensorsShapeAndType()
61 const TfLiteTensor* src_tensor = in CopyTensorsData() local
64 TF_LITE_ENSURE_EQ(context, src_tensor->bytes, dst_tensor->bytes); in CopyTensorsData()
65 memcpy(dst_tensor->data.raw, src_tensor->data.raw, src_tensor->bytes); in CopyTensorsData()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | hierarchical_tree_broadcaster.h | 71 const Tensor* src_tensor, const StatusCallback& done);
|
| D | hierarchical_tree_broadcaster.cc | 400 const Tensor* src_tensor, in DispatchSend() argument
415 src_tensor, col_ctx_->device_locality, done); in DispatchSend()
|