Searched defs:broadcast_sizes (Results 1 – 3 of 3) sorted by relevance
471 absl::Span<const int64> broadcast_sizes) { in Broadcast()
2608 const Shape& operand, absl::Span<const int64> broadcast_sizes) { in InferBroadcastShape()
648 absl::Span<const int64> broadcast_sizes) { in Broadcast()2839 XlaOp Broadcast(const XlaOp& operand, absl::Span<const int64> broadcast_sizes) { in Broadcast()