Searched refs:extent1 (Results 1 – 1 of 1) sorted by relevance
793 const int extent1 = ArraySize(input1_dims, i); in NdArrayDescsForElementwiseBroadcast() local794 if (extent0 != extent1) { in NdArrayDescsForElementwiseBroadcast()797 desc0_out->extents[i] = extent1; in NdArrayDescsForElementwiseBroadcast()799 TFLITE_DCHECK_EQ(extent1, 1); in NdArrayDescsForElementwiseBroadcast()838 const int extent1 = extended_input1_shape.Dims(i); in NdArrayDescsForElementwiseBroadcast() local839 if (extent0 != extent1) { in NdArrayDescsForElementwiseBroadcast()842 desc0_out->extents[i] = extent1; in NdArrayDescsForElementwiseBroadcast()844 TFLITE_DCHECK_EQ(extent1, 1); in NdArrayDescsForElementwiseBroadcast()875 const int extent1 = extended_input1_shape.Dims(i); in NdArrayDescsForElementwiseBroadcast() local879 if (extent1 != 1) extent = extent1; in NdArrayDescsForElementwiseBroadcast()[all …]