Searched refs:bshape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment_test.cc | 353 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/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_zeros.py | 276 bshape = array_ops.broadcast_static_shape(x.get_shape(), special_shape)
279 if bshape == x.get_shape():
|
| D | linear_operator_identity.py | 332 bshape = array_ops.broadcast_static_shape(x.get_shape(), special_shape)
335 if bshape == x.get_shape():
|