Searched refs:cshape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_fusion_test.cc | 153 auto cshape = ShapeUtil::MakeShape(F32, {6}); in TEST_F() local
155 HloInstruction::CreateConcatenate(cshape, {ceil, ceil}, /*dimension=*/0)); in TEST_F()
170 cshape, in TEST_F()
179 HloInstruction::CreateUnary(cshape, HloOpcode::kExp, reduce)); in TEST_F()
181 HloInstruction::CreateUnary(cshape, HloOpcode::kFloor, exp)); in TEST_F()
183 cshape, in TEST_F()
188 HloInstruction::CreateBinary(cshape, HloOpcode::kMultiply, two, floor)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment_test.cc | 433 Shape cshape = ShapeUtil::MakeShape(F32, {4, 3, 2}); in TEST_F() local
439 HloInstruction::CreateTranspose(cshape, broadcast, {2, 1, 0})); in TEST_F()
445 Shape output_shape_with_layout(cshape); in TEST_F()
548 Shape cshape = ShapeUtil::MakeShape(F32, {100}); in TEST_F() local
556 HloInstruction::CreateReshape(cshape, concatenate)); in TEST_F()
|