| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_softmax_op.cc | 53 // src_tensor now points to the 0-th input of global data struct "context"
55 const Tensor& src_tensor = MklGetInput(context, src_idx); variable
61 // src_dims is the dimenstion of src_tensor
65 : src_tensor.shape();
84 // Basically following two functions maps the TF "src_tensor" to mkl
89 src.SetUsrMem(src_md, &src_tensor);
|
| debug_ops.h | 62 const Tensor& src_tensor = context->input(0); variable
64 if (src_tensor.IsInitialized() &&
65 DataTypeCanUseMemcpy(src_tensor.dtype()) &&
69 OP_REQUIRES_OK(context, context->allocate_output(0, src_tensor.shape(),
83 device, device_ctxt, &src_tensor, copied_tensor,
88 *copied_tensor = tensor::DeepCopy(src_tensor);
97 SYCLmemcpy(context->eigen_sycl_device(), src_tensor, copied_tensor);
99 *copied_tensor = tensor::DeepCopy(src_tensor);
102 *copied_tensor = tensor::DeepCopy(src_tensor);
107 context->set_output(0, src_tensor);
[all...] |
| mkl_conv_ops.cc | 517 const Tensor& src_tensor = MklGetInput(context, kInputIndex_Src); variable
577 src.SetUsrMem(src_md, &src_tensor);
[all...] |
| mkl_relu_op.cc | 387 const Tensor& src_tensor = MklGetInput(context, src_index); variable
392 if (src_tensor.dims() == 0) {
406 auto src_dims = TFShapeToMklDnnDims(src_tensor.shape());
411 src.SetUsrMem(src_md, &src_tensor);
436 tf_shape_dst = src_tensor.shape();
483 const Tensor& src_tensor = MklGetInput(context, src_index); local
491 int src_dims_size = src_tensor.dims();
513 auto src_dims = TFShapeToMklDnnDims(src_tensor.shape());
541 auto src_dims = TFShapeToMklDnnDimsInNCHW(src_tensor.shape(),
552 src.SetUsrMem(src_md, &src_tensor);
642 const Tensor& src_tensor = MklGetInput(context, src_index); local
672 const Tensor& src_tensor = MklGetInput(context, src_index); local
705 const Tensor& src_tensor = MklGetInput(context, src_index); local
739 const Tensor& src_tensor = MklGetInput(context, src_index); local
777 const Tensor& src_tensor = MklGetInput(context, src_index); local
810 const Tensor& src_tensor = MklGetInput(context, src_index); local
[all...] |
| mkl_fused_batch_norm_op.cc | 712 const Tensor& src_tensor = MklGetInput(context, kSrcIndex); variable
1055 const Tensor& src_tensor = MklGetInput(context, kSrcIndex); variable
[all...] |
