Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_conversion_folding_test.cc85 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F() local
89 HloInstruction::CreateParameter(0, f32_shape, "a")); in TEST_F()
91 HloInstruction::CreateParameter(1, f32_shape, "b")); in TEST_F()
93 HloInstruction::CreateParameter(2, f32_shape, "c")); in TEST_F()
96 HloInstruction::CreateBinary(f32_shape, HloOpcode::kAdd, a, b)); in TEST_F()
100 HloInstruction::CreateConvert(f32_shape, convert0)); in TEST_F()
103 HloInstruction::CreateBinary(f32_shape, HloOpcode::kAdd, convert1, c)); in TEST_F()
119 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F() local
123 HloInstruction::CreateParameter(0, f32_shape, "a")); in TEST_F()
125 HloInstruction::CreateParameter(1, f32_shape, "b")); in TEST_F()
[all …]
Dbfloat16_normalization_test.cc94 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F() local
98 HloInstruction::CreateParameter(0, f32_shape, "a")); in TEST_F()
102 HloInstruction::CreateParameter(2, f32_shape, "c")); in TEST_F()
108 HloInstruction::CreateBinary(f32_shape, HloOpcode::kAdd, add0, c)); in TEST_F()
122 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F() local
126 HloInstruction::CreateParameter(0, f32_shape, "a")); in TEST_F()
130 HloInstruction::CreateParameter(2, f32_shape, "c")); in TEST_F()
152 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F() local
156 HloInstruction::CreateParameter(0, f32_shape, "a")); in TEST_F()
160 HloInstruction::CreateParameter(2, f32_shape, "c")); in TEST_F()
[all …]
Dbfloat16_propagation_test.cc901 Shape f32_shape = ShapeUtil::MakeShape(F32, {4, 4}); in TEST_F() local
905 HloInstruction::CreateParameter(0, f32_shape, "param")); in TEST_F()
907 HloInstruction::CreateBinary(f32_shape, HloOpcode::kAdd, param, param)); in TEST_F()
909 HloInstruction::CreateBinary(f32_shape, HloOpcode::kAdd, param, param)); in TEST_F()
913 HloInstruction::CreateGetTupleElement(f32_shape, tuple, 0)); in TEST_F()
915 HloInstruction::CreateGetTupleElement(f32_shape, tuple, 1)); in TEST_F()