Searched refs:desc0_out (Results 1 – 1 of 1) sorted by relevance
773 NdArrayDesc<N>* desc0_out, in NdArrayDescsForElementwiseBroadcast() argument775 TFLITE_DCHECK(desc0_out != nullptr); in NdArrayDescsForElementwiseBroadcast()780 desc0_out->extents[i] = input0_dims.sizes[i]; in NdArrayDescsForElementwiseBroadcast()781 desc0_out->strides[i] = input0_dims.strides[i]; in NdArrayDescsForElementwiseBroadcast()794 desc0_out->strides[i] = 0; in NdArrayDescsForElementwiseBroadcast()795 desc0_out->extents[i] = extent1; in NdArrayDescsForElementwiseBroadcast()820 NdArrayDesc<N>* desc0_out, NdArrayDesc<N>* desc1_out) { in NdArrayDescsForElementwiseBroadcast() argument821 TFLITE_DCHECK(desc0_out != nullptr); in NdArrayDescsForElementwiseBroadcast()828 CopyDimsToDesc<N>(extended_input0_shape, desc0_out); in NdArrayDescsForElementwiseBroadcast()839 desc0_out->strides[i] = 0; in NdArrayDescsForElementwiseBroadcast()[all …]