Home
last modified time | relevance | path

Searched defs:xla_computation (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_embedding_ops.cc91 auto xla_computation = in Compile() local
164 auto xla_computation = in Compile() local
277 auto xla_computation = in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dmlir_to_hlo.cc35 XlaComputation& xla_computation, in MlirToXlaComputation()
93 absl::string_view mlir_module_str, XlaComputation& xla_computation, in ParseMlirModuleStringAndConvertToXlaComputation()
H A Dtfrt_cpu_pjrt_client_test.cc48 XlaComputation xla_computation(hlo_module->ToProto()); in TEST() local
H A Dtfrt_cpu_pjrt_client.cc423 XlaComputation xla_computation; in Compile() local
H A Dpjrt_stream_executor_client.cc2488 XlaComputation xla_computation; in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/
H A Dtpu_ops_c_api.h674 TpuSerializedProto* xla_computation; member
688 TpuSerializedProto* xla_computation; member
707 TpuSerializedProto* xla_computation; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dxla_call_module_op.cc102 xla::XlaComputation xla_computation; in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dtf_xla_mlir_translate.cc297 xla::XlaComputation* xla_computation = compilation_result->computation.get(); in CompileMlirToXlaHloViaBuilder() local
H A Dcompile_mlir_util.cc518 xla::XlaComputation* xla_computation, bool use_tuple_args, in ConvertMLIRToXlaComputation()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
H A Dtpu_client.cc883 XlaComputation xla_computation; in CompileMlir() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter_unnested.cc2456 xla::XlaComputation xla_computation; in GetOrCreateSubComputationFromRegion() local