Searched refs:ashape (Results 1 – 4 of 4) 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 | 44 Shape ashape = ShapeUtil::MakeShape(F32, {42, 12}); in TEST_F() local
45 Shape ashape_in_row_major(ashape); in TEST_F()
46 Shape ashape_in_col_major(ashape); in TEST_F()
61 HloInstruction::CreateParameter(0, ashape, "x")); in TEST_F()
63 HloInstruction::CreateParameter(1, ashape, "y")); in TEST_F()
65 HloInstruction::CreateBinary(ashape, HloOpcode::kAdd, x, y)); in TEST_F()
|
| /external/llvm-project/flang/test/Semantics/ |
| D | call03.f90 | 148 subroutine test09(ashape, polyarray, c) ! 15.5.2.4(14), 15.5.2.11 argument
151 real :: ashape(:) local
159 call assumedsize(ashape(1))
|
| /external/llvm-project/flang/lib/Evaluate/ |
| D | shape.cpp | 547 if (auto ashape{(*this)(call.arguments()[0])}) { in operator ()() local
549 if (ashape->size() == 1 && bshape->size() == 2) { in operator ()()
552 } else if (ashape->size() == 2 && bshape->size() == 1) { in operator ()()
553 ashape->pop_back(); in operator ()()
554 return std::move(*ashape); // matmul(matrix, vector) in operator ()()
555 } else if (ashape->size() == 2 && bshape->size() == 2) { in operator ()()
556 (*ashape)[1] = std::move((*bshape)[1]); in operator ()()
557 return std::move(*ashape); // matmul(matrix, matrix) in operator ()()
|