Searched defs:ashape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment_test.cc | 97 Shape ashape = ShapeUtil::MakeShape(F32, {42, 12}); in TEST_F() local
127 Shape ashape = ShapeUtil::MakeShape(F32, {42, 12}); in TEST_F() local
170 Shape ashape = constant_literal1.shape(); in TEST_F() local
318 Shape ashape = ShapeUtil::MakeShape(F32, {1, 2, 3, 1}); in TEST_F() local
358 Shape ashape = ShapeUtil::MakeShape(F32, {42, 12}); in TEST_F() local
395 Shape ashape = ShapeUtil::MakeShape(F32, {3, 4}); in TEST_F() local
509 Shape ashape = ShapeUtil::MakeShape(F32, {50, 1}); in TEST_F() local
1354 Shape ashape = ShapeUtil::MakeShape(F32, {12, 8}); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment_test.cc | 47 Shape ashape = ShapeUtil::MakeShape(F32, {42, 12}); in TEST_F() local
|