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