Searched refs:ShapeSize (Results 1 – 8 of 8) sorted by relevance
215 auto shape_size = abstract::ShapeSize(host_shape); in SyncDeviceToHost()295 auto shape_size = abstract::ShapeSize(host_shape); in SyncDeviceToHostAndConvertFormatBasedOnTransData()341 auto shape_size = abstract::ShapeSize(host_shape); in SyncDeviceToHostAndConvertFormat()382 auto shape_size = abstract::ShapeSize(host_shape); in SyncHostToDevice()449 auto shape_size = abstract::ShapeSize(host_shape); in ConvertFormatAndSyncHostToDevice()
55 size_t ShapeSize(const std::vector<size_t> &shape);
307 size_t ShapeSize(const std::vector<size_t> &shape) { in ShapeSize() function
1076 *total_size = abstract::ShapeSize(args.device_shape) * (*size); in CheckArgs()1296 auto total_size = abstract::ShapeSize(args.device_shape) * size; in FracZToNchw()1441 auto dst_size = abstract::ShapeSize(args.device_shape) * size; in NchwToFracNz()1503 auto dst_size = abstract::ShapeSize(args.device_shape) * size; in FracNzToNchw()1559 auto total_size = abstract::ShapeSize(args.device_shape) * size; in NchwToNc1hwc0()1613 auto total_size = abstract::ShapeSize(args.device_shape) * size; in Nc1hwc0ToNchw()1741 auto total_size = abstract::ShapeSize(args.device_shape) * size; in Ndc1hwc0ToNcdhw()1796 auto total_size = abstract::ShapeSize(args.device_shape) * size; in NcdhwToNdc1hwc0()1855 auto total_size = abstract::ShapeSize(args.device_shape) * size; in NcdhwToFracZ3D()1911 auto total_size = abstract::ShapeSize(args.device_shape) * size; in FracZ3DToNcdhw()
313 size = abstract::ShapeSize(shape_tmp) * abstract::TypeIdSize(tensor->data_type()); in UpdateGraphInputAbstract()
192 int64_t ShapeSize(const std::vector<int64_t> &shape) { in ShapeSize() function216 temp_shape = ShapeSize(max_shape) > ShapeSize(temp_shape) ? max_shape : temp_shape; in CreateNodeOutputTensor()
486 size = abstract::ShapeSize(shape_tmp) * abstract::TypeIdSize(tensor->data_type()); in LoadInputData()
655 auto size = abstract::ShapeSize(shape_tmp) * abstract::TypeIdSize(data_type()); in data_sync()