Searched defs:reserve_space (Results 1 – 2 of 2) sorted by relevance
797 const Tensor* output_c_backprop, const Tensor* reserve_space, in DoBackward()1546 Tensor reserve_space = reserve_space_allocator.get_allocated_tensor(0); in MaybeAutoTune() local1690 const Tensor* reserve_space = nullptr; in ComputeImpl() local1747 const Tensor** output_c_backprop, const Tensor** reserve_space) { in ExtractBackwardInputs()
2049 DeviceMemory<uint8> reserve_space; in DoRnnForwardImpl() local