Searched refs:broadcast_shape_or_status (Results 1 – 1 of 1) sorted by relevance
216 auto broadcast_shape_or_status = builder->GetShape(input); in BroadcastSpatialDimensions() local217 if (!broadcast_shape_or_status.ok()) { in BroadcastSpatialDimensions()218 return builder->ReportError(broadcast_shape_or_status.status()); in BroadcastSpatialDimensions()220 xla::Shape broadcast_shape = broadcast_shape_or_status.ValueOrDie(); in BroadcastSpatialDimensions()