Searched refs:desc2_out (Results 1 – 1 of 1) sorted by relevance
854 NdArrayDesc<N>* desc1_out, NdArrayDesc<N>* desc2_out) { in NdArrayDescsForElementwiseBroadcast() argument857 TFLITE_DCHECK(desc2_out != nullptr); in NdArrayDescsForElementwiseBroadcast()866 CopyDimsToDesc<N>(extended_input2_shape, desc2_out); in NdArrayDescsForElementwiseBroadcast()894 desc2_out->strides[i] = 0; in NdArrayDescsForElementwiseBroadcast()895 desc2_out->extents[i] = extent; in NdArrayDescsForElementwiseBroadcast()