Searched refs:extended_input2_shape (Results 1 – 1 of 1) sorted by relevance
861 auto extended_input2_shape = RuntimeShape::ExtendedShape(N, input2_shape); in NdArrayDescsForElementwiseBroadcast() local866 CopyDimsToDesc<N>(extended_input2_shape, desc2_out); in NdArrayDescsForElementwiseBroadcast()874 const int extent2 = extended_input2_shape.Dims(i); in NdArrayDescsForElementwiseBroadcast()