Home
last modified time | relevance | path

Searched refs:fallback_tensor (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/
H A Dfallback_tensor_test.cc57 FallbackTensor fallback_tensor(tensor); in TEST() local
58 EXPECT_FALSE(fallback_tensor.is_immutable()); in TEST()
60 ASSERT_EQ(fallback_tensor.tensor().NumElements(), 1); in TEST()
61 ASSERT_EQ(fallback_tensor.tensor().dtype(), tensorflow::DT_INT32); in TEST()
62 auto flat = fallback_tensor.tensor().flat<int32_t>(); in TEST()
69 FallbackTensor fallback_tensor(&immutable_tensor); in TEST() local
70 EXPECT_TRUE(fallback_tensor.is_immutable()); in TEST()
72 ASSERT_EQ(fallback_tensor.tensor().NumElements(), 1); in TEST()
73 ASSERT_EQ(fallback_tensor.tensor().dtype(), tensorflow::DT_INT32); in TEST()
74 auto flat = fallback_tensor.tensor().flat<int32_t>(); in TEST()
H A DBUILD264 name = "fallback_tensor",
265 srcs = ["fallback_tensor.cc"],
266 hdrs = ["fallback_tensor.h"],
279 ":fallback_tensor",
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/ir/
H A Dtfrt_fallback_sync.td159 fallback_tensor: The output fallback tensor
167 TFTensorType:$fallback_tensor
170 let assemblyFormat = "operands attr-dict `:` type($fallback_tensor)";
181 fallback_tensor: The output fallback tensor
189 TFTensorType:$fallback_tensor
192 let assemblyFormat = "operands attr-dict `:` type($fallback_tensor)";
202 fallback_tensor: The input fallback tensor
207 TFTensorType:$fallback_tensor
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/gpu/
H A Dconversion_function.cc218 auto fallback_tensor = CopyRefGpuTensorToRuntimeFallbackTensor( in ConvertDenseGpuTensorToRuntimeFallbackTensor() local
220 if (fallback_tensor) { in ConvertDenseGpuTensorToRuntimeFallbackTensor()
222 host, std::move(*fallback_tensor)); in ConvertDenseGpuTensorToRuntimeFallbackTensor()
224 return EmitErrorAsync(exec_ctx, fallback_tensor.takeError()); in ConvertDenseGpuTensorToRuntimeFallbackTensor()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
H A Dkernel_fallback_execute_compat.cc523 auto& fallback_tensor = arg->get<tensorflow::tfrt_stub::FallbackTensor>(); in KernelFallbackExecuteOpInternal() local
529 if (!fallback_tensor.is_immutable() && !arg->IsUnique()) { in KernelFallbackExecuteOpInternal()
530 input_tf_tensors.push_back(fallback_tensor.tensor()); in KernelFallbackExecuteOpInternal()
532 input_tf_tensor_values[i].tensor = &fallback_tensor.tensor(); in KernelFallbackExecuteOpInternal()
903 const auto& fallback_tensor = arg.get(); in FallbackCopyTensorIfSmall() local
904 const auto& tensor = fallback_tensor.tensor(); in FallbackCopyTensorIfSmall()
906 if (!fallback_tensor.is_immutable()) { in FallbackCopyTensorIfSmall()
923 tensorflow::tfrt_stub::FallbackTensor>(fallback_tensor); in FallbackCopyTensorIfSmall()
H A DBUILD365 "//tensorflow/core/tfrt/utils:fallback_tensor",
403 "//tensorflow/core/tfrt/utils:fallback_tensor",
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_kernels.cc792 auto fallback_tensor = CreateRuntimeFallbackTensorFromTfTensorHandle( in RuntimeFallbackKernel() local
794 if (!fallback_tensor) { in RuntimeFallbackKernel()
796 host, tfrt::StrCat(fallback_tensor.takeError())); in RuntimeFallbackKernel()
800 host, std::move(*fallback_tensor)); in RuntimeFallbackKernel()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/
H A DBUILD157 "//tensorflow/core/tfrt/utils:fallback_tensor",
218 "//tensorflow/core/tfrt/utils:fallback_tensor",
314 "//tensorflow/core/tfrt/utils:fallback_tensor",
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/graph_executor/
H A DBUILD60 "//tensorflow/core/tfrt/utils:fallback_tensor",
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/saved_model/
H A DBUILD73 "//tensorflow/core/tfrt/utils:fallback_tensor",