Home
last modified time | relevance | path

Searched refs:fn_body (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/python/autograph/pyct/static_analysis/
Dreaching_definitions_test.py99 fn_body = node.body
101 self.assertHasDefs(fn_body[0].targets[0], 1)
102 self.assertHasDefs(fn_body[1].test, 1)
103 self.assertHasDefs(fn_body[1].body[0].targets[0], 1)
104 self.assertHasDefs(fn_body[2].value, 2)
106 self.assertHasDefinedIn(fn_body[1], ('a', 'b'))
120 fn_body = node.body
122 self.assertHasDefinedIn(fn_body[1], ('a', 'b'))
123 self.assertHasDefinedIn(fn_body[1].body[0], ('a', 'b'))
138 fn_body = node.body
[all …]
Dliveness_test.py89 fn_body = node.body
91 self.assertHasLiveOut(fn_body[0], 'x')
92 self.assertHasLiveOut(fn_body[0].body[0], 'x')
106 fn_body = node.body
108 self.assertHasLiveOut(fn_body[0], 'x')
109 self.assertHasLiveOut(fn_body[0].body[0], 'x')
110 self.assertHasLiveOut(fn_body[0].body[0].handlers[0].body[0], 'x')
122 fn_body = node.body
124 self.assertHasLiveOut(fn_body[0], ('a', 'x'))
125 self.assertHasLiveOut(fn_body[1], 'x')
[all …]
Dtype_inference_test.py97 fn_body = node.body
101 anno.hasanno(fn_body[0].value.elts[0], anno.Static.TYPES))
103 anno.hasanno(fn_body[0].value.elts[1], anno.Static.TYPES))
135 fn_body = node.body
137 self.assertTypes(fn_body[0].value.elts[0], 'a_type')
138 self.assertTypes(fn_body[0].value.elts[1], 'b_type')
172 fn_body = node.body
174 self.assertTypes(fn_body[0].body[0].value, (('x_type', 'y_type'),))
175 self.assertTypes(fn_body[0].body[0].value.elts[0], 'x_type')
176 self.assertTypes(fn_body[0].body[0].value.elts[1], 'y_type')
[all …]
Dreaching_definitions_py3_test.py43 fn_body = node.body
45 self.assertHasDefs(fn_body[2].test, 1)
46 self.assertHasDefs(fn_body[2].body[0].targets[0], 1)
47 self.assertHasDefs(fn_body[3].value.elts[0], 1)
48 self.assertHasDefs(fn_body[3].value.elts[1], 2)
50 self.assertSameDef(fn_body[2].test, fn_body[3].value.elts[0])
52 self.assertHasDefinedIn(fn_body[2], ('a', 'b'))
/external/tensorflow/tensorflow/c/
Dc_api_function.cc51 const TF_Graph* fn_body, const char* fn_name, int ninputs, in ProcessInputs() argument
54 TF_EXCLUSIVE_LOCKS_REQUIRED(fn_body->mu) { in ProcessInputs()
61 fn_body->graph.IsValidOutputTensor(node, idx), in ProcessInputs()
87 Status ProcessOutputs(const TF_Graph* fn_body, const char* fn_name, in ProcessOutputs() argument
90 TF_EXCLUSIVE_LOCKS_REQUIRED(fn_body->mu) { in ProcessOutputs()
96 fn_body->graph.IsValidOutputTensor(node, idx), in ProcessOutputs()
110 const TF_Graph* fn_body, const char* fn_name, int num_opers, in ComputeBodyNodes() argument
114 TF_EXCLUSIVE_LOCKS_REQUIRED(fn_body->mu) { in ComputeBodyNodes()
116 for (const Node* node : fn_body->graph.op_nodes()) { in ComputeBodyNodes()
151 const TF_Graph* fn_body, const char* fn_name, in TF_GraphToFunctionWithControlOutputs() argument
[all …]
Dc_api.h1107 const TF_Graph* fn_body, const char* fn_name,
1127 const TF_Graph* fn_body, const char* fn_name,
/external/tensorflow/tensorflow/python/autograph/pyct/
Dtransformer_test.py92 fn_body = node.body
93 outer_while_body = fn_body[1].body
94 self.assertSameAnno(fn_body[0], outer_while_body[0], 'cond_state')
95 self.assertDifferentAnno(fn_body[0], outer_while_body[0], 'loop_state')
146 fn_body = node.body
147 outer_if_body = fn_body[1].body
148 self.assertDifferentAnno(fn_body[0], outer_if_body[0], 'cond_state')
/external/tensorflow/tensorflow/core/framework/
Dgraph_to_functiondef.h31 Status GraphToFunctionDef(const Graph& fn_body, const string& fn_name,
Dgraph_to_functiondef.cc367 Status GraphToFunctionDef(const Graph& fn_body, const string& fn_name, in GraphToFunctionDef() argument
/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.h207 const TF_Graph* fn_body, const char* fn_name, bool append_hash_to_fn_name,
Dtf_session_helper.cc579 const TF_Graph* fn_body, const char* fn_name, bool append_hash_to_fn_name, in TF_GraphToFunction_wrapper() argument
613 fn_body, fn_name, append_hash_to_fn_name, nopers, opers_array, in TF_GraphToFunction_wrapper()
Dtf_session_wrapper.cc190 [](const TF_Graph* fn_body, const char* fn_name, in PYBIND11_MODULE()
213 fn_body, fn_name, append_hash_to_fn_name, in PYBIND11_MODULE()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dllvm_ir_runtime.cc110 llvm::BasicBlock* fn_body = llvm::BasicBlock::Create(*context, "body", fn); in RewriteCalls() local
111 llvm::IRBuilder<> b(fn_body); in RewriteCalls()
/external/tensorflow/tensorflow/core/kernels/data/
Dcaptured_function.cc189 const FunctionBody* fn_body = in CreateShortCircuitInfo() local
191 indices.resize(fn_body->ret_nodes.size()); in CreateShortCircuitInfo()
193 for (size_t i = 0; i < fn_body->ret_nodes.size(); ++i) { in CreateShortCircuitInfo()
194 Node* ret_node = fn_body->ret_nodes[i]; in CreateShortCircuitInfo()
/external/rust/cxx/macro/src/
Dexpand.rs662 let fn_body = quote_spanned!(efn.semi_token.span=> { in expand_cxx_function_shim() localVariable
674 #visibility #unsafety #fn_token #ident #generics #arg_list #ret #fn_body in expand_cxx_function_shim()
704 #visibility #unsafety #fn_token #ident #arg_list #ret #fn_body in expand_cxx_function_shim()