Home
last modified time | relevance | path

Searched refs:ashape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment_test.cc96 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/
Dgpu_layout_assignment_test.cc44 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/
Dcall03.f90148 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/
Dshape.cpp547 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 ()()