Searched defs:src_tensor (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_relu_op.cc | 460 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute() local
571 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute() local
730 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute_Scalar() local
760 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute_Scalar() local
793 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute_Scalar() local
827 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute_Scalar() local
865 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute_Scalar() local
898 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute_Scalar() local
937 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute_Scalar() local
968 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute_Scalar() local
[all …]
|
| D | mkl_softmax_op.cc | 52 const Tensor& src_tensor = MklGetInput(context, src_idx); in Compute() local
|
| D | debug_ops.h | 62 const Tensor& src_tensor = context->input(0); in Compute() local
|
| D | mkl_lrn_op.cc | 95 const Tensor& src_tensor = MklGetInput(context, kIdxInput); in Compute() local
289 const Tensor& src_tensor = MklGetInput(context, kIdxInput); in SanityCheckInputs() local
|
| D | mkl_fused_batch_norm_op.cc | 518 const Tensor& src_tensor = MklGetInput(context, kSrcIndex); in Compute() local
842 const Tensor& src_tensor = MklGetInput(context, kSrcIndex); in Compute() local
|
| D | mkl_conv_grad_input_ops.cc | 317 const Tensor& src_tensor = MklGetInput(context, kInputIdx); in Compute() local
|
| D | mkl_conv_grad_filter_ops.cc | 380 const Tensor& src_tensor = MklGetInput(context, kInputIdx); in Compute() local
|
| D | mkl_conv_ops.cc | 923 const Tensor& src_tensor = MklGetInput(context, kInputIndex_Src); in Compute() local
|
| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| D | sycl_util.h | 34 Tensor const& src_tensor, Tensor* dst_tensor) { in SYCLmemcpy()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | while.cc | 40 const TfLiteTensor* src_tensor = in CopyTensorsShapeAndType() local
61 const TfLiteTensor* src_tensor = in CopyTensorsData() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | hierarchical_tree_broadcaster.cc | 400 const Tensor* src_tensor, in DispatchSend()
|