Searched refs:ashape (Results 1 – 3 of 3) 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
98 HloInstruction::CreateParameter(0, ashape, "param0")); in TEST_F()
100 HloInstruction::CreateParameter(1, ashape, "param1")); in TEST_F()
102 HloInstruction::CreateBinary(ashape, HloOpcode::kAdd, param0, param1)); in TEST_F()
107 Shape shape(ashape); in TEST_F()
126 Shape ashape = ShapeUtil::MakeShape(F32, {42, 12}); in TEST_F() local
128 HloInstruction::CreateParameter(0, ashape, "param0")); in TEST_F()
130 HloInstruction::CreateParameter(1, ashape, "param1")); in TEST_F()
132 HloInstruction::CreateBinary(ashape, HloOpcode::kAdd, param0, param1)); in TEST_F()
137 Shape col_major_shape(ashape); in TEST_F()
[all …]
|
| /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
43 Shape ashape_in_row_major(ashape); in TEST_F()
44 Shape ashape_in_col_major(ashape); in TEST_F()
59 HloInstruction::CreateParameter(0, ashape, "x")); in TEST_F()
61 HloInstruction::CreateParameter(1, ashape, "y")); in TEST_F()
63 HloInstruction::CreateBinary(ashape, HloOpcode::kAdd, x, y)); in TEST_F()
|
| /external/toolchain-utils/cros_utils/ |
| D | pstat.py | 946 ashape = a.shape
961 return N.reshape(work, ashape)
|