Searched refs:kCubeSize (Results 1 – 10 of 10) sorted by relevance
301 const size_t cout16 = ((shape[kN] + kCubeSize - 1) / kCubeSize) * kCubeSize; in FracZDeviceShape()302 const size_t cin16 = ((shape[kC] + kCubeSize - 1) / kCubeSize) * kCubeSize; in FracZDeviceShape()303 device_shape.push_back(shape[kH] * shape[kW] * cin16 / kCubeSize); in FracZDeviceShape()304 device_shape.push_back(cout16 / kCubeSize); in FracZDeviceShape()305 device_shape.push_back(kCubeSize); in FracZDeviceShape()306 device_shape.push_back(kCubeSize); in FracZDeviceShape()314 auto tmp = SizeToLong(kCubeSize); in FracZDeviceDynamicShape()338 const size_t C1 = (shape[kC] + kCubeSize - 1) / kCubeSize; in Nc1hwc0DeviceShape()339 const size_t C0 = kCubeSize; in Nc1hwc0DeviceShape()353 auto tmp = SizeToLong(kCubeSize); in Nc1hwc0DeviceDynamicShape()[all …]
34 if (shape.size() == kShapeSize1 && shape[0] % kCubeSize != 0) { in CheckShapeDimInfo()38 … (shape[shape.size() - 1] % kCubeSize != 0 || shape[shape.size() - kShapeSize2] % kCubeSize != 0)); in CheckShapeDimInfo()
33 if (shape.size() == 1 && shape[0] % kCubeSize != 0) { in CheckShapeDimInfo()37 … (shape[shape.size() - 1] % kCubeSize != 0 || shape[shape.size() - kShape2Dim] % kCubeSize != 0)); in CheckShapeDimInfo()
59 kDimMultiNum * (((origin_input9_shape[kDim1] + kCubeSize - 1) / kCubeSize) * kCubeSize)}; in CreateTLoopNode()87 … SizeToLong((origin_output2_shape[kDim2] + kCubeSize - 1) / kCubeSize * kCubeSize), in CreateTLoopNode()88 … SizeToLong((origin_output3_shape[kDim1] + kCubeSize - 1) / kCubeSize * kCubeSize)}), in CreateTLoopNode()544 if (hidden_size % kCubeSize != 0) { in Process()
52 if (pad_shape[N_nchw] % kCubeSize != 0 || pad_shape[C_nchw] % kCubeSize != 0) { in GetKernelFormat()
54 if (input_origin_shape.size() == 1 && input_origin_shape[0] % kCubeSize == 0) { in CheckFormatForConsistency()
77 …e.size() == 1 && (device_shape[0] == 1 || static_cast<size_t>(device_shape[0]) % kCubeSize == 0)) { in FractalNzAbstractShape()
497 const size_t kCubeSize = 16; variable
37 …_check = origin_shape.size() > 1 || (origin_shape.size() == 1 && origin_shape[0] % kCubeSize != 0); in NeedInsertTransData()
110 if (shape[shape.size() - i] != 1 && shape[shape.size() - i] % kCubeSize != 0) { in CanConvertDefaultShapeToNZ()