Searched defs:broadcast_dimensions (Results 1 – 7 of 7) sorted by relevance
420 absl::Span<const int64> broadcast_dimensions) { in InDimBroadcast()448 std::vector<int64> broadcast_dimensions; in AddBroadcastSequence() local483 absl::Span<const int64> broadcast_dimensions, in BinaryOp()673 const absl::Span<const int64> broadcast_dimensions) { in BroadcastInDim()2845 const absl::Span<const int64> broadcast_dimensions) { in BroadcastInDim()2922 absl::Span<const int64> broadcast_dimensions) { in Eq()2927 absl::Span<const int64> broadcast_dimensions) { in Ne()2932 absl::Span<const int64> broadcast_dimensions) { in Ge()2937 absl::Span<const int64> broadcast_dimensions) { in Gt()2942 absl::Span<const int64> broadcast_dimensions) { in Le()[all …]
177 absl::Span<const int64> broadcast_dimensions, in MakeBroadcastHlo()439 absl::Span<const int64> broadcast_dimensions) { in BroadcastZeros()
745 absl::Span<const int64> broadcast_dimensions) { in InferInDimBroadcastShape()862 absl::Span<const int64> broadcast_dimensions) { in InferElementwiseBinaryOpShape()917 absl::Span<const int64> broadcast_dimensions) { in InferBinaryOpShape()2627 absl::Span<const int64> broadcast_dimensions) { in InferBroadcastShape()
1102 absl::Span<const int64> broadcast_dimensions) { in CreateBroadcast()1135 std::vector<int64> broadcast_dimensions; in CreateBroadcastSequence() local
1156 optional<std::vector<int64>> broadcast_dimensions; in ParseInstructionRhs() local
477 absl::Span<const int64> broadcast_dimensions) { in BroadcastInDim()
1104 def BroadcastInDim(self, operand, shape, broadcast_dimensions): argument