Searched refs:RemoveTensorMutation (Results 1 – 12 of 12) sorted by relevance
11 return RemoveTensorMutation(graph_->block()); in removeTensorMutation()226 bool MutationRemover::RemoveTensorMutation(Block* block) { in RemoveTensorMutation() function in torch::jit::MutationRemover233 changed |= RemoveTensorMutation(sub_block); in RemoveTensorMutation()358 bool RemoveTensorMutation( in RemoveTensorMutation() function375 return RemoveTensorMutation(graph, [](Node* node) { in InplaceToFunctionalActivation()
50 bool RemoveTensorMutation(Block* block);73 TORCH_API bool RemoveTensorMutation(
244 RemoveTensorMutation(graph); in metalRemoveMutation()
493 RemoveTensorMutation(graph); in vulkanRemoveMutation()
1146 RemoveTensorMutation(graph, [](Node* node_to_functionalize) { in ConvertFrozenOpsToMKLDNN()
46 RemoveTensorMutation(g, [](Node* nodeToFunctionalize) { in fuseGraph()
85 If it has an inplace variant, you should add it in the lambda being passed to RemoveTensorMutation …
29 RemoveTensorMutation(graph); in optimizeAndTraceGraph()
31 RemoveTensorMutation(graph); in fuseStaticSubgraphs()
173 RemoveTensorMutation(graph); in OptimizeGraph()
2922 RemoveTensorMutation(graph, [](Node*) { return false; }); in TEST()2924 RemoveTensorMutation(graph, [](Node*) { return true; }); in TEST()
505 return RemoveTensorMutation(g); in initJITBindings()