/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
D | cpu_topk_test.cc | 35 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()
|
D | BUILD | 204 "//tensorflow/compiler/xla/client:xla_computation",
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | BUILD | 73 ":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/ |
D | BUILD | 192 "//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 …]
|
D | multiple_devices_on_host_test.cc | 67 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/ |
D | rng_bit_generator_expander.cc | 87 TF_ASSIGN_OR_RETURN(XlaComputation xla_computation, builder.Build()); in GetGeneratorComputation() 90 xla_computation.GetProgramShape()); in GetGeneratorComputation() 93 xla_computation.proto(), config)); in GetGeneratorComputation()
|
D | rng_expander.cc | 111 TF_ASSIGN_OR_RETURN(XlaComputation xla_computation, builder.Build()); in GetComputationForRng() 114 xla_computation.GetProgramShape()); in GetComputationForRng() 117 xla_computation.proto(), config)); in GetComputationForRng()
|
D | cholesky_expander.cc | 251 TF_ASSIGN_OR_RETURN(XlaComputation xla_computation, builder.Build()); in ExpandInstruction() 254 xla_computation.GetProgramShape()); in ExpandInstruction() 257 xla_computation.proto(), config)); in ExpandInstruction()
|
D | qr_expander.cc | 450 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()
|
D | triangular_solve_expander.cc | 600 TF_ASSIGN_OR_RETURN(XlaComputation xla_computation, builder.Build()); in ExpandInstruction() 603 xla_computation.GetProgramShape()); in ExpandInstruction() 606 xla_computation.proto(), config)); in ExpandInstruction()
|
D | BUILD | 1007 "//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/ |
D | BUILD | 66 "//tensorflow/compiler/xla/client:xla_computation", 85 "//tensorflow/compiler/xla/client:xla_computation",
|
/external/tensorflow/tensorflow/compiler/xrt/kernels/ |
D | BUILD | 68 "//tensorflow/compiler/xla/client:xla_computation", 125 "//tensorflow/compiler/xla/client:xla_computation",
|
/external/tensorflow/tensorflow/compiler/xla/tools/ |
D | BUILD | 74 "//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/ |
D | BUILD | 144 "//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/ |
D | BUILD | 143 "//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/ |
D | BUILD | 36 "//tensorflow/compiler/xla/client:xla_computation",
|
D | raw_api_test.cc | 1515 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/ |
D | BUILD | 35 "//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/ |
D | BUILD | 292 "//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/ |
D | compile_mlir_util.h | 72 xla::XlaComputation* xla_computation, bool use_tuple_args,
|
D | tf_xla_mlir_translate.cc | 283 xla::XlaComputation* xla_computation = compilation_result->computation.get(); in CompileMlirToXlaHloViaBuilder() local 284 *xla_computation = xla::XlaComputation(hlo_proto.hlo_module()); in CompileMlirToXlaHloViaBuilder()
|
D | compile_mlir_util.cc | 390 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/ |
D | BUILD | 175 "//tensorflow/compiler/xla/client:xla_computation", 302 "//tensorflow/compiler/xla/client:xla_computation",
|
/external/tensorflow/tensorflow/compiler/aot/ |
D | BUILD | 64 "//tensorflow/compiler/xla/client:xla_computation",
|