Home
last modified time | relevance | path

Searched defs:reserve_space (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc878 const Tensor* output_c_backprop, const Tensor* reserve_space, in DoBackward()
1775 Tensor reserve_space = reserve_space_allocator.get_allocated_tensor(0); in MaybeAutotune() local
1931 const Tensor* reserve_space = nullptr; in ComputeImpl() local
1988 const Tensor** output_c_backprop, const Tensor** reserve_space) { in ExtractBackwardInputs()
Dfused_batch_norm_op.cc1133 const Tensor& reserve_space = context->input(5); in operator ()() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1905 DeviceMemory<uint8> reserve_space; in DoRnnForwardImpl() local
4631 DeviceMemory<uint8> reserve_space; in DoBatchNormalizationForwardImpl() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2181 DeviceMemory<uint8> reserve_space; in DoRnnForwardImpl() local