Searched defs:reserve_space (Results 1 – 4 of 4) sorted by relevance
879 const Tensor* output_c_backprop, const Tensor* reserve_space, in DoBackward()1777 Tensor reserve_space = reserve_space_allocator.get_allocated_tensor(0); in MaybeAutotune() local1933 const Tensor* reserve_space = nullptr; in ComputeImpl() local1990 const Tensor** output_c_backprop, const Tensor** reserve_space) { in ExtractBackwardInputs()
1135 const Tensor& reserve_space = context->input(5); in operator ()() local
1965 DeviceMemory<uint8> reserve_space; in DoRnnForwardImpl() local2036 DeviceMemory<uint8> reserve_space; in DoRnnForwardImpl() local5482 DeviceMemory<uint8> reserve_space; in DoBatchNormalizationForwardImpl() local
2184 DeviceMemory<uint8> reserve_space; in DoRnnForwardImpl() local