Home
last modified time | relevance | path

Searched defs:workspace_size_in_bytes (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc1861 size_t workspace_size_in_bytes = 0; in CreateRnnWorkspace() local
1882 size_t workspace_size_in_bytes = 0; in CreateBatchNormForwardWorkspace() local
1906 size_t workspace_size_in_bytes = 0; in CreateBatchNormBackwardWorkspace() local
1973 size_t workspace_size_in_bytes = 0; in DoRnnForwardImpl() local
2185 size_t workspace_size_in_bytes = 0; in DoRnnBackwardImpl() local
5847 size_t workspace_size_in_bytes = 0; in DoPrepareForCtcLoss() local
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc2108 size_t workspace_size_in_bytes = 0; in CreateRnnWorkspace() local
2458 size_t workspace_size_in_bytes = 0; in DoPrepareForCtcLoss() local
4203 size_t workspace_size_in_bytes = 0; in DoPoolBackward() local
4347 size_t workspace_size_in_bytes = 0; in DoNormalizeBackwardWithDimensions() local