Searched refs:dest2_size (Results 1 – 1 of 1) sorted by relevance
3601 int dest2_size = 0; in DoPoolBackward() local3613 dest2_size = dims[0] * dims[1] * dims[2] * dims[3] * sizeof(float); in DoPoolBackward()3615 if (dest2_size > 0) { in DoPoolBackward()3617 auto allocated = workspace_allocator->AllocateBytes(stream, dest2_size); in DoPoolBackward()3697 int dest2_size = 0; in DoPoolBackward() local3709 dest2_size = dims[0] * dims[1] * dims[2] * dims[3] * sizeof(float); in DoPoolBackward()3711 if (dest2_size > 0) { in DoPoolBackward()3713 auto allocated = workspace_allocator->AllocateBytes(stream, dest2_size); in DoPoolBackward()3832 int dest2_size = 0; in DoNormalizeBackwardWithDimensions() local3844 dest2_size = in DoNormalizeBackwardWithDimensions()[all …]