Searched refs:workspace_size_in_bytes (Results 1 – 2 of 2) sorted by relevance
1973 size_t workspace_size_in_bytes = 0; in CreateRnnWorkspace() local1977 &workspace_size_in_bytes /*sizeInBytes*/); in CreateRnnWorkspace()1983 if (workspace_size_in_bytes > 0) { in CreateRnnWorkspace()1985 workspace_allocator->AllocateBytes(stream, workspace_size_in_bytes); in CreateRnnWorkspace()1991 stream->ThenMemZero(workspace, workspace_size_in_bytes); in CreateRnnWorkspace()3578 size_t workspace_size_in_bytes = 0; in DoPoolBackward() local3580 &workspace_size_in_bytes); in DoPoolBackward()3590 if (workspace_size_in_bytes > 0) { in DoPoolBackward()3593 workspace_allocator->AllocateBytes(stream, workspace_size_in_bytes); in DoPoolBackward()3630 workspace.opaque(), workspace_size_in_bytes); in DoPoolBackward()[all …]
1491 size_t workspace_size_in_bytes = 0; in CreateRnnWorkspace() local1495 /*sizeInBytes=*/&workspace_size_in_bytes)); in CreateRnnWorkspace()1497 if (workspace_size_in_bytes == 0) { in CreateRnnWorkspace()1500 return workspace_allocator->AllocateBytes(stream, workspace_size_in_bytes); in CreateRnnWorkspace()