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 ()()
|