Searched refs:D_ncdhw (Results 1 – 3 of 3) sorted by relevance
41 D_ncdhw, enumerator149 case D_ncdhw: in PaddingShapeTo5dDefault()151 shape_5d[D_ncdhw] = shape[C_ncdhw]; in PaddingShapeTo5dDefault()155 shape_5d[D_ncdhw] = shape[C_ncdhw]; in PaddingShapeTo5dDefault()156 shape_5d[H_ncdhw] = shape[D_ncdhw]; in PaddingShapeTo5dDefault()160 shape_5d[D_ncdhw] = shape[C_ncdhw]; in PaddingShapeTo5dDefault()161 shape_5d[H_ncdhw] = shape[D_ncdhw]; in PaddingShapeTo5dDefault()
373 device_shape.push_back(shape[D_ncdhw]); in Ndc1hwc0DeviceShape()391 device_shape.push_back(shape[D_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()957 reshape_type_vec->push_back(D_ncdhw); in StringToAxisVector5D()1748 auto d = args.host_shape[D_ncdhw]; in Ndc1hwc0ToNcdhw()1804 auto d = args.host_shape[D_ncdhw]; in NcdhwToNdc1hwc0()1863 auto d = args.host_shape[D_ncdhw]; in NcdhwToFracZ3D()1918 auto d = args.host_shape[D_ncdhw]; in FracZ3DToNcdhw()
34 enum k3Axis : int { N_ncdhw = 0, C_ncdhw, D_ncdhw, H_ncdhw, W_ncdhw, kNcdhwDims }; enumerator43 case D_ncdhw: in PaddingRangeTo5D()45 dst_range[D_ncdhw] = ori_range[C_ncdhw]; in PaddingRangeTo5D()49 dst_range[D_ncdhw] = ori_range[C_ncdhw]; in PaddingRangeTo5D()50 dst_range[H_ncdhw] = ori_range[D_ncdhw]; in PaddingRangeTo5D()54 dst_range[D_ncdhw] = ori_range[C_ncdhw]; in PaddingRangeTo5D()55 dst_range[H_ncdhw] = ori_range[D_ncdhw]; in PaddingRangeTo5D()219 dst_range.push_back(range[D_ncdhw]); in NdhwcRange()232 dst_range.push_back(range[D_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()[all …]