Searched refs:output_allocator (Results 1 – 1 of 1) sorted by relevance
327 Allocator* output_allocator, bool allocate_xla_tensors, se::Stream* stream, in GetOrCreateTensorForOutput() argument372 MakeTensor(output_dtype, output_shape, output_buffer, output_allocator); in GetOrCreateTensorForOutput()