| /external/tensorflow/tensorflow/core/framework/ |
| D | function_test.cc | 67 auto fdef = FDH::Create( in TEST() local
95 EXPECT_EQ(DebugString(fdef), e); in TEST()
100 InstantiateFunction(fdef, Attrs({{"T", DT_FLOAT}}), GetOpSig, &result)); in TEST()
114 auto fdef = FDH::Create( in TEST() local
141 EXPECT_EQ(DebugString(fdef), e); in TEST()
146 InstantiateFunction(fdef, Attrs({{"T", DT_FLOAT}}), GetOpSig, &result)); in TEST()
160 auto fdef = FDH::Create( in TEST() local
185 EXPECT_EQ(DebugString(fdef), e); in TEST()
190 InstantiateFunction(fdef, Attrs({{"T", DT_FLOAT}}), GetOpSig, &result)); in TEST()
209 auto fdef = FDH::Create( in TEST() local
[all …]
|
| D | graph_to_functiondef.cc | 117 FunctionDef* fdef) { in GraphToFunctionDef() argument
118 fdef->mutable_signature()->set_name(name); in GraphToFunctionDef()
130 while (fdef->signature().input_arg_size() <= index) { in GraphToFunctionDef()
131 fdef->mutable_signature()->add_input_arg(); in GraphToFunctionDef()
134 fdef->mutable_signature()->mutable_input_arg(index); in GraphToFunctionDef()
147 while (fdef->signature().output_arg_size() <= index) { in GraphToFunctionDef()
148 fdef->mutable_signature()->add_output_arg(); in GraphToFunctionDef()
151 fdef->mutable_signature()->mutable_output_arg(index); in GraphToFunctionDef()
162 NodeDef* node_def = fdef->add_node_def(); in GraphToFunctionDef()
221 for (int i = 0; i < fdef->signature().input_arg_size(); ++i) { in GraphToFunctionDef()
[all …]
|
| D | function.cc | 541 string Print(const FunctionDef& fdef) { in Print() argument
543 const OpDef& sig = fdef.signature(); in Print()
569 for (const auto& n : fdef.node_def()) { in Print()
572 for (const auto& cr : fdef.control_ret()) { in Print()
575 for (const auto& r : fdef.ret()) { in Print()
682 Status InstantiateFunction(const FunctionDef& fdef, AttrSlice attr_values, in InstantiateFunction() argument
685 VLOG(4) << "Instantiation Function: " << Print(fdef); in InstantiateFunction()
687 const OpDef& sig = fdef.signature(); in InstantiateFunction()
691 fdef.attr().count(FunctionLibraryDefinition::kIntsOnDeviceAttr) != 0 && in InstantiateFunction()
692 fdef.attr().at(FunctionLibraryDefinition::kIntsOnDeviceAttr).b(); in InstantiateFunction()
[all …]
|
| D | graph_to_functiondef_test.cc | 72 FunctionDef fdef; in TEST() local
73 TF_EXPECT_OK(GraphToFunctionDef(*graph, "test_fn", &fdef)); in TEST()
91 EqualFunctionDef(fdef_expected, RemoveDebugInfo(fdef), &diff); in TEST()
109 FunctionDef fdef; in TEST() local
110 TF_EXPECT_OK(GraphToFunctionDef(*graph, "test_fn", &fdef)); in TEST()
125 EqualFunctionDef(fdef_expected, RemoveDebugInfo(fdef), &diff); in TEST()
|
| D | function.h | 229 Status InstantiateFunction(const FunctionDef& fdef, AttrSlice attr_values,
256 uint64 FunctionDefHash(const FunctionDef& fdef);
339 Status AddFunctionDef(const FunctionDef& fdef) LOCKS_EXCLUDED(mu_);
352 Status ReplaceFunction(const string& func, const FunctionDef& fdef);
448 FunctionDef fdef; member
459 Status AddFunctionDefHelper(const FunctionDef& fdef, bool* added)
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | function_def_to_graph_test.py | 49 fdef = graph_to_function_def.graph_to_function_def(
54 fdef.signature.name = "_whats_in_a_name"
55 return fdef
59 fdef = self._build_function_def()
60 g = function_def_to_graph.function_def_to_graph(fdef)
69 fdef = self._build_function_def()
71 g = function_def_to_graph.function_def_to_graph(fdef)
78 fdef, input_shapes=[tensor_shape.vector(5),
86 fdef, input_shapes=[None, tensor_shape.matrix(5, 7)])
96 fdef, input_shapes=[tensor_shape.matrix(5, 7)])
[all …]
|
| D | function_def_to_graph.py | 31 def function_def_to_graph(fdef, input_shapes=None): argument
50 func_graph = FuncGraph(fdef.signature.name)
52 fdef, input_shapes)
62 nested_to_flat_tensor_name[arg.name] for arg in fdef.signature.input_arg
70 nested_to_flat_tensor_name[fdef.ret[arg.name]]
71 for arg in fdef.signature.output_arg
77 func_graph.get_operation_by_name(fdef.control_ret[ret_name])
78 for ret_name in fdef.signature.control_output
92 def function_def_to_graph_def(fdef, input_shapes=None): argument
128 if input_shapes and len(input_shapes) != len(fdef.signature.input_arg):
[all …]
|
| D | function.py | 297 fdef = function_pb2.FunctionDef()
299 fdef.ParseFromString(compat.as_bytes(proto_data))
300 return fdef
1034 def _from_definition(fdef, grad_func=None): argument
1051 argnames = [arg.name for arg in fdef.signature.input_arg]
1053 dtypes.as_dtype(arg.type) for arg in fdef.signature.input_arg)
1054 func_name = fdef.signature.name
1058 out_names = [arg.name for arg in fdef.signature.output_arg]
1062 serialized = fdef.SerializeToString()
1066 result._op_def = fdef.signature
[all …]
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | function_deserialization.py | 243 for fdef in _sort_function_defs(library):
244 copy = _fix_fdef(fdef, functions, load_shared_name_suffix)
247 for dep in _list_function_deps(fdef):
252 functions[fdef.signature.name] = func
266 for fdef in library.function:
267 for dep in _list_function_deps(fdef):
268 edges[dep].append(fdef.signature.name)
269 in_count[fdef.signature.name] += 1
272 fdef.signature.name
273 for fdef in library.function
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | create_xla_launch_op_test.cc | 67 for (const auto& fdef : flib) { in Init() local
68 *(proto.add_function()) = fdef; in Init()
95 FunctionDef fdef = XTimesY(); in TEST_F() local
96 (*fdef.mutable_attr())["_XlaCompile"] = BoolAttr(true); in TEST_F()
97 Init({fdef}); in TEST_F()
120 FunctionDef fdef = XTimesY(); in TEST_F() local
121 Init({fdef}); in TEST_F()
133 FunctionDef fdef = XTimesY(); in TEST_F() local
134 (*fdef.mutable_attr())["_XlaCompile"] = BoolAttr(false); in TEST_F()
135 Init({fdef}); in TEST_F()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_function.cc | 164 FunctionDef* fdef) { in FillFunctionBody() argument
166 for (const auto& func_attr : fdef->signature().attr()) { in FillFunctionBody()
173 NodeDef* node_def = fdef->add_node_def(); in FillFunctionBody()
249 fdef->mutable_signature()->set_is_stateful(true); in FillFunctionBody()
287 OpDef::AttrDef* attr_def = fdef->mutable_signature()->add_attr(); in FillFunctionBody()
307 const char* description, FunctionDef* fdef) { in GraphToFunctionDef() argument
313 fdef->mutable_signature()->set_description(description); in GraphToFunctionDef()
337 OpDef::ArgDef* argdef = fdef->mutable_signature()->add_output_arg(); in GraphToFunctionDef()
351 OpDef::ArgDef* argdef = fdef->mutable_signature()->add_input_arg(); in GraphToFunctionDef()
393 FillFunctionBody(fn_name, node_names, body_nodes, tensor_renaming, fdef)); in GraphToFunctionDef()
[all …]
|
| D | c_api_function_test.cc | 229 FunctionDef fdef() { in fdef() function in tensorflow::__anon1a216e640111::CApiFunctionTest
230 tensorflow::FunctionDef fdef; in fdef() local
231 EXPECT_TRUE(GetFunctionDef(func_, &fdef)); in fdef()
232 return fdef; in fdef()
252 void VerifyFDefNodes(const tensorflow::FunctionDef& fdef, in VerifyFDefNodes() argument
254 ASSERT_EQ(nodes.size(), fdef.node_def_size()) in VerifyFDefNodes()
257 << "] Actual nodes in fdef: " << fdef.DebugString(); in VerifyFDefNodes()
258 for (const NodeDef& node_def : fdef.node_def()) { in VerifyFDefNodes()
261 << " in fdef: " << fdef.DebugString(); in VerifyFDefNodes()
265 void VerifyFDefInputs(const tensorflow::FunctionDef& fdef, in VerifyFDefInputs() argument
[all …]
|
| /external/protobuf/ruby/src/main/java/com/google/protobuf/jruby/ |
| D | RubyMessage.java | 190 for (Descriptors.FieldDescriptor fdef : descriptor.getFields()) { in eq()
191 IRubyObject thisVal = getField(context, fdef); in eq()
192 IRubyObject thatVal = message.getField(context, fdef); in eq()
352 for (Descriptors.FieldDescriptor fdef : this.descriptor.getFields()) { in toHash()
353 IRubyObject value = getField(context, fdef); in toHash()
361 ret.fastASet(runtime.newSymbol(fdef.getName()), value); in toHash()
403 for (Descriptors.FieldDescriptor fdef : this.descriptor.getFields()) { in deepCopy()
404 if (fdef.isRepeated()) { in deepCopy()
405 copy.addRepeatedField(fdef, this.getRepeatedField(context, fdef).deepCopy(context)); in deepCopy()
406 } else if (fields.containsKey(fdef)) { in deepCopy()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | test_util.cc | 26 const FunctionDef* fdef = library.Find(name); in InstantiateFunctionForTest() local
27 TF_RET_CHECK(fdef != nullptr); in InstantiateFunctionForTest()
34 InstantiateFunction(*fdef, AttrSlice(), get_func_sig, &inst)); in InstantiateFunctionForTest()
|
| /external/python/cpython3/Objects/ |
| D | moduleobject.c | 149 PyMethodDef *fdef; in _add_methods_to_object() local
151 for (fdef = functions; fdef->ml_name != NULL; fdef++) { in _add_methods_to_object()
152 if ((fdef->ml_flags & METH_CLASS) || in _add_methods_to_object()
153 (fdef->ml_flags & METH_STATIC)) { in _add_methods_to_object()
159 func = PyCFunction_NewEx(fdef, (PyObject*)module, name); in _add_methods_to_object()
163 if (PyObject_SetAttrString(module, fdef->ml_name, func) != 0) { in _add_methods_to_object()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | dump_graph_test.cc | 55 FunctionDef fdef; in TEST() local
57 string ret = DumpFunctionDefToFile("function", fdef); in TEST()
|
| D | dump_graph.cc | 129 string DumpFunctionDefToFile(const string& name, FunctionDef const& fdef, in DumpFunctionDefToFile() argument
131 return WriteTextProtoToUniqueFile(Env::Default(), name, "FunctionDef", fdef, in DumpFunctionDefToFile()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function.cc | 403 Status FunctionDefToBody(const FunctionDef& fdef, AttrSlice attrs,
591 &fbody->fdef.signature(), this, fbody->arg_types, input_memory_types, in CreateKernel()
600 const FunctionDef& fdef, AttrSlice attrs, in FunctionDefToBody() argument
603 return FunctionDefToBodyHelper(fdef, attrs, lib_def, get_func_sig_, fbody); in FunctionDefToBody()
608 return FunctionDefToBodyHelper(fdef, attrs, lib_def, get_func_sig, fbody); in FunctionDefToBody()
615 const FunctionDef* fdef = lib_def->Find(func.name()); in InstantiateSymbolicGradient() local
616 if (fdef == nullptr) { in InstantiateSymbolicGradient()
718 const FunctionDef* fdef = lib_def->Find(function_name); in Instantiate() local
719 if (fdef == nullptr) { in Instantiate()
722 TF_RETURN_IF_ERROR(FunctionDefToBody(*fdef, attrs, lib_def, &fbody)); in Instantiate()
[all …]
|
| D | process_function_library_runtime.cc | 427 const FunctionDef& fdef, in ValidateMultiDeviceOptions() argument
429 const OpDef& signature = fdef.signature(); in ValidateMultiDeviceOptions()
436 if (fdef.attr().count(FunctionLibraryDefinition::kIntsOnDeviceAttr) != 0 && in ValidateMultiDeviceOptions()
437 fdef.attr().at(FunctionLibraryDefinition::kIntsOnDeviceAttr).b()) { in ValidateMultiDeviceOptions()
479 const FunctionDef* fdef, in GetGraphAndRets() argument
490 FunctionDefToBodyHelper(*fdef, attrs, lib_def, get_func_sig, &tmp_fbody)); in GetGraphAndRets()
545 const FunctionDef* fdef = lib_def->Find(function_name); in InstantiateMultiDevice() local
546 if (fdef == nullptr) { in InstantiateMultiDevice()
551 TF_RETURN_IF_ERROR(ValidateMultiDeviceOptions(*fdef, options)); in InstantiateMultiDevice()
557 TF_RETURN_IF_ERROR(GetGraphAndRets(function_name, attrs, fdef, lib_def, in InstantiateMultiDevice()
[all …]
|
| D | graph_execution_state.cc | 702 for (const FunctionDef& fdef : new_graph.library().function()) { in OptimizeGraph() local
703 const string& func_name = fdef.signature().name(); in OptimizeGraph()
707 TF_RETURN_IF_ERROR((*optimized_flib)->ReplaceFunction(func_name, fdef)); in OptimizeGraph()
710 TF_RETURN_IF_ERROR((*optimized_flib)->AddFunctionDef(fdef)); in OptimizeGraph()
804 const FunctionDef* fdef = optimized_flib->Find(node->def().op()); in BuildGraph() local
805 if (fdef != nullptr) { in BuildGraph()
806 for (const NodeDef& ndef : fdef->node_def()) { in BuildGraph()
|
| D | function.h | 76 FunctionDef fdef; member
287 const FunctionDef& fdef, const AttrSlice& attrs,
|
| D | lower_if_op.cc | 241 const FunctionDef* fdef = flib.Find(n->type_string()); in InlineCallInGraph() local
242 CHECK(fdef != nullptr); in InlineCallInGraph()
245 FunctionDefToBodyHelper(*fdef, n->attrs(), &flib, in InlineCallInGraph()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | while_v2.py | 411 fdef = while_op.graph._get_function(func_name).definition
418 func_graph = function_def_to_graph.function_def_to_graph(fdef, input_shapes)
883 fdef = functions[func_name].definition
884 output_arg_name = fdef.signature.output_arg[output_idx].name
885 output_tensor_name = fdef.ret[output_arg_name]
888 [arg.name for arg in fdef.signature.input_arg],
889 {ndef.name: ndef for ndef in fdef.node_def},
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 307 Status EagerContext::MaybeRegisterFunctionRemotely(const FunctionDef& fdef) { in MaybeRegisterFunctionRemotely() argument
320 *requests[i].mutable_function_def() = fdef; in MaybeRegisterFunctionRemotely()
343 Status EagerContext::AddFunctionDef(const FunctionDef& fdef) { in AddFunctionDef() argument
345 TF_RETURN_IF_ERROR(func_lib_def_.AddFunctionDef(fdef)); in AddFunctionDef()
347 return MaybeRegisterFunctionRemotely(fdef); in AddFunctionDef()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | partitioned_function_ops.cc | 132 const FunctionDef* fdef = flib->Find(func_->name()); in FillOutputDevices() local
133 if (fdef == nullptr) { in FillOutputDevices()
139 for (const OpDef::ArgDef& ret_def : fdef->signature().output_arg()) { in FillOutputDevices()
|