Home
last modified time | relevance | path

Searched refs:computation1 (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_test.cc76 auto computation1 = module->AddEntryComputation(CreateConstantComputation()); in TEST_F() local
81 ::testing::UnorderedElementsAre(computation1, computation2)); in TEST_F()
85 EXPECT_EQ(computation1->name(), "Constant"); in TEST_F()
92 auto computation1 = in TEST_F() local
95 module->AddEmbeddedComputation(CreateCallComputation({computation1})); in TEST_F()
97 module->AddEmbeddedComputation(CreateCallComputation({computation1})); in TEST_F()
158 auto computation1 = in TEST_F() local
161 module->AddEmbeddedComputation(CreateCallComputation({computation1})); in TEST_F()
163 module->AddEmbeddedComputation(CreateCallComputation({computation1})); in TEST_F()
169 ::testing::UnorderedElementsAre(computation1, computation2, in TEST_F()
[all …]
Dhlo_dataflow_analysis_test.cc1800 auto computation1 = module_->AddEmbeddedComputation( in TEST_P() local
1823 scalar_shape_, pred_cond, true_operand_cond, computation1, in TEST_P()
1856 EXPECT_TRUE(analysis.ValueIsDefinedAt(computation1->root_instruction())); in TEST_P()
1860 auto computation1_param = computation1->parameter_instruction(0); in TEST_P()
1898 analysis.GetValueDefinedAt(computation1->root_instruction()), in TEST_P()
1903 analysis.GetValueDefinedAt(computation1->root_instruction()), in TEST_P()
/external/tensorflow/tensorflow/compiler/xrt/tests/
Draw_api_test.cc477 auto computation1 = in TEST() local
479 auto c_handle1 = ops::XRTCompile(root, computation1); in TEST()