Searched refs:broadcast_dimension (Results 1 – 5 of 5) sorted by relevance
300 bcast1 = original_shape.broadcast_dimension(axis, row_length)302 bcast2 = bcast_shape.broadcast_dimension(axis, row_length)304 bcast3 = bcast_shape.broadcast_dimension(axis, 1)
294 def broadcast_dimension(self, axis, lengths): member in RaggedTensorDynamicShape470 shape_x = shape_x.broadcast_dimension(axis, shape_y.dimension_size(axis))471 shape_y = shape_y.broadcast_dimension(axis, shape_x.dimension_size(axis))
1045 absl::Span<const int64> broadcast_dimension) in HloBroadcastInstruction() argument1047 dimensions_(broadcast_dimension.begin(), broadcast_dimension.end()) { in HloBroadcastInstruction()
694 absl::Span<const int64> broadcast_dimension);
416 broadcast_dimension\[i\]'th dimension of the output shape. The dimensions of