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