Searched defs:broadcast_sizes (Results 1 – 4 of 4) sorted by relevance
308 std::vector<int64> broadcast_sizes; in EinsumInverseDiagonal() local
341 absl::Span<const int64> broadcast_sizes = {}) { in Uint128AddUint64()
2836 const Shape& operand, absl::Span<const int64> broadcast_sizes) { in InferBroadcastShape()
867 absl::Span<const int64> broadcast_sizes) { in Broadcast()4098 XlaOp Broadcast(const XlaOp operand, absl::Span<const int64> broadcast_sizes) { in Broadcast()