Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/autograph/pyct/static_analysis/
Dreaching_definitions_test.py89 fn_body = node.body
91 self.assertHasDefs(fn_body[0].targets[0], 1)
92 self.assertHasDefs(fn_body[1].test, 1)
93 self.assertHasDefs(fn_body[1].body[0].targets[0], 1)
94 self.assertHasDefs(fn_body[2].value, 2)
96 self.assertHasDefinedIn(fn_body[1], ('a', 'b'))
108 fn_body = node.body
110 self.assertHasDefs(fn_body[0].value.args[0], 1)
111 self.assertHasDefs(fn_body[1].body[0].targets[0], 1)
112 self.assertHasDefs(fn_body[1].body[1].targets[0], 1)
[all …]
Dliveness_test.py78 fn_body = node.body
80 self.assertHasLiveOut(fn_body[0], ('a', 'x'))
81 self.assertHasLiveOut(fn_body[1], 'x')
95 fn_body = node.body
97 self.assertHasLiveOut(fn_body[0], 'a')
98 self.assertHasLiveOut(fn_body[1], 'x')
108 fn_body = node.body
110 self.assertHasLiveOut(fn_body[0], 'x')
120 fn_body = node.body
122 self.assertHasLiveOut(fn_body[0], ('x.y', 'x'))
[all …]
/external/tensorflow/tensorflow/c/
Dc_api_function.cc299 Status GraphToFunctionDef(const Graph& fn_body, const string& fn_name, in GraphToFunctionDef() argument
461 const TF_Graph* fn_body, const char* fn_name, int ninputs, in ProcessInputs() argument
464 EXCLUSIVE_LOCKS_REQUIRED(fn_body->mu) { in ProcessInputs()
471 fn_body->graph.IsValidOutputTensor(node, idx), in ProcessInputs()
497 Status ProcessOutputs(const TF_Graph* fn_body, const char* fn_name, in ProcessOutputs() argument
500 EXCLUSIVE_LOCKS_REQUIRED(fn_body->mu) { in ProcessOutputs()
506 fn_body->graph.IsValidOutputTensor(node, idx), in ProcessOutputs()
520 const TF_Graph* fn_body, const char* fn_name, int num_opers, in ComputeBodyNodes() argument
524 EXCLUSIVE_LOCKS_REQUIRED(fn_body->mu) { in ComputeBodyNodes()
526 for (const Node* node : fn_body->graph.op_nodes()) { in ComputeBodyNodes()
[all …]
Dc_api.h1311 const TF_Graph* fn_body, const char* fn_name,
1331 const TF_Graph* fn_body, const char* fn_name,
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_utils.cc46 const FunctionBody* fn_body = in ComputeShortCircuitIndices() local
48 indices->resize(fn_body->ret_nodes.size()); in ComputeShortCircuitIndices()
50 for (size_t i = 0; i < fn_body->ret_nodes.size(); ++i) { in ComputeShortCircuitIndices()
51 Node* ret_node = fn_body->ret_nodes[i]; in ComputeShortCircuitIndices()
/external/tensorflow/tensorflow/python/autograph/pyct/
Dtransformer_test.py147 fn_body = node.body
148 outer_while_body = fn_body[1].body
149 self.assertSameAnno(fn_body[0], outer_while_body[0], 'cond_state')
150 self.assertDifferentAnno(fn_body[0], outer_while_body[0], 'loop_state')
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dllvm_ir_runtime.cc73 llvm::BasicBlock* fn_body = llvm::BasicBlock::Create(*context, "body", fn); in RewriteCalls() local
74 llvm::IRBuilder<> b(fn_body); in RewriteCalls()
/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.h208 const TF_Graph* fn_body, const char* fn_name, bool append_hash_to_fn_name,
Dtf_session_helper.cc590 const TF_Graph* fn_body, const char* fn_name, bool append_hash_to_fn_name, in TF_GraphToFunction_wrapper() argument
624 fn_body, fn_name, append_hash_to_fn_name, nopers, opers_array, in TF_GraphToFunction_wrapper()