Searched refs:xpose (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_propagation_test.cc | 116 HloInstruction* xpose = in TEST_F() local
120 CreateDot(ShapeUtil::MakeShape(F32, {4, 4}), xpose, a)); in TEST_F()
130 EXPECT_TRUE(OutputsBF16(xpose)); in TEST_F()
177 HloInstruction* xpose = in TEST_F() local
181 CreateDot(ShapeUtil::MakeShape(F32, {4, 4}), xpose, rw)); in TEST_F()
191 EXPECT_TRUE(OutputsBF16(xpose)); in TEST_F()
278 HloInstruction* xpose = in TEST_F() local
285 builder.AddInstruction(HloInstruction::CreateTuple({tuple0, xpose})); in TEST_F()
288 HloInstruction::CreateGetTupleElement(xpose->shape(), tuple1, 1)); in TEST_F()
307 EXPECT_TRUE(OutputsBF16(xpose)); in TEST_F()
[all …]
|