| /external/tensorflow/tensorflow/python/autograph/converters/ |
| D | directives.py | 55 def _map_args(call_node, function): argument
67 args = call_node.args
68 kwds = {kwd.arg: kwd.value for kwd in call_node.keywords}
93 def _process_symbol_directive(self, call_node, directive): argument
94 if len(call_node.args) < 1:
97 target = call_node.args[0]
100 def_.directives[directive] = _map_args(call_node, directive)
101 return call_node
103 def _process_statement_directive(self, call_node, directive): argument
113 node_anno[directive] = _map_args(call_node, directive)
[all …]
|
| /external/mesa3d/src/compiler/glsl/ |
| D | ir_function_detect_recursion.cpp | 131 struct call_node : public exec_node { struct
210 call_node *node = new(mem_ctx) call_node; in visit_enter()
216 node = new(mem_ctx) call_node; in visit_enter()
233 foreach_in_list_safe(call_node, node, list) { in destroy_links()
255 struct call_node *n = (struct call_node *) f->callers.pop_head(); in remove_unlinked_functions()
260 struct call_node *n = (struct call_node *) f->callees.pop_head(); in remove_unlinked_functions()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | build_xla_ops_pass_test.cc | 91 NodeDef call_node; in MakeXlaCompiledKernel() local
92 call_node.set_name(node_name); in MakeXlaCompiledKernel()
93 call_node.set_op(callee_name); in MakeXlaCompiledKernel()
94 AddNodeAttr(kXlaCompiledKernelAttr, true, &call_node); in MakeXlaCompiledKernel()
95 AddNodeAttr(kXlaNumConstantArgsAttr, num_constant_args, &call_node); in MakeXlaCompiledKernel()
96 AddNodeAttr(kXlaNumResourceArgsAttr, num_resource_args, &call_node); in MakeXlaCompiledKernel()
98 *result = graph->AddNode(call_node, &s); in MakeXlaCompiledKernel()
|
| D | resource_operation_safety_analysis_test.cc | 254 NodeDef call_node; in MakeCall() local
255 call_node.set_name(node_name); in MakeCall()
256 call_node.set_op(callee_name); in MakeCall()
257 return graph->AddNode(call_node, status); in MakeCall()
|
| D | extract_outside_compilation_pass.cc | 311 const Node* call_node, const std::map<string, int>& host_compute_core, in BuildXlaHostComputeNodeDef() argument
315 call_node->attrs(), "_outside_compilation_subgraph", &original_oc_name)); in BuildXlaHostComputeNodeDef()
325 for (const auto& attr : call_node->attrs()) { in BuildXlaHostComputeNodeDef()
358 TF_RETURN_IF_ERROR(GetNodeAttr(call_node->attrs(), "Tinputs", &input_dtypes)); in BuildXlaHostComputeNodeDef()
360 for (auto e : call_node->in_edges()) { in BuildXlaHostComputeNodeDef()
381 Graph* g, Node* call_node, const std::map<string, int>& host_compute_core, in ReplaceOutsideCompilationCallNode() argument
386 BuildXlaHostComputeNodeDef(call_node, host_compute_core, cluster_deps)); in ReplaceOutsideCompilationCallNode()
388 ReplaceNode(g, call_node, node_def)); in ReplaceOutsideCompilationCallNode()
1786 Node* call_node = host_graph.AddNode(call_def, &s); in BuildHostGraphForFuncCallNode() local
1788 host_graph.AddEdge(key_placeholder, 0, call_node, 0); in BuildHostGraphForFuncCallNode()
|
| D | mark_for_compilation_pass_test.cc | 964 NodeDef call_node; in TEST() local
965 call_node.set_name("fn_call"); in TEST()
966 call_node.set_op("Stateful_func"); in TEST()
968 Node* call = root.graph()->AddNode(call_node, &status); in TEST()
|
| D | extract_outside_compilation_pass_test.cc | 796 Node *call_node = node_name_index["oc_call_fn"]; in TEST_F() local
797 EXPECT_NE(call_node, nullptr); in TEST_F()
|
| D | encapsulate_subgraphs_pass_test.cc | 386 Node* KeyPlaceholder(const string& call_node, in KeyPlaceholder() argument
389 NodeBuilder node_builder(absl::StrCat(call_node, "_key_placeholder"), in KeyPlaceholder()
395 .WithAttr("_host_compute_call_node", call_node) in KeyPlaceholder()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 1846 Node* call_node, const FunctionBody& fbody) { in FindArgsToLiftForCallNode() argument
1849 std::vector<DataType> dtypes(call_node->input_types().begin(), in FindArgsToLiftForCallNode()
1850 call_node->input_types().end()); in FindArgsToLiftForCallNode()
2266 Graph* g, Node* call_node, FunctionLibraryRuntime* flr, in LiftOutsideCompilationOnlyArgsFromCallNode() argument
2272 if (fld->Contains(call_node->type_string())) { in LiftOutsideCompilationOnlyArgsFromCallNode()
2273 func.set_name(call_node->type_string()); in LiftOutsideCompilationOnlyArgsFromCallNode()
2274 *func.mutable_attr() = call_node->def().attr(); in LiftOutsideCompilationOnlyArgsFromCallNode()
2275 } else if (call_node->IsPartitionedCall()) { in LiftOutsideCompilationOnlyArgsFromCallNode()
2276 TF_RETURN_IF_ERROR(GetNodeAttr(call_node->def(), "f", &func)); in LiftOutsideCompilationOnlyArgsFromCallNode()
2278 TF_RET_CHECK(call_node->type_string() == in LiftOutsideCompilationOnlyArgsFromCallNode()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | resource_util.cc | 130 const Node& call_node, in UpdateResourceUsageFromFunctionBodyAnalysis() argument
157 TF_RETURN_IF_ERROR(call_node.input_edge(index, &e)); in UpdateResourceUsageFromFunctionBodyAnalysis()
172 TF_ASSIGN_OR_RETURN(outs, OutputEdgesByIndex(call_node, index)); in UpdateResourceUsageFromFunctionBodyAnalysis()
|
| /external/tensorflow/tensorflow/python/autograph/pyct/static_analysis/ |
| D | activity_test.py | 217 call_node = node.body[2].value
221 anno.getanno(call_node, NodeAnno.ARGS_SCOPE), ('a', 'b'), ())
234 call_node = node.body[1].value
236 anno.getanno(call_node, NodeAnno.ARGS_SCOPE), ('a', 'a.b', 'a.c'), ())
250 call_node = node.body[2].value
252 anno.getanno(call_node, NodeAnno.ARGS_SCOPE),
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | host_compute_ops.cc | 274 for (Node* call_node : call_nodes) { in LowerFunctionalOps()
276 call_node, g, flib_def, /*keep_caller_fetchable=*/false)); in LowerFunctionalOps()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 2483 NodeDef call_node = *node; in SimplifyCase() local
2484 call_node.set_op("PartitionedCall"); in SimplifyCase()
2485 call_node.clear_input(); in SimplifyCase()
2487 call_node.add_input(node->input(i)); in SimplifyCase()
2489 auto* new_func = (*call_node.mutable_attr())["f"].mutable_func(); in SimplifyCase()
2497 (*call_node.mutable_attr())["_output_shapes"] in SimplifyCase()
2504 call_node.mutable_attr()->erase("output_shapes"); in SimplifyCase()
2505 call_node.mutable_attr()->erase("branches"); in SimplifyCase()
2507 *node = std::move(call_node); in SimplifyCase()
|
| /external/selinux/libsepol/cil/src/ |
| D | cil_resolve_ast.c | 2688 void cil_print_recursive_call(struct cil_tree_node *call_node, struct cil_tree_node *terminating_no… in cil_print_recursive_call() argument
2696 for (curr = call_node; curr != terminating_node; curr = curr->parent) { in cil_print_recursive_call()
2698 if (curr != call_node) { in cil_print_recursive_call()
2723 int cil_check_recursive_call(struct cil_tree_node *call_node, struct cil_tree_node *macro_node) in cil_check_recursive_call() argument
2729 for (curr = call_node; curr != NULL; curr = curr->parent) { in cil_check_recursive_call()
2731 if (curr == call_node) { in cil_check_recursive_call()
2749 cil_print_recursive_call(call_node, curr); in cil_check_recursive_call()
|