Searched refs:nodeFreesManagedTensors (Results 1 – 4 of 4) sorted by relevance
1135 EXPECT_FALSE(ranges.nodeFreesManagedTensors(nodes[0])); in TEST()1137 EXPECT_TRUE(ranges.nodeFreesManagedTensors(nodes[1])); in TEST()1142 EXPECT_TRUE(ranges.nodeFreesManagedTensors(nodes[2])); in TEST()1176 EXPECT_FALSE(ranges.nodeFreesManagedTensors(nodes[i])); in TEST()1179 EXPECT_TRUE(ranges.nodeFreesManagedTensors(nodes[4])); in TEST()1184 EXPECT_TRUE(ranges.nodeFreesManagedTensors(nodes[5])); in TEST()
119 if (ranges.nodeFreesManagedTensors(node)) { in assignStorageToManagedTensors()
116 bool nodeFreesManagedTensors(Node* node) const;
470 bool ManagedTensorRanges::nodeFreesManagedTensors(Node* node) const { in nodeFreesManagedTensors() function in torch::jit::ManagedTensorRanges