Home
last modified time | relevance | path

Searched refs:param0_data (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dmap_test.cc173 std::unique_ptr<GlobalData> param0_data = in TEST_F() local
179 ComputeAndCompareR0<float>(&builder, 43.0, {param0_data.get()}, in TEST_F()
187 std::unique_ptr<GlobalData> param0_data = in XLA_TEST_F() local
193 ComputeAndCompareR1<float>(&builder, {}, {param0_data.get()}, in XLA_TEST_F()
202 std::unique_ptr<GlobalData> param0_data = in TEST_F() local
209 {param0_data.get()}, ErrorSpec(0.01f)); in TEST_F()
216 std::unique_ptr<GlobalData> param0_data = in TEST_F() local
222 ComputeAndCompareR1<int32>(&builder, {1, 1, 1, 1}, {param0_data.get()}); in TEST_F()
229 std::unique_ptr<GlobalData> param0_data = in TEST_F() local
235 ComputeAndCompareR1<uint32>(&builder, {1, 1, 1, 1}, {param0_data.get()}); in TEST_F()
[all …]
Dparams_test.cc46 std::unique_ptr<GlobalData> param0_data = in XLA_TEST_F() local
51 ComputeAndCompareR0<float>(&builder, 3.14159f, {param0_data.get()}, in XLA_TEST_F()
58 std::unique_ptr<GlobalData> param0_data = in XLA_TEST_F() local
63 ComputeAndCompareR1<float>(&builder, {}, {param0_data.get()}, in XLA_TEST_F()
70 std::unique_ptr<GlobalData> param0_data = in XLA_TEST_F() local
75 ComputeAndCompareR1<float>(&builder, {3.14f, -100.25f}, {param0_data.get()}, in XLA_TEST_F()
83 std::unique_ptr<GlobalData> param0_data = in XLA_TEST_F() local
90 ComputeAndCompareR1U8(&builder, str, {param0_data.get()}); in XLA_TEST_F()
97 std::unique_ptr<GlobalData> param0_data = in XLA_TEST_F() local
103 {param0_data.get()}, ErrorSpec(0.01f)); in XLA_TEST_F()
[all …]
Dfmax_fmin_test.cc47 std::unique_ptr<GlobalData> param0_data = CreateR1Parameter<float>( in TEST_F() local
61 {param0_data.get(), param1_data.get()}, in TEST_F()
69 std::unique_ptr<GlobalData> param0_data = CreateR1Parameter<float>( in TEST_F() local
83 {param0_data.get(), param1_data.get()}, in TEST_F()
Dcheck_execution_arity_test.cc45 auto param0_data = in TEST_F() local
57 computation, {param0_data.get(), param1_data.get()}, &execution_options_); in TEST_F()
61 client_->Execute(computation, {param0_data.get()}, &execution_options_); in TEST_F()
Dselect_test.cc142 std::unique_ptr<GlobalData> param0_data = CreateR1Parameter<float>( in TEST_F() local
152 {param0_data.get(), param1_data.get()}, in TEST_F()
183 std::unique_ptr<GlobalData> param0_data = in TEST_F() local
193 {param0_data.get(), param1_data.get()}, in TEST_F()
Ddeconstruct_tuple_test.cc174 std::unique_ptr<GlobalData> param0_data = in XLA_TEST_F() local
178 auto global_data = ExecuteAndCheckTransfer(&builder, {param0_data.get()}); in XLA_TEST_F()
183 EXPECT_NE(handles[0]->handle().handle(), param0_data->handle().handle()); in XLA_TEST_F()
Dvector_ops_simple_test.cc183 std::unique_ptr<GlobalData> param0_data = CreateR1Parameter<float>( in XLA_TEST_F() local
192 {param0_data.get(), param1_data.get()}, in XLA_TEST_F()
223 std::unique_ptr<GlobalData> param0_data = in XLA_TEST_F() local
232 {param0_data.get(), param1_data.get()}, in XLA_TEST_F()
Dprng_test.cc198 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<GlobalData> param0_data, in XLA_TEST_F()
212 /*arguments=*/{param0_data.get()}, &execution_options)); in XLA_TEST_F()
Darray_elementwise_ops_test.cc2169 std::unique_ptr<GlobalData> param0_data = in XLA_TEST_F() local
2182 {param0_data.get(), param1_data.get()}, in XLA_TEST_F()
2191 std::unique_ptr<GlobalData> param0_data = in XLA_TEST_F() local
2205 &builder, expected, {param0_data.get(), param1_data.get()}, error_spec_); in XLA_TEST_F()
2213 std::unique_ptr<GlobalData> param0_data = in XLA_TEST_F() local
2221 {param0_data.get()}, error_spec_); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dsample_harness.cc42 std::unique_ptr<xla::GlobalData> param0_data = in main() local
63 /*arguments=*/{param0_data.get(), param1_data.get()}, in main()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler_test.cc216 std::unique_ptr<xla::GlobalData> param0_data = in TEST_F() local
223 ->Execute(*result.computation, {param0_data.get(), param1_data.get()}) in TEST_F()
267 std::unique_ptr<xla::GlobalData> param0_data = in TEST_F() local
274 ->Execute(*result.computation, {param0_data.get(), param1_data.get()}) in TEST_F()
503 std::unique_ptr<xla::GlobalData> param0_data = in TEST_F() local
507 client_->Execute(*result.computation, {param0_data.get()}) in TEST_F()
535 std::unique_ptr<xla::GlobalData> param0_data = in TEST_F() local
539 client_->Execute(*result.computation, {param0_data.get()}) in TEST_F()
774 std::unique_ptr<xla::GlobalData> param0_data = in TEST_F() local
778 client_->Execute(*result.computation, {param0_data.get()}) in TEST_F()
[all …]