Searched refs:output_allocator (Results 1 – 1 of 1) sorted by relevance
307 se::DeviceMemoryBase output_buffer, Allocator* output_allocator) { in GetOrCreateTensorForOutput() argument327 output_allocator); in GetOrCreateTensorForOutput()