Home
last modified time | relevance | path

Searched refs:computation_status (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dmap_test.cc58 auto computation_status = mapped_builder.Build(); in CreateAdderToOne() local
59 TF_CHECK_OK(computation_status.status()); in CreateAdderToOne()
60 return computation_status.ConsumeValueOrDie(); in CreateAdderToOne()
68 auto computation_status = b.Build(); in CreateMax() local
69 TF_CHECK_OK(computation_status.status()); in CreateMax()
70 return computation_status.ConsumeValueOrDie(); in CreateMax()
80 auto computation_status = mapped_builder.Build(); in CreateScalarOne() local
81 TF_CHECK_OK(computation_status.status()); in CreateScalarOne()
82 return computation_status.ConsumeValueOrDie(); in CreateScalarOne()
95 auto computation_status = mapped_builder.Build(); in CreateMulByTwo() local
[all …]
Dcheck_execution_arity_test.cc50 auto computation_status = builder.Build(); in TEST_F() local
51 ASSERT_IS_OK(computation_status.status()); in TEST_F()
52 auto computation = computation_status.ConsumeValueOrDie(); in TEST_F()
84 auto computation_status = builder.Build(); in XLA_TEST_F() local
85 ASSERT_IS_OK(computation_status.status()); in XLA_TEST_F()
86 auto computation = computation_status.ConsumeValueOrDie(); in XLA_TEST_F()
Dvector_ops_simple_test.cc331 auto computation_status = builder.Build(); in XLA_TEST_F() local
332 ASSERT_IS_OK(computation_status.status()); in XLA_TEST_F()
333 add_half = computation_status.ConsumeValueOrDie(); in XLA_TEST_F()
344 auto computation_status = builder.Build(); in XLA_TEST_F() local
345 ASSERT_IS_OK(computation_status.status()); in XLA_TEST_F()
346 clamp = computation_status.ConsumeValueOrDie(); in XLA_TEST_F()
361 auto computation_status = builder.Build(); in XLA_TEST_F() local
362 ASSERT_IS_OK(computation_status.status()); in XLA_TEST_F()
363 mult_relu_add = computation_status.ConsumeValueOrDie(); in XLA_TEST_F()
Dtest_utils_test.cc39 auto computation_status = builder.Build(); in XLA_TEST_F() local
40 TF_ASSERT_OK(computation_status.status()); in XLA_TEST_F()
46 computation_status.ValueOrDie(), {0}); in XLA_TEST_F()
47 computation_status = builder.Build(); in XLA_TEST_F()
48 TF_ASSERT_OK(computation_status.status()); in XLA_TEST_F()
51 computation_status.ValueOrDie(), {&pair_float, &single_float}, in XLA_TEST_F()
Dclient_library_test_base.cc147 auto computation_status = builder->Build(); in ExecuteToString() local
148 if (!computation_status.ok()) { in ExecuteToString()
149 return computation_status.status().ToString(); in ExecuteToString()
151 auto computation = computation_status.ConsumeValueOrDie(); in ExecuteToString()
536 auto computation_status = builder.Build(); in CreateScalarRelu() local
537 TF_CHECK_OK(computation_status.status()); in CreateScalarRelu()
538 return computation_status.ConsumeValueOrDie(); in CreateScalarRelu()
547 auto computation_status = builder.Build(); in CreateScalarMax() local
548 TF_CHECK_OK(computation_status.status()); in CreateScalarMax()
549 return computation_status.ConsumeValueOrDie(); in CreateScalarMax()
[all …]
Dconcat_test.cc43 StatusOr<XlaComputation> computation_status = builder.Build(); in XLA_TEST_F() local
44 ASSERT_FALSE(computation_status.ok()); in XLA_TEST_F()
45 EXPECT_THAT(computation_status.status().ToString(), in XLA_TEST_F()
75 StatusOr<XlaComputation> computation_status = builder.Build(); in XLA_TEST_F() local
76 ASSERT_FALSE(computation_status.ok()); in XLA_TEST_F()
77 EXPECT_THAT(computation_status.status().ToString(), in XLA_TEST_F()
416 StatusOr<XlaComputation> computation_status = builder.Build(); in XLA_TEST_F() local
417 ASSERT_FALSE(computation_status.ok()); in XLA_TEST_F()
419 computation_status.status().ToString(), in XLA_TEST_F()
431 StatusOr<XlaComputation> computation_status = builder.Build(); in XLA_TEST_F() local
[all …]
Dmatrix_ops_simple_test.cc83 auto computation_status = builder.Build(); in XLA_TYPED_TEST() local
84 ASSERT_IS_OK(computation_status.status()); in XLA_TYPED_TEST()
85 add_half = computation_status.ConsumeValueOrDie(); in XLA_TYPED_TEST()
Dtuple_test.cc311 auto computation_status = b.Build(); in XLA_TEST_F() local
312 ASSERT_IS_OK(computation_status.status()); in XLA_TEST_F()
313 tuple_computation = computation_status.ConsumeValueOrDie(); in XLA_TEST_F()
Dparams_test.cc161 auto computation_status = builder.Build(); in XLA_TEST_F() local
163 ASSERT_NE(computation_status.status(), Status::OK()); in XLA_TEST_F()
Dreduce_test.cc936 auto computation_status = builder.Build(); in XLA_TEST_F() local
937 TF_ASSERT_OK(computation_status.status()); in XLA_TEST_F()
942 computation_status.ValueOrDie(), {0}); in XLA_TEST_F()
Darray_elementwise_ops_test.cc2920 auto computation_status = builder.Build(); in XLA_TEST_F() local
2921 ASSERT_FALSE(computation_status.ok()); in XLA_TEST_F()
2922 EXPECT_THAT(computation_status.status().ToString(), in XLA_TEST_F()
2948 auto computation_status = builder.Build(); in XLA_TEST_F() local
2949 ASSERT_FALSE(computation_status.ok()); in XLA_TEST_F()
2950 EXPECT_THAT(computation_status.status().error_message(), in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis_test.cc66 auto computation_status = builder.Build(); in HloCostAnalysisTest() local
67 TF_CHECK_OK(computation_status.status()); in HloCostAnalysisTest()
68 add_and_exp_ = computation_status.ConsumeValueOrDie(); in HloCostAnalysisTest()
77 auto computation_status = builder.Build(); in HloCostAnalysisTest() local
78 TF_CHECK_OK(computation_status.status()); in HloCostAnalysisTest()
79 add_ = computation_status.ConsumeValueOrDie(); in HloCostAnalysisTest()
88 auto computation_status = builder.Build(); in HloCostAnalysisTest() local
89 TF_CHECK_OK(computation_status.status()); in HloCostAnalysisTest()
90 sigmoid_ = computation_status.ConsumeValueOrDie(); in HloCostAnalysisTest()
99 auto computation_status = builder.Build(); in HloCostAnalysisTest() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/tools/
Ddumped_computation_to_operation_list.cc69 auto computation_status = client->LoadSnapshot(snapshot); in RealMain() local
70 if (!computation_status.ok()) { in RealMain()
72 computation_status.status().ToString().c_str()); in RealMain()
75 XlaComputation computation = computation_status.ConsumeValueOrDie(); in RealMain()
Ddumped_computation_to_text.cc44 auto computation_status = client->LoadSnapshot(snapshot); in RealMain() local
45 if (!computation_status.ok()) { in RealMain()
47 computation_status.status().ToString().c_str()); in RealMain()
50 XlaComputation computation = computation_status.ConsumeValueOrDie(); in RealMain()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dsample_harness.cc56 xla::StatusOr<xla::XlaComputation> computation_status = builder.Build(); in main() local
57 xla::XlaComputation computation = computation_status.ConsumeValueOrDie(); in main()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.cc317 xla::StatusOr<xla::XlaComputation> computation_status = builder->Build(); in BuildComputation() local
318 if (!computation_status.ok()) { in BuildComputation()
319 return computation_status.status(); in BuildComputation()
321 *computation = computation_status.ConsumeValueOrDie(); in BuildComputation()