Searched defs:broadcast_sizes (Results 1 – 4 of 4) sorted by relevance
339 std::vector<int64> broadcast_sizes; in EinsumInverseDiagonal() local
342 absl::Span<const int64> broadcast_sizes = {}) { in Uint128AddUint64()
2973 const Shape& operand, absl::Span<const int64> broadcast_sizes) { in InferBroadcastShape()
877 absl::Span<const int64> broadcast_sizes) { in Broadcast()3910 XlaOp Broadcast(const XlaOp operand, absl::Span<const int64> broadcast_sizes) { in Broadcast()