Home
last modified time | relevance | path

Searched refs:computation_status (Results 1 – 13 of 13) 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()
Dclient_library_test_base.cc122 StatusOr<Computation> computation_status = builder->Build(); in ExecuteToString() local
123 if (!computation_status.ok()) { in ExecuteToString()
124 return computation_status.status().ToString(); in ExecuteToString()
126 Computation computation = computation_status.ConsumeValueOrDie(); in ExecuteToString()
467 auto computation_status = builder.Build(); in CreateScalarRelu() local
468 TF_CHECK_OK(computation_status.status()); in CreateScalarRelu()
469 return computation_status.ConsumeValueOrDie(); in CreateScalarRelu()
478 auto computation_status = builder.Build(); in CreateScalarMax() local
479 TF_CHECK_OK(computation_status.status()); in CreateScalarMax()
480 return computation_status.ConsumeValueOrDie(); in CreateScalarMax()
[all …]
Dvector_ops_simple_test.cc360 auto computation_status = builder.Build(); in XLA_TEST_F() local
361 ASSERT_IS_OK(computation_status.status()); in XLA_TEST_F()
362 add_half = 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 clamp = computation_status.ConsumeValueOrDie(); in XLA_TEST_F()
390 auto computation_status = builder.Build(); in XLA_TEST_F() local
391 ASSERT_IS_OK(computation_status.status()); in XLA_TEST_F()
392 mult_relu_add = computation_status.ConsumeValueOrDie(); in XLA_TEST_F()
Dmatrix_ops_simple_test.cc51 auto computation_status = builder.Build(); in BuildSum() local
52 TF_CHECK_OK(computation_status.status()); in BuildSum()
53 return computation_status.ConsumeValueOrDie(); in BuildSum()
104 auto computation_status = builder.Build(); in TEST_F() local
105 ASSERT_IS_OK(computation_status.status()); in TEST_F()
106 add_half = computation_status.ConsumeValueOrDie(); in TEST_F()
Dconcat_test.cc43 StatusOr<Computation> 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<Computation> 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()
413 StatusOr<Computation> computation_status = builder.Build(); in XLA_TEST_F() local
414 ASSERT_FALSE(computation_status.ok()); in XLA_TEST_F()
416 computation_status.status().ToString(), in XLA_TEST_F()
Dtuple_test.cc303 auto computation_status = b.Build(); in XLA_TEST_F() local
304 ASSERT_IS_OK(computation_status.status()); in XLA_TEST_F()
305 tuple_computation = computation_status.ConsumeValueOrDie(); in XLA_TEST_F()
Darray_elementwise_ops_test.cc2682 StatusOr<Computation> computation_status = builder.Build(); in XLA_TEST_F() local
2683 ASSERT_FALSE(computation_status.ok()); in XLA_TEST_F()
2684 EXPECT_THAT(computation_status.status().ToString(), in XLA_TEST_F()
2710 StatusOr<Computation> computation_status = builder.Build(); in XLA_TEST_F() local
2711 ASSERT_FALSE(computation_status.ok()); in XLA_TEST_F()
2712 EXPECT_THAT(computation_status.status().error_message(), in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis_test.cc69 auto computation_status = builder.Build(); in HloCostAnalysisTest() local
70 TF_CHECK_OK(computation_status.status()); in HloCostAnalysisTest()
71 add_and_exp_ = computation_status.ConsumeValueOrDie(); in HloCostAnalysisTest()
80 auto computation_status = builder.Build(); in HloCostAnalysisTest() local
81 TF_CHECK_OK(computation_status.status()); in HloCostAnalysisTest()
82 add_ = computation_status.ConsumeValueOrDie(); in HloCostAnalysisTest()
91 auto computation_status = builder.Build(); in HloCostAnalysisTest() local
92 TF_CHECK_OK(computation_status.status()); in HloCostAnalysisTest()
93 sigmoid_ = computation_status.ConsumeValueOrDie(); in HloCostAnalysisTest()
102 auto computation_status = builder.Build(); in HloCostAnalysisTest() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/tools/
Ddumped_computation_to_operation_list.cc72 auto computation_status = client->LoadSnapshot(session_module); in RealMain() local
73 if (!computation_status.ok()) { in RealMain()
75 computation_status.status().ToString().c_str()); in RealMain()
78 Computation computation = computation_status.ConsumeValueOrDie(); in RealMain()
Ddumped_computation_to_text.cc46 auto computation_status = client->LoadSnapshot(session_module); in RealMain() local
47 if (!computation_status.ok()) { in RealMain()
49 computation_status.status().ToString().c_str()); in RealMain()
52 Computation computation = computation_status.ConsumeValueOrDie(); in RealMain()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dsample_harness.cc56 xla::StatusOr<xla::Computation> computation_status = builder.Build(); in main() local
57 xla::Computation computation = computation_status.ConsumeValueOrDie(); in main()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.cc415 xla::StatusOr<xla::Computation> computation_status = builder->Build(); in BuildComputation() local
416 if (!computation_status.ok()) { in BuildComputation()
417 return computation_status.status(); in BuildComputation()
419 *computation = computation_status.ConsumeValueOrDie(); in BuildComputation()