Searched defs:broadcast_dimensions (Results 1 – 11 of 11) sorted by relevance
474 absl::Span<const int64> broadcast_dimensions) { in InDimBroadcast()503 std::vector<int64> broadcast_dimensions; in AddBroadcastSequence() local548 absl::Span<const int64> broadcast_dimensions, in BinaryOp()784 const absl::Span<const int64> broadcast_dimensions) { in BroadcastInDim()2900 const absl::Span<const int64> broadcast_dimensions) { in BroadcastInDim()2987 absl::Span<const int64> broadcast_dimensions) { in Eq()2992 absl::Span<const int64> broadcast_dimensions) { in Ne()2997 absl::Span<const int64> broadcast_dimensions) { in Ge()3002 absl::Span<const int64> broadcast_dimensions) { in Gt()3007 absl::Span<const int64> broadcast_dimensions) { in Le()[all …]
68 std::vector<int64> broadcast_dimensions(cond_shape.dims()); in Compile() local
67 std::vector<int64> broadcast_dimensions(other_dims.size() + 1); in CreateDiagonal() local
158 std::vector<int64> broadcast_dimensions(input_rank - 1); in SetMatrixDiag() local
193 absl::Span<const int64> broadcast_dimensions, in MakeBroadcastHlo()204 absl::Span<const int64> broadcast_dimensions, in MakeBroadcastHlo()492 absl::Span<const int64> broadcast_dimensions) { in BroadcastZeros()
760 absl::Span<const int64> broadcast_dimensions) { in InferInDimBroadcastShape()877 absl::Span<const int64> broadcast_dimensions) { in InferElementwiseBinaryOpShape()939 absl::Span<const int64> broadcast_dimensions) { in InferBinaryOpShape()2750 absl::Span<const int64> broadcast_dimensions) { in InferBroadcastShape()
1197 absl::Span<const int64> broadcast_dimensions) { in CreateBroadcast()1239 std::vector<int64> broadcast_dimensions; in CreateBroadcastSequence() local
1218 optional<std::vector<int64>> broadcast_dimensions; in ParseInstructionRhs() local
888 DenseIntElementsAttr broadcast_dimensions) { in GetBroadcastType()1247 Value rhs, DenseIntElementsAttr broadcast_dimensions, in build()
116 llvm::Optional<mlir::DenseIntElementsAttr> broadcast_dimensions) { in Convert_broadcast_dimensions()
1326 SmallVector<int64_t, 4> broadcast_dimensions(rank); in matchAndRewrite() local2338 SmallVector<int64_t, 4> broadcast_dimensions; in matchAndRewrite() local