Searched refs:computation1 (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_test.cc | 77 auto computation1 = module->AddEntryComputation(CreateConstantComputation()); in TEST_F() local
82 ::testing::UnorderedElementsAre(computation1, computation2)); in TEST_F()
86 EXPECT_EQ(computation1->name(), "Constant"); in TEST_F()
93 auto computation1 = in TEST_F() local
96 module->AddEmbeddedComputation(CreateCallComputation({computation1})); in TEST_F()
98 module->AddEmbeddedComputation(CreateCallComputation({computation1})); in TEST_F()
159 auto computation1 = in TEST_F() local
162 module->AddEmbeddedComputation(CreateCallComputation({computation1})); in TEST_F()
164 module->AddEmbeddedComputation(CreateCallComputation({computation1})); in TEST_F()
170 ::testing::UnorderedElementsAre(computation1, computation2, in TEST_F()
[all …]
|
| D | hlo_dataflow_analysis_test.cc | 1927 auto computation1 = module_->AddEmbeddedComputation( in TEST_P() local
1950 scalar_shape_, pred_cond, true_operand_cond, computation1, in TEST_P()
1983 EXPECT_TRUE(analysis.ValueIsDefinedAt(computation1->root_instruction())); in TEST_P()
1987 auto computation1_param = computation1->parameter_instruction(0); in TEST_P()
2025 analysis.GetValueDefinedAt(computation1->root_instruction()), in TEST_P()
2030 analysis.GetValueDefinedAt(computation1->root_instruction()), in TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client_backend_independent_test.py | 102 computation1 = builder1.build()
104 self.assertEqual(computation0.hash(), computation1.hash())
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 671 auto computation1 = in TEST() local
673 auto c_handle1 = ops::XRTCompile(root, computation1); in TEST()
|