| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| all_reduce_test.cc | 49 EXPECT_EQ(literal, ExecuteAndTransfer(std::move(module), {&literal}));
72 ExecuteAndTransfer(std::move(module), {&literal0, &literal1}));
98 ExecuteAndTransfer(std::move(module), {&literal0}));
|
| broadcast_test.cc | 47 auto result = ExecuteAndTransfer(std::move(hlo_module), {});
63 auto result = ExecuteAndTransfer(std::move(hlo_module), {});
86 auto result = ExecuteAndTransfer(std::move(hlo_module), {});
107 auto result = ExecuteAndTransfer(std::move(hlo_module), {});
126 auto result = ExecuteAndTransfer(std::move(hlo_module), {});
143 auto result = ExecuteAndTransfer(std::move(hlo_module), {});
163 auto result = ExecuteAndTransfer(std::move(hlo_module), {});
188 auto result = ExecuteAndTransfer(std::move(hlo_module), {});
219 auto result = ExecuteAndTransfer(std::move(hlo_module), {});
236 auto result = ExecuteAndTransfer(std::move(hlo_module), {})
[all...] |
| prng_test.cc | 60 ExecuteAndTransfer(&builder, /*arguments=*/{}).ConsumeValueOrDie();
151 ExecuteAndTransfer(&builder, /*arguments=*/{}).ConsumeValueOrDie();
210 auto actual, client_->ExecuteAndTransfer(
223 // * If a seed is passed in via Execute (ExecuteAndTransfer) then the output is
245 result1, client_->ExecuteAndTransfer(computation, /*arguments=*/{},
253 result2, client_->ExecuteAndTransfer(computation, /*arguments=*/{},
256 result3, client_->ExecuteAndTransfer(computation, /*arguments=*/{},
266 result4, client_->ExecuteAndTransfer(computation, /*arguments=*/{},
269 result5, client_->ExecuteAndTransfer(computation, /*arguments=*/{},
272 result6, client_->ExecuteAndTransfer(computation, /*arguments=*/{}
[all...] |
| custom_call_test.cc | 92 Literal result = ExecuteAndTransfer(std::move(module), {});
113 Literal result = ExecuteAndTransfer(std::move(module), {});
136 Literal result = ExecuteAndTransfer(std::move(module), {});
159 Literal result = ExecuteAndTransfer(std::move(module), {&argument});
186 Literal result = ExecuteAndTransfer(std::move(module), {&argument});
206 Literal result = ExecuteAndTransfer(std::move(module), {&arg0, &arg1});
|
| copy_test.cc | 51 Literal result = ExecuteAndTransfer(std::move(module), {});
104 Literal result = ExecuteAndTransfer(std::move(module), {&literal});
124 Literal result = ExecuteAndTransfer(std::move(module), {});
150 Literal result = ExecuteAndTransfer(std::move(module), {});
183 Literal result = ExecuteAndTransfer(std::move(module), {});
217 Literal result = ExecuteAndTransfer(std::move(module), {});
253 auto actual = ExecuteAndTransfer(&builder, {input_data.get()}, &out_shape)
|
| fusion_test.cc | 131 auto actual = ExecuteAndTransfer(std::move(hlo_module), {});
242 ExecuteAndTransfer(std::move(hlo_module), {}), ErrorSpec(1e-4)));
268 ExecuteAndTransfer(std::move(hlo_module), {}), ErrorSpec(1e-4)));
298 auto result = ExecuteAndTransfer(std::move(hlo_module), {});
328 ExecuteAndTransfer(std::move(hlo_module), {}), ErrorSpec(1e-4)));
343 ExecuteAndTransfer(std::move(hlo_module), {})));
358 ExecuteAndTransfer(std::move(hlo_module), {})));
373 ExecuteAndTransfer(std::move(hlo_module), {})));
388 ExecuteAndTransfer(std::move(hlo_module), {})));
403 ExecuteAndTransfer(std::move(hlo_module), {})))
[all...] |
| replay_test.cc | 64 ->ExecuteAndTransfer(replayed, /*arguments=*/{}, &execution_options_)
103 ->ExecuteAndTransfer(replayed,
144 ->ExecuteAndTransfer(replayed, /*arguments=*/{}, &execution_options_)
|
| client_library_test_base.cc | 112 StatusOr<Literal> ClientLibraryTestBase::ExecuteAndTransfer(
120 return client_->ExecuteAndTransfer(computation, arguments,
124 StatusOr<Literal> ClientLibraryTestBase::ExecuteAndTransfer(
129 return ExecuteAndTransfer(computation, arguments, shape_with_output_layout);
141 return ref_client_->ExecuteAndTransfer(computation, arguments,
154 client_->ExecuteAndTransfer(computation, arguments, &execution_options_);
191 TF_ASSIGN_OR_RETURN(auto actual, ExecuteAndTransfer(computation, arguments));
202 ExecuteAndTransfer(computation, arguments, &layout));
259 ExecuteAndTransfer(computation,
293 return ExecuteAndTransfer(computation, arguments, shape_with_layout)
[all...] |
| outfeed_in_nested_computation_test.cc | 77 comp_result = local_client_->ExecuteAndTransfer(computation, {})
152 comp_result = local_client_->ExecuteAndTransfer(computation, {})
|
| client_test.cc | 93 client_->ExecuteAndTransfer(computation, {}, &execution_options));
|
| compilation_cache_test.cc | 47 ->ExecuteAndTransfer(computation, arguments,
|
| multioutput_fusion_test.cc | 119 ExecuteAndTransfer(std::move(hlo_module), {&literal_r0, &arg1});
182 auto actual = ExecuteAndTransfer(std::move(hlo_module), {&input0, &input1});
|
| constants_test.cc | 168 Literal result = ExecuteAndTransfer(&builder, {}).ConsumeValueOrDie();
|
| hlo_test_base.h | 173 Literal ExecuteAndTransfer(std::unique_ptr<HloModule> module,
|
| scatter_test.cc | 162 auto actual = ExecuteAndTransfer(std::move(module), {&permutation});
|
| client_library_test_base.h | 98 StatusOr<Literal> ExecuteAndTransfer(
102 StatusOr<Literal> ExecuteAndTransfer(
|
| reshape_test.cc | 625 ->ExecuteAndTransfer(computation, {input.get()}, &execution_options)
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| sample_harness.cc | 61 xla::StatusOr<xla::Literal> result = client->ExecuteAndTransfer(
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| map_inliner_test.cc | 72 auto result = ExecuteAndTransfer(hlo_module->Clone(), {});
106 auto result = ExecuteAndTransfer(hlo_module->Clone(), {});
144 auto result = ExecuteAndTransfer(hlo_module->Clone(), {});
175 auto result = ExecuteAndTransfer(hlo_module->Clone(), {});
|
| hlo_cse_test.cc | 74 auto result = ExecuteAndTransfer(module->Clone(), {});
106 auto result = ExecuteAndTransfer(module->Clone(), {});
136 auto result = ExecuteAndTransfer(module->Clone(), {});
|
| /external/tensorflow/tensorflow/compiler/xla/rpc/ |
| grpc_client_test.cc | 100 TF_ASSERT_OK_AND_ASSIGN(auto result_literal, client_->ExecuteAndTransfer(
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| cpu_fusion_test.cc | 85 auto result = ExecuteAndTransfer(module->Clone(), {});
132 auto result = ExecuteAndTransfer(module->Clone(), {});
218 auto result = ExecuteAndTransfer(module->Clone(), {});
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| tf2xla_test.cc | 91 client->ExecuteAndTransfer(computation, {x_global.get(), y_global.get()});
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| client.h | 170 StatusOr<Literal> ExecuteAndTransfer(
|
| client.cc | 166 StatusOr<Literal> Client::ExecuteAndTransfer(
|
