| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | LinearAlgebraUtils.h | 550 …Tensor get_linear_indices(int64_t numel, IntArrayRef original_shape, IntArrayRef broadcast_shape) { in get_linear_indices()
564 IntArrayRef broadcast_shape) : is_broadcasting_(!original_shape.equals(broadcast_shape)) { in BroadcastLinearIndices()
|
| H A D | TensorAdvancedIndexing.cpp | 2172 auto broadcast_shape = infer_size_symint(self_sizes, indices.sym_sizes()); in _take_along_dim_helper() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | common_shapes.py | 87 def broadcast_shape(shape_x, shape_y): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | broadcast.cc | 46 std::vector<int64_t> broadcast_shape; in BroadcastTo() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | xla_broadcast_helper_op.cc | 81 std::vector<int64_t> broadcast_shape(max_rank_shape->dims(), 1LL); in Compile() local
|
| H A D | image_resize_ops.cc | 226 xla::Shape broadcast_shape = broadcast_shape_or_status.ValueOrDie(); in BroadcastSpatialDimensions() local
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/core/ |
| H A D | TensorShape.h | 211 static TensorShape broadcast_shape(const Shapes &... shapes) in broadcast_shape() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | optimize.cc | 77 ArrayRef<int64_t> broadcast_shape = output_type.getShape(); in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | while_loop_concat_code_motion.cc | 873 std::vector<int64_t> broadcast_shape; in RewriteLoopWithConcatGroups() local
|
| H A D | hlo_creation_utils.cc | 242 Shape broadcast_shape = ShapeUtil::MakeShape(operand->shape().element_type(), in MakeBroadcastHlo() local
|
| H A D | hlo_instruction.cc | 1690 Shape broadcast_shape = ShapeUtil::ChangeElementType( in CreateBroadcastSequence() local
|
| H A D | algebraic_simplifier_test.cc | 4715 Shape broadcast_shape = ShapeUtil::MakeShape(F32, {4, 5, 6, 7}); in TEST_F() local
4748 Shape broadcast_shape = ShapeUtil::MakeShape(F32, {4, 5, 6}); in TEST_F() local
|
| H A D | algebraic_simplifier.cc | 3800 auto broadcast_shape = broadcast->shape(); in HandlePad() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| H A D | legalize_tf.cc | 655 auto broadcast_shape = in rewriteOpWithDynamicInput() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_instruction_fusion_test.cc | 320 Shape broadcast_shape = ShapeUtil::MakeShape(F32, {1, 8, 8}); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/rnn/ |
| H A D | lstm_ops.cc | 67 Eigen::array<Eigen::DenseIndex, 2> broadcast_shape({cell.batch_size(), 1}); in LSTMBlockCellFpropWithEigen() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | arithmetic_optimizer.cc | 3342 TensorShapeProto broadcast_shape; in TrySimplifyInternal() local
3411 TensorShapeProto broadcast_shape; in TrySimplify() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A D | types.h | 672 int broadcast_shape[5]; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/ |
| H A D | pass.cc | 609 SmallVector<int64_t> broadcast_shape; in IsValidConstShapeForMulConvPushDown() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_ops_n_z.cc | 3343 llvm::SmallVector<int64_t, 4> broadcast_shape(output_rank, 1LL); in inferReturnTypeComponents() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| H A D | tfl_ops.cc | 122 Value broadcast_shape = defining_op->getOperand( in getDefiningBroadcastArgsOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | legalize_tf.cc | 4776 SmallVector<int64_t, 4> broadcast_shape(input_rank * 2, in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | xla_builder.cc | 635 Shape broadcast_shape = in AddBroadcastSequence() local
|