Searched refs:f32vec4_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | heap_simulator_test.cc | 362 Shape f32vec4_ = ShapeUtil::MakeShape(F32, {4}); member in xla::__anon72fb25430111::HeapSimulatorTest
394 HloInstruction::CreateParameter(1, f32vec4_, "paramX")); in TEST_F()
396 f32vec4_, HloOpcode::kMultiply, paramA, paramX)); in TEST_F()
418 HloInstruction::CreateParameter(1, f32vec4_, "paramX")); in TEST_F()
420 HloInstruction::CreateParameter(2, f32vec4_, "paramY")); in TEST_F()
422 f32vec4_, HloOpcode::kMultiply, paramA, paramX)); in TEST_F()
424 HloInstruction::CreateBinary(f32vec4_, HloOpcode::kAdd, mul, paramY)); in TEST_F()
452 /*parameter_number=*/0, f32vec4_, "A")); in TEST_F()
454 HloInstruction::CreateUnary(f32vec4_, HloOpcode::kExp, a_param)); in TEST_F()
456 HloInstruction::CreateUnary(f32vec4_, HloOpcode::kNegate, a_param)); in TEST_F()
[all …]
|
| D | buffer_assignment_test.cc | 301 Shape f32vec4_ = ShapeUtil::MakeShape(F32, {4}); member in xla::__anon3198ac380111::BufferAssignmentTest
305 Shape t_s32_f32v4_ = ShapeUtil::MakeTupleShape({s32_, f32vec4_});
361 HloInstruction::CreateBinary(f32vec4_, HloOpcode::kAdd, const0, const1)); in TEST_F()
1386 auto elem_shape = f32vec4_; in TEST_F()
1427 auto elem_shape = f32vec4_; in TEST_F()
|