Searched refs:managed_output_tensor_values (Results 1 – 3 of 3) sorted by relevance
183 const c10::FastSet<const Value*>& managed_output_tensor_values) { in assignStorageToOutputTensors() argument189 if (!setIncludes(managed_output_tensor_values, val) || in assignStorageToOutputTensors()209 const auto& managed_output_tensor_values = in MemoryPlanner() local210 block_info.managed_output_tensor_values(); in MemoryPlanner()229 setIncludes(managed_output_tensor_values, out_v)) || in MemoryPlanner()271 block_runner, managed_output_tensor_values); in MemoryPlanner()
374 const auto& managed_output_tensor_values() const { in managed_output_tensor_values() function
1969 const auto& managed_outputs = block_info_.managed_output_tensor_values(); in isManagedOutputTensorValue()