Searched refs:vector_shape_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_dataflow_analysis_test.cc | 96 const Shape vector_shape_ = ShapeUtil::MakeShape(F32, {42}); member in xla::__anondaadebfd0111::HloDataflowAnalysisTest
1236 HloInstruction::CreateParameter(0, vector_shape_, "param")); in TEST_P()
1238 HloInstruction::CreateUnary(vector_shape_, HloOpcode::kNegate, param)); in TEST_P()
1240 HloInstruction::CreateUnary(vector_shape_, HloOpcode::kExp, negate)); in TEST_P()
1242 HloInstruction::CreateUnary(vector_shape_, HloOpcode::kLog, exp)); in TEST_P()
1272 HloInstruction::CreateParameter(0, vector_shape_, "param0")); in TEST_P()
1274 HloInstruction::CreateParameter(1, vector_shape_, "param1")); in TEST_P()
1276 HloInstruction::CreateUnary(vector_shape_, HloOpcode::kNegate, param0)); in TEST_P()
1278 HloInstruction::CreateUnary(vector_shape_, HloOpcode::kExp, param1)); in TEST_P()
1280 vector_shape_, HloOpcode::kAdd, negate, exp)); in TEST_P()
[all …]
|