Searched refs:C_ncdhw (Results 1 – 3 of 3) sorted by relevance
40 C_ncdhw, enumerator146 case C_ncdhw: in PaddingShapeTo5dDefault()147 shape_5d[C_ncdhw] = shape[N_ncdhw]; in PaddingShapeTo5dDefault()150 shape_5d[C_ncdhw] = shape[N_ncdhw]; in PaddingShapeTo5dDefault()151 shape_5d[D_ncdhw] = shape[C_ncdhw]; in PaddingShapeTo5dDefault()154 shape_5d[C_ncdhw] = shape[N_ncdhw]; in PaddingShapeTo5dDefault()155 shape_5d[D_ncdhw] = shape[C_ncdhw]; in PaddingShapeTo5dDefault()159 shape_5d[C_ncdhw] = shape[N_ncdhw]; in PaddingShapeTo5dDefault()160 shape_5d[D_ncdhw] = shape[C_ncdhw]; in PaddingShapeTo5dDefault()
421 if (HasShapeDynamic({shape[C_ncdhw], shape[D_ncdhw], shape[H_ncdhw], shape[W_ncdhw]})) { in Fracz3DDeviceDynamicShape()954 reshape_type_vec->push_back(C_ncdhw); in StringToAxisVector5D()1747 auto c = args.host_shape[C_ncdhw]; in Ndc1hwc0ToNcdhw()1803 auto c = args.host_shape[C_ncdhw]; in NcdhwToNdc1hwc0()1862 auto c = args.host_shape[C_ncdhw]; in NcdhwToFracZ3D()1917 auto c = args.host_shape[C_ncdhw]; in FracZ3DToNcdhw()
34 enum k3Axis : int { N_ncdhw = 0, C_ncdhw, D_ncdhw, H_ncdhw, W_ncdhw, kNcdhwDims }; enumerator40 case C_ncdhw: in PaddingRangeTo5D()41 dst_range[C_ncdhw] = ori_range[N_ncdhw]; in PaddingRangeTo5D()44 dst_range[C_ncdhw] = ori_range[N_ncdhw]; in PaddingRangeTo5D()45 dst_range[D_ncdhw] = ori_range[C_ncdhw]; in PaddingRangeTo5D()48 dst_range[C_ncdhw] = ori_range[N_ncdhw]; in PaddingRangeTo5D()49 dst_range[D_ncdhw] = ori_range[C_ncdhw]; in PaddingRangeTo5D()53 dst_range[C_ncdhw] = ori_range[N_ncdhw]; in PaddingRangeTo5D()54 dst_range[D_ncdhw] = ori_range[C_ncdhw]; in PaddingRangeTo5D()222 dst_range.push_back(range[C_ncdhw]); in NdhwcRange()[all …]