| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | xla_builder.cc | 615 absl::Span<const int64_t> broadcast_dimensions) { in InDimBroadcast()
644 std::vector<int64_t> broadcast_dimensions; in AddBroadcastSequence() local
689 absl::Span<const int64_t> broadcast_dimensions, in BinaryOp()
958 const absl::Span<const int64_t> broadcast_dimensions) { in BroadcastInDim()
4060 const absl::Span<const int64_t> broadcast_dimensions) { in BroadcastInDim()
4151 absl::Span<const int64_t> broadcast_dimensions) { in Eq()
4156 absl::Span<const int64_t> broadcast_dimensions, in CompareTotalOrder()
4172 absl::Span<const int64_t> broadcast_dimensions) { in EqTotalOrder()
4178 absl::Span<const int64_t> broadcast_dimensions) { in Ne()
4183 absl::Span<const int64_t> broadcast_dimensions) { in NeTotalOrder()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_creation_utils.cc | 238 absl::Span<const int64_t> broadcast_dimensions, in MakeBroadcastHlo()
252 absl::Span<const int64_t> broadcast_dimensions, in MakeBroadcastHlo()
670 absl::Span<const int64_t> broadcast_dimensions) { in BroadcastZeros()
679 absl::Span<const int64_t> broadcast_dimensions) { in BroadcastOnes()
|
| H A D | shape_inference.cc | 784 absl::Span<const int64_t> broadcast_dimensions) { in InferInDimBroadcastShape()
901 absl::Span<const int64_t> broadcast_dimensions) { in InferElementwiseBinaryOpShape()
963 absl::Span<const int64_t> broadcast_dimensions) { in InferBinaryOpShape()
3001 absl::Span<const int64_t> broadcast_dimensions) { in InferBroadcastShape()
|
| H A D | hlo_instruction.cc | 1662 absl::Span<const int64_t> broadcast_dimensions) { in CreateBroadcast()
1704 std::vector<int64_t> broadcast_dimensions; in CreateBroadcastSequence() local
|
| H A D | hlo_parser.cc | 2114 optional<std::vector<int64_t>> broadcast_dimensions; in CreateInstruction() local
|
| H A D | algebraic_simplifier.cc | 3781 std::vector<int64_t> broadcast_dimensions; in HandlePad() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | diag_op.cc | 68 std::vector<int64_t> broadcast_dimensions(other_dims.size() + 1); in CreateDiagonal() local
|
| H A D | select_op.cc | 69 std::vector<int64_t> broadcast_dimensions(cond_shape.dims()); in Compile() local
|
| H A D | matrix_diag_ops.cc | 161 std::vector<int64_t> broadcast_dimensions(input_rank - 1); in SetMatrixDiag() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_prims.py | 41 def _wrapper(a, b, broadcast_dimensions): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| H A D | mlir_hlo_builder.cc | 513 absl::Span<const int64_t> broadcast_dimensions) { in InDimBroadcast()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | legalize_hlo.cc | 3269 bool IsTFStyleBroadcast(DenseIntElementsAttr broadcast_dimensions, in IsTFStyleBroadcast()
3283 DenseIntElementsAttr broadcast_dimensions, in ExpandedShape()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | legalize_tf.cc | 327 llvm::SmallVector<int64_t, 4> broadcast_dimensions; in GetStaticBroadcastType() local
1624 llvm::SmallVector<int64_t, 4> broadcast_dimensions; in matchAndRewrite() local
4653 SmallVector<int64_t, 4> broadcast_dimensions; in matchAndRewrite() local
4763 SmallVector<int64_t, 4> broadcast_dimensions; in matchAndRewrite() local
|
| /aosp_15_r20/external/pytorch/torch/_prims/ |
| H A D | __init__.py | 1313 def _broadcast_in_dim_aten(a, shape, broadcast_dimensions): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ |
| H A D | mlir_hlo_to_hlo.cc | 176 llvm::Optional<mlir::DenseIntElementsAttr> broadcast_dimensions) { in Convert_broadcast_dimensions()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | ir_emitter_unnested.cc | 1202 llvm::SmallVector<int64_t> broadcast_dimensions; in RowVectorizationEnabled() local
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | lowering.py | 949 def broadcast_in_dim(a, shape, broadcast_dimensions): argument
|