| /external/tensorflow/tensorflow/python/framework/ |
| common_shapes.py | 503 """Helper functions for is_broadcast_compatible and broadcast_shape.
566 def broadcast_shape(shape_x, shape_y): function
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| diag_op.cc | 75 auto broadcast_shape = local
77 xla::ComputationDataHandle zeros = Zeros(builder, broadcast_shape);
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_instruction_fusion_test.cc | 274 Shape broadcast_shape = ShapeUtil::MakeShape(F32, {1, 8, 8}); local
282 HloInstruction::CreateBroadcast(broadcast_shape, param0, {1}));
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| algebraic_simplifier_test.cc | 2161 Shape broadcast_shape = ShapeUtil::MakeShape(F32, {4, 5, 6, 7}); local
2198 Shape broadcast_shape = ShapeUtil::MakeShape(F32, {4, 5, 6}); local
[all...] |
| hlo_instruction.cc | 682 Shape broadcast_shape = ShapeUtil::ChangeElementType( local
687 HloInstruction::CreateBroadcast(broadcast_shape, operand, {});
719 broadcast_shape, reshaped_operand, broadcast_dimensions);
[all...] |
| user_computation.cc | 3442 Shape broadcast_shape = ShapeUtil::MakeShape( local
[all...] |
