Home
last modified time | relevance | path

Searched refs:reserve_space_size_in_bytes (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2051 size_t reserve_space_size_in_bytes = 0; in DoRnnForwardImpl() local
2055 &reserve_space_size_in_bytes /*sizeInBytes*/); in DoRnnForwardImpl()
2061 if (reserve_space_size_in_bytes > 0) { in DoRnnForwardImpl()
2063 stream, reserve_space_size_in_bytes); in DoRnnForwardImpl()
2069 stream->ThenMemZero(&reserve_space, reserve_space_size_in_bytes); in DoRnnForwardImpl()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1541 size_t reserve_space_size_in_bytes = 0; in DoRnnForwardImpl() local
1545 /*sizeInBytes=*/&reserve_space_size_in_bytes)); in DoRnnForwardImpl()
1547 if (reserve_space_size_in_bytes > 0) { in DoRnnForwardImpl()
1550 stream, reserve_space_size_in_bytes)); in DoRnnForwardImpl()