Searched refs:managed_tensor_values (Results 1 – 4 of 4) sorted by relevance
28 const c10::FastSet<const Value*>& managed_tensor_values) { in tensorValueToTensor() argument35 if (managed_tensor_values.find(output) == managed_tensor_values.end()) { in tensorValueToTensor()208 const auto& managed_tensor_values = block_info.managed_tensor_values(); in MemoryPlanner() local220 const bool in_managed_tensors = setIncludes(managed_tensor_values, out_v); in MemoryPlanner()383 const auto& managed_tensor_values = block_info.managed_tensor_values(); in StandardMemoryPlanner() local386 tensorValueToTensor(block_runner->nodes(), managed_tensor_values); in StandardMemoryPlanner()
110 const c10::FastSet<const Value*>& managed_tensor_values);378 const auto& managed_tensor_values() const { in managed_tensor_values() function
400 const c10::FastSet<const Value*>& managed_tensor_values) { in ManagedTensorRanges() argument457 for (auto* managed_tensor : managed_tensor_values) { in ManagedTensorRanges()
1346 FastSet<const Value*> managed_tensor_values; in testAssignStorageToManagedTensors() local1353 managed_tensor_values.insert(vmap_it->second); in testAssignStorageToManagedTensors()1356 ASSERT_EQ(managed_tensor_values.size(), tensor_value_to_tensor.size()); in testAssignStorageToManagedTensors()1360 ManagedTensorRanges(*graph->block(), alias_db, managed_tensor_values); in testAssignStorageToManagedTensors()