Home
last modified time | relevance | path

Searched refs:availableTensorValuesAfterNode (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/benchmarks/static_runtime/
Dtest_static_module.cc1139 ranges.availableTensorValuesAfterNode(nodes[1]), in TEST()
1144 ranges.availableTensorValuesAfterNode(nodes[2]), in TEST()
1181 ranges.availableTensorValuesAfterNode(nodes[4]), in TEST()
1186 ranges.availableTensorValuesAfterNode(nodes[5]); in TEST()
/external/pytorch/torch/csrc/jit/runtime/static/
Dmemory_planner.cpp121 ranges.availableTensorValuesAfterNode(node); in assignStorageToManagedTensors()
Dimpl.h117 const std::vector<const Value*>& availableTensorValuesAfterNode(
Dimpl.cpp476 availableTensorValuesAfterNode(Node* node) const { in availableTensorValuesAfterNode() function in torch::jit::ManagedTensorRanges