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()
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 auto executables, local_client_->Compile(computation_status.ValueOrDie(), in XLA_TEST_F()
Dvector_ops_simple_test.cc343 auto computation_status = builder.Build(); in XLA_TEST_F() local
344 ASSERT_IS_OK(computation_status.status()); in XLA_TEST_F()
345 add_half = computation_status.ConsumeValueOrDie(); in XLA_TEST_F()
356 auto computation_status = builder.Build(); in XLA_TEST_F() local
357 ASSERT_IS_OK(computation_status.status()); in XLA_TEST_F()
358 clamp = computation_status.ConsumeValueOrDie(); in XLA_TEST_F()
373 auto computation_status = builder.Build(); in XLA_TEST_F() local
374 ASSERT_IS_OK(computation_status.status()); in XLA_TEST_F()
375 mult_relu_add = computation_status.ConsumeValueOrDie(); 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.cc46 StatusOr<XlaComputation> computation_status = builder.Build(); in XLA_TEST_F() local
47 ASSERT_FALSE(computation_status.ok()); in XLA_TEST_F()
48 EXPECT_THAT(computation_status.status().ToString(), in XLA_TEST_F()
78 StatusOr<XlaComputation> computation_status = builder.Build(); in XLA_TEST_F() local
79 ASSERT_FALSE(computation_status.ok()); in XLA_TEST_F()
80 EXPECT_THAT(computation_status.status().ToString(), in XLA_TEST_F()
419 StatusOr<XlaComputation> computation_status = builder.Build(); in XLA_TEST_F() local
420 ASSERT_FALSE(computation_status.ok()); in XLA_TEST_F()
422 computation_status.status().ToString(), in XLA_TEST_F()
434 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.cc937 auto computation_status = builder.Build(); in XLA_TEST_F() local
938 TF_ASSERT_OK(computation_status.status()); in XLA_TEST_F()
943 computation_status.ValueOrDie(), {0}); in XLA_TEST_F()
Darray_elementwise_ops_test.cc3026 auto computation_status = builder.Build(); in XLA_TEST_F() local
3027 ASSERT_FALSE(computation_status.ok()); in XLA_TEST_F()
3028 EXPECT_THAT(computation_status.status().ToString(), in XLA_TEST_F()
3054 auto computation_status = builder.Build(); in XLA_TEST_F() local
3055 ASSERT_FALSE(computation_status.ok()); in XLA_TEST_F()
3056 EXPECT_THAT(computation_status.status().error_message(), in XLA_TEST_F()
/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/
Dhlo_cost_analysis_test.cc65 auto computation_status = builder.Build(); in HloCostAnalysisTest() local
66 TF_CHECK_OK(computation_status.status()); in HloCostAnalysisTest()
67 add_and_exp_ = computation_status.ConsumeValueOrDie(); in HloCostAnalysisTest()
76 auto computation_status = builder.Build(); in HloCostAnalysisTest() local
77 TF_CHECK_OK(computation_status.status()); in HloCostAnalysisTest()
78 add_ = computation_status.ConsumeValueOrDie(); in HloCostAnalysisTest()
87 auto computation_status = builder.Build(); in HloCostAnalysisTest() local
88 TF_CHECK_OK(computation_status.status()); in HloCostAnalysisTest()
89 sigmoid_ = computation_status.ConsumeValueOrDie(); in HloCostAnalysisTest()
98 auto computation_status = builder.Build(); in HloCostAnalysisTest() local
[all …]
/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.cc419 xla::StatusOr<xla::XlaComputation> computation_status = builder->Build(); in BuildComputation() local
420 if (!computation_status.ok()) { in BuildComputation()
421 return computation_status.status(); in BuildComputation()
423 *computation = computation_status.ConsumeValueOrDie(); in BuildComputation()