Searched refs:kernel_fallback_tensor (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | BUILD | 152 ":kernel_fallback_tensor",
170 name = "kernel_fallback_tensor",
171 srcs = ["kernel_fallback_tensor.cc"],
172 hdrs = ["kernel_fallback_tensor.h"],
217 ":kernel_fallback_tensor",
266 ":kernel_fallback_tensor",
296 ":kernel_fallback_tensor",
321 ":kernel_fallback_tensor",
354 ":kernel_fallback_tensor",
|
| /external/tensorflow/tensorflow/core/runtime_fallback/conversion/ |
| D | BUILD | 32 "//tensorflow/core/runtime_fallback/kernel:kernel_fallback_tensor",
|
| /external/tensorflow/tensorflow/core/tfrt/utils/ |
| D | tensor_util.cc | 182 if (auto* kernel_fallback_tensor = in CreateTFTensorFromTensorHandle() local
184 return *kernel_fallback_tensor->GetTensor(); in CreateTFTensorFromTensorHandle()
|
| D | BUILD | 92 "//tensorflow/core/runtime_fallback/kernel:kernel_fallback_tensor",
116 "//tensorflow/core/runtime_fallback/kernel:kernel_fallback_tensor",
|
| /external/tensorflow/tensorflow/core/tfrt/runtime/ |
| D | BUILD | 38 "//tensorflow/core/runtime_fallback/kernel:kernel_fallback_tensor",
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | BUILD | 80 "//tensorflow/core/runtime_fallback/kernel:kernel_fallback_tensor",
|
| D | runtime_fallback_kernels.cc | 1007 AsyncValueRef<tfrt::Tensor> kernel_fallback_tensor = in FallbackTensorToCoreRTTensorHandleInternal() local
1009 auto metadata = kernel_fallback_tensor.get().metadata(); in FallbackTensorToCoreRTTensorHandleInternal()
1016 metadata, std::move(kernel_fallback_tensor)); in FallbackTensorToCoreRTTensorHandleInternal()
|