Searched defs:reserve_space (Results 1 – 4 of 4) sorted by relevance
856 const Tensor* output_c_backprop, const Tensor* reserve_space, in DoBackward()1733 Tensor reserve_space = reserve_space_allocator.get_allocated_tensor(0); in MaybeAutoTune() local1889 const Tensor* reserve_space = nullptr; in ComputeImpl() local1946 const Tensor** output_c_backprop, const Tensor** reserve_space) { in ExtractBackwardInputs()
1112 const Tensor& reserve_space = context->input(5); in operator ()() local
3520 DeviceMemory<uint8> reserve_space; in DoBatchNormalizationForwardImpl() local
2130 DeviceMemory<uint8> reserve_space; in DoRnnForwardImpl() local