Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment_test.cc353 Shape bshape = ShapeUtil::MakeShape(F32, {3, 1, 2}); in TEST_F() local
359 builder.AddInstruction(HloInstruction::CreateReshape(bshape, log)); in TEST_F()
361 HloInstruction::CreateUnary(bshape, HloOpcode::kTanh, reshape)); in TEST_F()
367 Shape bshape_with_layout(bshape); in TEST_F()
395 Shape bshape = ShapeUtil::MakeShape(F32, {12, 42}); in TEST_F() local
401 HloInstruction::CreateTranspose(bshape, log, {1, 0})); in TEST_F()
403 HloInstruction::CreateUnary(bshape, HloOpcode::kTanh, transpose)); in TEST_F()
408 Shape bshape_with_layout(bshape); in TEST_F()
432 Shape bshape = ShapeUtil::MakeShape(F32, {2, 3, 4}); in TEST_F() local
437 HloInstruction::CreateBroadcast(bshape, param, {1, 2})); in TEST_F()
[all …]
/external/llvm-project/flang/lib/Evaluate/
Dshape.cpp548 if (auto bshape{(*this)(call.arguments()[1])}) { in operator ()() local
549 if (ashape->size() == 1 && bshape->size() == 2) { in operator ()()
550 bshape->erase(bshape->begin()); in operator ()()
551 return std::move(*bshape); // matmul(vector, matrix) in operator ()()
552 } else if (ashape->size() == 2 && bshape->size() == 1) { in operator ()()
555 } else if (ashape->size() == 2 && bshape->size() == 2) { in operator ()()
556 (*ashape)[1] = std::move((*bshape)[1]); in operator ()()
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_zeros.py294 bshape = array_ops.broadcast_static_shape(x.shape, special_shape)
297 if bshape == x.shape:
Dlinear_operator_identity.py348 bshape = array_ops.broadcast_static_shape(x.shape, special_shape)
351 if bshape == x.shape: