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