Searched refs:XDims (Results 1 – 1 of 1) sorted by relevance
3179 auto XDims = X.dims(); in createElementwiseLinear() local3188 assert(XDims.size() == 2 && "X must be 2D"); in createElementwiseLinear()3192 assert(wDims[0] == XDims[axis] && in createElementwiseLinear()3194 assert(bDims[0] == XDims[axis] && in createElementwiseLinear()3199 createBroadcast(name.str() + ".broadcastW", w, XDims, axis); in createElementwiseLinear()3201 createBroadcast(name.str() + ".broadcastB", b, XDims, axis); in createElementwiseLinear()