Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_normalization_test.cc95 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F() local
100 HloInstruction::CreateParameter(1, bf16_shape, "b")); in TEST_F()
105 HloInstruction::CreateBinary(bf16_shape, HloOpcode::kAdd, a, b)); in TEST_F()
123 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F() local
128 HloInstruction::CreateParameter(1, bf16_shape, "b")); in TEST_F()
133 HloInstruction::CreateBinary(bf16_shape, HloOpcode::kMultiply, a, b)); in TEST_F()
136 HloInstruction::CreateBinary(bf16_shape, HloOpcode::kMultiply, mul0, c)); in TEST_F()
153 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F() local
158 HloInstruction::CreateParameter(1, bf16_shape, "b")); in TEST_F()
163 HloInstruction::CreateBinary(bf16_shape, HloOpcode::kSubtract, a, b)); in TEST_F()
[all …]
Dbfloat16_conversion_folding_test.cc86 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F() local
98 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, add0)); in TEST_F()
104 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, add1)); in TEST_F()
120 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F() local
132 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, mul0)); in TEST_F()
139 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, mul1)); in TEST_F()
155 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F() local
167 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, sub0)); in TEST_F()
174 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, sub1)); in TEST_F()
190 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F() local
[all …]
Dbfloat16_propagation_test.cc902 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {4, 4}); in TEST_F() local
917 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, gte0)); in TEST_F()
919 builder.AddInstruction(HloInstruction::CreateConvert(bf16_shape, gte1)); in TEST_F()
921 bf16_shape, HloOpcode::kAdd, convert0, convert1)); in TEST_F()