Home
last modified time | relevance | path

Searched refs:functional_node (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/csrc/jit/passes/
Dcreate_functional_graphs.cpp122 for (Node* functional_node : functional_graph_nodes) { in CreateFunctionalGraphsImpl()
123 if (!inlineIfTooSmall(functional_node)) { in CreateFunctionalGraphsImpl()
124 ConstantPooling(functional_node->g(attr::Subgraph)); in CreateFunctionalGraphsImpl()
/external/tensorflow/tensorflow/compiler/jit/
Dcompilability_check_util_test.cc193 Node* functional_node = ops::UnaryOp(kUncompilableFunctionName, const0, in TEST_F() local
198 EXPECT_FALSE(checker_->IsCompilableNode(*functional_node, flib_runtime)); in TEST_F()
200 checker_->FindUncompilableNodes(*functional_node, flib_runtime); in TEST_F()