Searched refs:func_node (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 105 bool IsDirectFunctionCall(const FunctionDef& func, const NodeDef& func_node) { in IsDirectFunctionCall() argument
106 return func_node.op() == func.signature().name(); in IsDirectFunctionCall()
111 bool IsIndirectFunctionCall(const FunctionDef& func, const NodeDef& func_node) { in IsIndirectFunctionCall() argument
112 auto* func_attr = AttrSlice(func_node).Find(kFuncAttrName); in IsIndirectFunctionCall()
118 const NodeDef& func_node) { in FunctionInstantiationAttributes() argument
119 if (IsDirectFunctionCall(func, func_node)) { in FunctionInstantiationAttributes()
120 return AttrSlice(func_node); in FunctionInstantiationAttributes()
122 } else if (IsIndirectFunctionCall(func, func_node)) { in FunctionInstantiationAttributes()
123 auto* func_attr = AttrSlice(func_node).Find(kFuncAttrName); in FunctionInstantiationAttributes()
128 << SummarizeNodeDef(func_node); in FunctionInstantiationAttributes()
[all …]
|
| /external/tensorflow/tensorflow/python/autograph/pyct/static_analysis/ |
| D | live_values_test.py | 97 func_node = node.body[0].value.func
98 self.assertEquals(foo, anno.getanno(func_node, 'live_val'))
99 self.assertEquals(('foo',), anno.getanno(func_node, 'fqn'))
107 func_node = node.body[0].value.func
108 self.assertEquals(constant_op.constant, anno.getanno(func_node, 'live_val'))
110 anno.getanno(func_node, 'fqn'))
125 func_node = node.body[0].value.func
126 self.assertEquals(TestClass.member, anno.getanno(func_node, 'live_val'))
127 self.assertEquals(TestClass, anno.getanno(func_node, 'parent_type'))
128 self.assertEquals(('TestClass', 'member'), anno.getanno(func_node, 'fqn'))
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.cc | 879 for (const NodeDef& func_node : item.function_body().node()) { in MakeFunctionDef() local
880 const string& name = func_node.name(); in MakeFunctionDef()
883 if (IsPlaceholder(func_node) && input_placeholders.count(name)) continue; in MakeFunctionDef()
885 if (IsIdentity(func_node) && output_tensors.count(name)) continue; in MakeFunctionDef()
888 *func_def_node = func_node; in MakeFunctionDef()
|
| /external/mesa3d/src/compiler/nir/ |
| D | nir.h | 1946 struct exec_node *func_node = exec_list_get_head(&shader->functions); in nir_shader_get_entrypoint() local
1947 nir_function *func = exec_node_data(nir_function, func_node, node); in nir_shader_get_entrypoint()
|