Searched defs:ashape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment_test.cc | 96 Shape ashape = ShapeUtil::MakeShape(F32, {42, 12}); in TEST_F() local
126 Shape ashape = ShapeUtil::MakeShape(F32, {42, 12}); in TEST_F() local
169 Shape ashape = constant_literal1.shape(); in TEST_F() local
352 Shape ashape = ShapeUtil::MakeShape(F32, {1, 2, 3, 1}); in TEST_F() local
394 Shape ashape = ShapeUtil::MakeShape(F32, {42, 12}); in TEST_F() local
431 Shape ashape = ShapeUtil::MakeShape(F32, {3, 4}); in TEST_F() local
546 Shape ashape = ShapeUtil::MakeShape(F32, {50, 1}); in TEST_F() local
1325 Shape ashape = ShapeUtil::MakeShape(F32, {12, 8}); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment_test.cc | 42 Shape ashape = ShapeUtil::MakeShape(F32, {42, 12}); in TEST_F() local
|