Searched refs:N_ncdhw (Results 1 – 3 of 3) sorted by relevance
39 N_ncdhw = 0, enumerator144 case N_ncdhw: in PaddingShapeTo5dDefault()147 shape_5d[C_ncdhw] = shape[N_ncdhw]; in PaddingShapeTo5dDefault()150 shape_5d[C_ncdhw] = shape[N_ncdhw]; in PaddingShapeTo5dDefault()154 shape_5d[C_ncdhw] = shape[N_ncdhw]; in PaddingShapeTo5dDefault()159 shape_5d[C_ncdhw] = shape[N_ncdhw]; in PaddingShapeTo5dDefault()
372 device_shape.push_back(shape[N_ncdhw]); in Ndc1hwc0DeviceShape()390 device_shape.push_back(shape[N_ncdhw]); in Ndc1hwc0DeviceDynamicShape()951 reshape_type_vec->push_back(N_ncdhw); in StringToAxisVector5D()1746 auto n = args.host_shape[N_ncdhw]; in Ndc1hwc0ToNcdhw()1802 auto n = args.host_shape[N_ncdhw]; in NcdhwToNdc1hwc0()1861 auto n = args.host_shape[N_ncdhw]; in NcdhwToFracZ3D()1916 auto n = args.host_shape[N_ncdhw]; in FracZ3DToNcdhw()
34 enum k3Axis : int { N_ncdhw = 0, C_ncdhw, D_ncdhw, H_ncdhw, W_ncdhw, kNcdhwDims }; enumerator38 case N_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()48 dst_range[C_ncdhw] = ori_range[N_ncdhw]; in PaddingRangeTo5D()53 dst_range[C_ncdhw] = ori_range[N_ncdhw]; in PaddingRangeTo5D()218 dst_range.push_back(range[N_ncdhw]); in NdhwcRange()231 dst_range.push_back(range[N_ncdhw]); in Ndc1hwc0Range()245 const std::pair<int64_t, int64_t> n1 = {(range[N_ncdhw].first + k16 - 1) / k16, in FracZ3DRange()246 (range[N_ncdhw].second + k16 - 1) / k16}; in FracZ3DRange()