Searched defs:scratch_memory (Results 1 – 4 of 4) sorted by relevance
627 DeviceMemory<uint8> scratch_memory) { in DoCtcLoss()
668 DeviceMemory<uint8> scratch_memory; in ThenConvolveWithAlgorithm() local710 DeviceMemory<uint8> scratch_memory; in ThenConvolveWithAlgorithm() local752 DeviceMemory<uint8> scratch_memory; in ThenConvolveWithAlgorithm() local794 DeviceMemory<uint8> scratch_memory; in ThenConvolveWithAlgorithm() local836 DeviceMemory<uint8> scratch_memory; in ThenConvolveWithAlgorithm() local981 DeviceMemory<uint8> scratch_memory; in ThenConvolveBackwardDataWithAlgorithm() local1026 DeviceMemory<uint8> scratch_memory; in ThenConvolveBackwardDataWithAlgorithm() local1071 DeviceMemory<uint8> scratch_memory; in ThenConvolveBackwardDataWithAlgorithm() local1116 DeviceMemory<uint8> scratch_memory; in ThenConvolveBackwardFilterWithAlgorithm() local1161 DeviceMemory<uint8> scratch_memory; in ThenConvolveBackwardFilterWithAlgorithm() local[all …]
2036 DeviceMemory<uint8> scratch_memory) { in DoCtcLossImpl()2902 DeviceMemory<uint8>* scratch_memory) { in DoPrepareForConvolution()2959 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve()3924 ScratchAllocator* scratch_allocator, DeviceMemory<uint8>* scratch_memory) { in DoPrepareForCtcLoss()3972 DeviceMemoryBase grads_data, DeviceMemory<uint8> scratch_memory) { in DoCtcLoss()
2645 DeviceMemory<uint8>* scratch_memory) { in DoPrepareForConvolution()2809 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve()