Home
last modified time | relevance | path

Searched refs:xla_computation (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_topk_test.cc35 TF_ASSERT_OK_AND_ASSIGN(XlaComputation xla_computation, builder.Build()); in TEST_F()
38 xla_computation.GetProgramShape()); in TEST_F()
41 auto module, HloModule::CreateFromProto(xla_computation.proto(), config)); in TEST_F()
62 TF_ASSERT_OK_AND_ASSIGN(XlaComputation xla_computation, builder.Build()); in TEST_F()
65 xla_computation.GetProgramShape()); in TEST_F()
68 auto module, HloModule::CreateFromProto(xla_computation.proto(), config)); in TEST_F()
DBUILD204 "//tensorflow/compiler/xla/client:xla_computation",
/external/tensorflow/tensorflow/compiler/xla/client/
DBUILD73 ":xla_computation",
118 ":xla_computation",
148 ":xla_computation",
200 name = "xla_computation",
201 srcs = ["xla_computation.cc"],
202 hdrs = ["xla_computation.h"],
222 ":xla_computation",
254 ":xla_computation",
/external/tensorflow/tensorflow/compiler/xla/tests/
DBUILD192 "//tensorflow/compiler/xla/client:xla_computation",
230 "//tensorflow/compiler/xla/client:xla_computation",
305 "//tensorflow/compiler/xla/client:xla_computation",
335 "//tensorflow/compiler/xla/client:xla_computation",
376 "//tensorflow/compiler/xla/client:xla_computation",
396 "//tensorflow/compiler/xla/client:xla_computation",
424 "//tensorflow/compiler/xla/client:xla_computation",
446 "//tensorflow/compiler/xla/client:xla_computation",
463 "//tensorflow/compiler/xla/client:xla_computation",
486 "//tensorflow/compiler/xla/client:xla_computation",
[all …]
Dmultiple_devices_on_host_test.cc67 TF_ASSERT_OK_AND_ASSIGN(XlaComputation xla_computation, BuildComputation()); in TestWithDeviceCount()
70 client->Compile(xla_computation, {}, xla::ExecutableBuildOptions{})); in TestWithDeviceCount()
/external/tensorflow/tensorflow/compiler/xla/service/
Drng_bit_generator_expander.cc87 TF_ASSIGN_OR_RETURN(XlaComputation xla_computation, builder.Build()); in GetGeneratorComputation()
90 xla_computation.GetProgramShape()); in GetGeneratorComputation()
93 xla_computation.proto(), config)); in GetGeneratorComputation()
Drng_expander.cc111 TF_ASSIGN_OR_RETURN(XlaComputation xla_computation, builder.Build()); in GetComputationForRng()
114 xla_computation.GetProgramShape()); in GetComputationForRng()
117 xla_computation.proto(), config)); in GetComputationForRng()
Dcholesky_expander.cc251 TF_ASSIGN_OR_RETURN(XlaComputation xla_computation, builder.Build()); in ExpandInstruction()
254 xla_computation.GetProgramShape()); in ExpandInstruction()
257 xla_computation.proto(), config)); in ExpandInstruction()
Dqr_expander.cc450 TF_ASSIGN_OR_RETURN(XlaComputation xla_computation, builder.Build(l)); in ExpandInstruction()
453 xla_computation.GetProgramShape()); in ExpandInstruction()
456 xla_computation.proto(), config)); in ExpandInstruction()
Dtriangular_solve_expander.cc600 TF_ASSIGN_OR_RETURN(XlaComputation xla_computation, builder.Build()); in ExpandInstruction()
603 xla_computation.GetProgramShape()); in ExpandInstruction()
606 xla_computation.proto(), config)); in ExpandInstruction()
DBUILD1007 "//tensorflow/compiler/xla/client:xla_computation",
1726 "//tensorflow/compiler/xla/client:xla_computation",
1895 "//tensorflow/compiler/xla/client:xla_computation",
3038 "//tensorflow/compiler/xla/client:xla_computation",
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
DBUILD66 "//tensorflow/compiler/xla/client:xla_computation",
85 "//tensorflow/compiler/xla/client:xla_computation",
/external/tensorflow/tensorflow/compiler/xrt/kernels/
DBUILD68 "//tensorflow/compiler/xla/client:xla_computation",
125 "//tensorflow/compiler/xla/client:xla_computation",
/external/tensorflow/tensorflow/compiler/xla/tools/
DBUILD74 "//tensorflow/compiler/xla/client:xla_computation",
159 "//tensorflow/compiler/xla/client:xla_computation",
178 "//tensorflow/compiler/xla/client:xla_computation",
/external/tensorflow/tensorflow/compiler/xla/pjrt/
DBUILD144 "//tensorflow/compiler/xla/client:xla_computation",
165 "//tensorflow/compiler/xla/client:xla_computation",
195 "//tensorflow/compiler/xla/client:xla_computation",
/external/tensorflow/tensorflow/compiler/tf2xla/
DBUILD143 "//tensorflow/compiler/xla/client:xla_computation",
174 "//tensorflow/compiler/xla/client:xla_computation",
301 "//tensorflow/compiler/xla/client:xla_computation",
372 "//tensorflow/compiler/xla/client:xla_computation",
441 "//tensorflow/compiler/xla/client:xla_computation",
535 "//tensorflow/compiler/xla/client:xla_computation",
697 "//tensorflow/compiler/xla/client:xla_computation",
/external/tensorflow/tensorflow/compiler/xrt/tests/
DBUILD36 "//tensorflow/compiler/xla/client:xla_computation",
Draw_api_test.cc1515 TF_ASSERT_OK_AND_ASSIGN(xla::XlaComputation xla_computation, builder.Build()); in TEST()
1517 auto result_shape = xla_computation.GetProgramShape().ValueOrDie().result(); in TEST()
1527 StoreComputationSnapshot(xla_computation, c.mutable_hlo_snapshot()); in TEST()
1553 XlaCompiledProgramShape(xla_computation, xla::ProgramShape(*shapes)); in TEST()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
DBUILD35 "//tensorflow/compiler/xla/client:xla_computation",
70 "//tensorflow/compiler/xla/client:xla_computation",
144 "//tensorflow/compiler/xla/client:xla_computation",
435 "//tensorflow/compiler/xla/client:xla_computation",
/external/tensorflow/tensorflow/compiler/xla/python/
DBUILD292 "//tensorflow/compiler/xla/client:xla_computation",
316 "//tensorflow/compiler/xla/client:xla_computation",
476 "//tensorflow/compiler/xla/client:xla_computation",
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.h72 xla::XlaComputation* xla_computation, bool use_tuple_args,
Dtf_xla_mlir_translate.cc283 xla::XlaComputation* xla_computation = compilation_result->computation.get(); in CompileMlirToXlaHloViaBuilder() local
284 *xla_computation = xla::XlaComputation(hlo_proto.hlo_module()); in CompileMlirToXlaHloViaBuilder()
Dcompile_mlir_util.cc390 xla::XlaComputation* xla_computation, bool use_tuple_args, in ConvertMLIRToXlaComputation() argument
402 *xla_computation = xla::XlaComputation(hlo_proto.hlo_module()); in ConvertMLIRToXlaComputation()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
DBUILD175 "//tensorflow/compiler/xla/client:xla_computation",
302 "//tensorflow/compiler/xla/client:xla_computation",
/external/tensorflow/tensorflow/compiler/aot/
DBUILD64 "//tensorflow/compiler/xla/client:xla_computation",

12