Searched refs:desc2_out (Results 1 – 1 of 1) sorted by relevance
633 NdArrayDesc<N>* desc1_out, NdArrayDesc<N>* desc2_out) { in NdArrayDescsForElementwiseBroadcast() argument636 TFLITE_DCHECK(desc2_out != nullptr); in NdArrayDescsForElementwiseBroadcast()645 CopyDimsToDesc<N>(extended_input2_shape, desc2_out); in NdArrayDescsForElementwiseBroadcast()673 desc2_out->strides[i] = 0; in NdArrayDescsForElementwiseBroadcast()674 desc2_out->extents[i] = extent; in NdArrayDescsForElementwiseBroadcast()