Searched defs:broadcast_dimension (Results 1 – 2 of 2) sorted by relevance
294 def broadcast_dimension(self, axis, lengths): member in RaggedTensorDynamicShape
1045 absl::Span<const int64> broadcast_dimension) in HloBroadcastInstruction()