Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_live_range_test.cc54 Shape f32vec4_ = ShapeUtil::MakeShape(F32, {4}); member in xla::__anon8ef0e3640111::HloLiveRangeTest
90 HloInstruction::CreateParameter(0, f32vec4_, "paramA")); in TEST_F()
92 HloInstruction::CreateParameter(1, f32vec4_, "paramX")); in TEST_F()
94 f32vec4_, HloOpcode::kMultiply, paramA, paramX)); in TEST_F()
115 HloInstruction::CreateParameter(0, f32vec4_, "paramA")); in TEST_F()
117 HloInstruction::CreateParameter(1, f32vec4_, "paramX")); in TEST_F()
119 f32vec4_, HloOpcode::kMultiply, paramA, paramX)); in TEST_F()
121 HloInstruction::CreateParameter(2, f32vec4_, "paramY")); in TEST_F()
123 HloInstruction::CreateBinary(f32vec4_, HloOpcode::kAdd, mul, paramY)); in TEST_F()
151 HloInstruction::CreateParameter(0, f32vec4_, "paramA")); in TEST_F()
[all …]
Dheap_simulator_test.cc413 Shape f32vec4_ = ShapeUtil::MakeShape(F32, {4}); member in xla::__anon72fb25430111::HeapSimulatorTest
445 HloInstruction::CreateParameter(1, f32vec4_, "paramX")); in TEST_F()
447 f32vec4_, HloOpcode::kMultiply, paramA, paramX)); in TEST_F()
469 HloInstruction::CreateParameter(1, f32vec4_, "paramX")); in TEST_F()
471 HloInstruction::CreateParameter(2, f32vec4_, "paramY")); in TEST_F()
473 f32vec4_, HloOpcode::kMultiply, paramA, paramX)); in TEST_F()
475 HloInstruction::CreateBinary(f32vec4_, HloOpcode::kAdd, mul, paramY)); in TEST_F()
514 HloInstruction::CreateParameter(0, f32vec4_, "paramA")); in TEST_F()
516 HloInstruction::CreateUnary(f32vec4_, HloOpcode::kNegate, paramA)); in TEST_F()
523 HloInstruction::CreateParameter(0, f32vec4_, "x")); in TEST_F()
[all …]
Dbuffer_assignment_test.cc308 Shape f32vec4_ = ShapeUtil::MakeShape(F32, {4}); member in xla::__anon3198ac380111::BufferAssignmentTest
312 Shape t_s32_f32v4_ = ShapeUtil::MakeTupleShape({s32_, f32vec4_});
368 HloInstruction::CreateBinary(f32vec4_, HloOpcode::kAdd, const0, const1)); in TEST_F()
1558 auto elem_shape = f32vec4_; in TEST_F()
1599 auto elem_shape = f32vec4_; in TEST_F()