Searched refs:input0_dims (Results 1 – 1 of 1) sorted by relevance
771 inline void NdArrayDescsForElementwiseBroadcast(const Dims<N>& input0_dims, in NdArrayDescsForElementwiseBroadcast() argument780 desc0_out->extents[i] = input0_dims.sizes[i]; in NdArrayDescsForElementwiseBroadcast()781 desc0_out->strides[i] = input0_dims.strides[i]; in NdArrayDescsForElementwiseBroadcast()790 const int extent0 = ArraySize(input0_dims, i); in NdArrayDescsForElementwiseBroadcast()