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