Searched refs:desc1_out (Results 1 – 1 of 1) sorted by relevance
774 NdArrayDesc<N>* desc1_out) { in NdArrayDescsForElementwiseBroadcast() argument776 TFLITE_DCHECK(desc1_out != nullptr); in NdArrayDescsForElementwiseBroadcast()782 desc1_out->extents[i] = input1_dims.sizes[i]; in NdArrayDescsForElementwiseBroadcast()783 desc1_out->strides[i] = input1_dims.strides[i]; in NdArrayDescsForElementwiseBroadcast()798 desc1_out->strides[i] = 0; in NdArrayDescsForElementwiseBroadcast()799 desc1_out->extents[i] = extent0; in NdArrayDescsForElementwiseBroadcast()820 NdArrayDesc<N>* desc0_out, NdArrayDesc<N>* desc1_out) { in NdArrayDescsForElementwiseBroadcast() argument822 TFLITE_DCHECK(desc1_out != nullptr); in NdArrayDescsForElementwiseBroadcast()829 CopyDimsToDesc<N>(extended_input1_shape, desc1_out); in NdArrayDescsForElementwiseBroadcast()843 desc1_out->strides[i] = 0; in NdArrayDescsForElementwiseBroadcast()[all …]