Searched refs:W_ncdhw (Results 1 – 3 of 3) sorted by relevance
34 enum k3Axis : int { N_ncdhw = 0, C_ncdhw, D_ncdhw, H_ncdhw, W_ncdhw, kNcdhwDims }; enumerator52 case W_ncdhw: in PaddingRangeTo5D()56 dst_range[W_ncdhw] = ori_range[H_ncdhw]; in PaddingRangeTo5D()221 dst_range.push_back(range[W_ncdhw]); in NdhwcRange()235 dst_range.push_back(range[W_ncdhw]); in Ndc1hwc0Range()247 …const int64_t r1_0 = range[D_ncdhw].first * c1.first * range[H_ncdhw].first * range[W_ncdhw].first; in FracZ3DRange()248 …const int64_t r1_1 = range[D_ncdhw].second * c1.second * range[H_ncdhw].second * range[W_ncdhw].se… in FracZ3DRange()
43 W_ncdhw, enumerator158 case W_ncdhw: in PaddingShapeTo5dDefault()162 shape_5d[W_ncdhw] = shape[H_ncdhw]; in PaddingShapeTo5dDefault()
376 device_shape.push_back(shape[W_ncdhw]); in Ndc1hwc0DeviceShape()394 device_shape.push_back(shape[W_ncdhw]); in Ndc1hwc0DeviceDynamicShape()407 device_shape.push_back(shape[D_ncdhw] * C1 * shape[H_ncdhw] * shape[W_ncdhw]); in Fracz3DDeviceShape()421 if (HasShapeDynamic({shape[C_ncdhw], shape[D_ncdhw], shape[H_ncdhw], shape[W_ncdhw]})) { in Fracz3DDeviceDynamicShape()425 device_shape.push_back(shape[D_ncdhw] * C1 * shape[H_ncdhw] * shape[W_ncdhw]); in Fracz3DDeviceDynamicShape()963 reshape_type_vec->push_back(W_ncdhw); in StringToAxisVector5D()1750 auto w = args.host_shape[W_ncdhw]; in Ndc1hwc0ToNcdhw()1806 auto w = args.host_shape[W_ncdhw]; in NcdhwToNdc1hwc0()1865 auto w = args.host_shape[W_ncdhw]; in NcdhwToFracZ3D()1920 auto w = args.host_shape[W_ncdhw]; in FracZ3DToNcdhw()