Searched refs:extent2 (Results 1 – 1 of 1) sorted by relevance
876 const int extent2 = extended_input2_shape.Dims(i); in NdArrayDescsForElementwiseBroadcast() local880 if (extent2 != 1) extent = extent2; in NdArrayDescsForElementwiseBroadcast()884 TFLITE_DCHECK(extent2 == 1 || extent2 == extent); in NdArrayDescsForElementwiseBroadcast()886 if (!(extent0 == extent1 && extent1 == extent2)) { in NdArrayDescsForElementwiseBroadcast()895 if (extent2 == 1) { in NdArrayDescsForElementwiseBroadcast()