Home
last modified time | relevance | path

Searched defs:broadcast_dimensions (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dchlo_ops.cc96 llvm::SmallVector<int64_t, 4> broadcast_dimensions; in GetBroadcastType() local
132 DenseIntElementsAttr broadcast_dimensions = in InferBroadcastBinaryOpReturnTypeComponents() local
166 auto broadcast_dimensions = op->getAttr("broadcast_dimensions") in ReifyBroadcastBinaryOpReturnTypeShapes() local
219 DenseIntElementsAttr broadcast_dimensions, in build()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc552 absl::Span<const int64> broadcast_dimensions) { in InDimBroadcast()
581 std::vector<int64> broadcast_dimensions; in AddBroadcastSequence() local
626 absl::Span<const int64> broadcast_dimensions, in BinaryOp()
892 const absl::Span<const int64> broadcast_dimensions) { in BroadcastInDim()
4104 const absl::Span<const int64> broadcast_dimensions) { in BroadcastInDim()
4199 absl::Span<const int64> broadcast_dimensions) { in Eq()
4204 absl::Span<const int64> broadcast_dimensions, in CompareTotalOrder()
4220 absl::Span<const int64> broadcast_dimensions) { in EqTotalOrder()
4226 absl::Span<const int64> broadcast_dimensions) { in Ne()
4231 absl::Span<const int64> broadcast_dimensions) { in NeTotalOrder()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dselect_op.cc68 std::vector<int64> broadcast_dimensions(cond_shape.dims()); in Compile() local
Ddiag_op.cc67 std::vector<int64> broadcast_dimensions(other_dims.size() + 1); in CreateDiagonal() local
Dmatrix_diag_ops.cc158 std::vector<int64> broadcast_dimensions(input_rank - 1); in SetMatrixDiag() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils.cc206 absl::Span<const int64> broadcast_dimensions, in MakeBroadcastHlo()
217 absl::Span<const int64> broadcast_dimensions, in MakeBroadcastHlo()
551 absl::Span<const int64> broadcast_dimensions) { in BroadcastZeros()
560 absl::Span<const int64> broadcast_dimensions) { in BroadcastOnes()
Dshape_inference.cc780 absl::Span<const int64> broadcast_dimensions) { in InferInDimBroadcastShape()
897 absl::Span<const int64> broadcast_dimensions) { in InferElementwiseBinaryOpShape()
959 absl::Span<const int64> broadcast_dimensions) { in InferBinaryOpShape()
2861 absl::Span<const int64> broadcast_dimensions) { in InferBroadcastShape()
Dhlo_instruction.cc1363 absl::Span<const int64> broadcast_dimensions) { in CreateBroadcast()
1405 std::vector<int64> broadcast_dimensions; in CreateBroadcastSequence() local
Dhlo_parser.cc1741 optional<std::vector<int64>> broadcast_dimensions; in ParseInstructionRhs() local
Dalgebraic_simplifier.cc3402 std::vector<int64> broadcast_dimensions; in HandlePad() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc1190 bool IsTFStyleBroadcast(DenseIntElementsAttr broadcast_dimensions, in IsTFStyleBroadcast()
1204 DenseIntElementsAttr broadcast_dimensions, in ExpandedShape()
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.cc392 absl::Span<const int64> broadcast_dimensions) { in InDimBroadcast()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dchlo_legalize_to_hlo.cc1065 auto broadcast_dimensions = op.broadcast_dimensions(); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc398 llvm::SmallVector<int64_t, 4> broadcast_dimensions; in GetStaticBroadcastType() local
1344 auto broadcast_dimensions = llvm::to_vector<4>( in matchAndRewrite() local
2599 SmallVector<int64_t, 4> broadcast_dimensions(rank); in matchAndRewrite() local
3996 SmallVector<int64_t, 4> broadcast_dimensions; in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc150 llvm::Optional<mlir::DenseIntElementsAttr> broadcast_dimensions) { in Convert_broadcast_dimensions()