HomeSort by relevance Sort by last modified time
    Searched refs:fdef (Results 1 - 25 of 58) sorted by null

1 2 3

  /external/tensorflow/tensorflow/core/framework/
graph_to_functiondef.h 25 // Converts 'graph' to a FunctionDef 'fdef', with name 'name'.
28 FunctionDef* fdef);
function_test.cc 67 auto fdef = FDH::Create( local
95 EXPECT_EQ(DebugString(fdef), e);
100 InstantiateFunction(fdef, Attrs({{"T", DT_FLOAT}}), GetOpSig, &result));
114 auto fdef = FDH::Create( local
141 EXPECT_EQ(DebugString(fdef), e);
146 InstantiateFunction(fdef, Attrs({{"T", DT_FLOAT}}), GetOpSig, &result));
160 auto fdef = FDH::Create( local
185 EXPECT_EQ(DebugString(fdef), e);
190 InstantiateFunction(fdef, Attrs({{"T", DT_FLOAT}}), GetOpSig, &result));
209 auto fdef = FDH::Create local
246 auto fdef = FDH::Create( local
300 auto fdef = FDH::Create( local
347 auto fdef = FDH::Define( local
416 auto fdef = FDH::Create( local
489 auto fdef = FDH::Define( local
541 auto fdef = test::function::XTimesTwoInt32(); local
549 auto fdef = test::function::XTimesTwoInt32(); local
564 auto fdef = local
584 auto fdef = local
593 auto fdef = FDH::Define("test", {}, {}, {"T:{float, double, int32, int64}"}, local
604 auto fdef = FDH::Define("test", {"x:float", "x:float"}, {}, {}, {}); local
611 auto fdef = FDH::Define("test", {"x:float"}, {}, {}, local
622 auto fdef = FDH::Create("test", {"x:float"}, {}, {}, local
633 auto fdef = FDH::Define("test", {"x:float"}, {}, {}, local
643 auto fdef = local
654 auto fdef = FDH::Create("test", {}, {"y: float"}, {}, local
665 auto fdef = FDH::Create("test", {}, {"y: float"}, {}, local
676 auto fdef = FDH::Create("test", {}, {"y: float"}, {}, local
699 auto fdef = FDH::Define("test", {}, {"y: float"}, {}, local
709 auto fdef = FDH::Create( local
735 auto fdef = FDH::Create( local
762 auto fdef = FDH::Create( local
789 auto fdef = FDH::Create( local
810 auto fdef = FDH::Create( local
831 auto fdef = FDH::Create( local
859 auto fdef = FDH::Create( local
887 auto fdef = FDH::Create( local
1033 FunctionDef fdef = test::function::XTimesTwo(); local
1083 FunctionDef fdef = test::function::XTimesFour(); local
1130 FunctionDef fdef = test::function::XTimesFour(); local
1300 auto fdef = proto.add_function(); local
1321 auto fdef = proto.add_function(); local
    [all...]
graph_to_functiondef_test.cc 72 FunctionDef fdef; local
73 TF_EXPECT_OK(GraphToFunctionDef(*graph, "test_fn", &fdef));
91 EqualFunctionDef(fdef_expected, RemoveDebugInfo(fdef), &diff);
109 FunctionDef fdef; local
110 TF_EXPECT_OK(GraphToFunctionDef(*graph, "test_fn", &fdef));
125 EqualFunctionDef(fdef_expected, RemoveDebugInfo(fdef), &diff);
graph_to_functiondef.cc 117 FunctionDef* fdef) {
118 fdef->mutable_signature()->set_name(name);
130 while (fdef->signature().input_arg_size() <= index) {
131 fdef->mutable_signature()->add_input_arg();
134 fdef->mutable_signature()->mutable_input_arg(index);
147 while (fdef->signature().output_arg_size() <= index) {
148 fdef->mutable_signature()->add_output_arg();
151 fdef->mutable_signature()->mutable_output_arg(index);
162 NodeDef* node_def = fdef->add_node_def();
221 for (int i = 0; i < fdef->signature().input_arg_size(); ++i)
    [all...]
function.cc 104 // attr_values should specify all attrs defined in fdef.
541 string Print(const FunctionDef& fdef) {
543 const OpDef& sig = fdef.signature();
569 for (const auto& n : fdef.node_def()) {
572 for (const auto& cr : fdef.control_ret()) {
575 for (const auto& r : fdef.ret()) {
682 Status InstantiateFunction(const FunctionDef& fdef, AttrSlice attr_values,
685 VLOG(4) << "Instantiation Function: " << Print(fdef);
687 const OpDef& sig = fdef.signature();
691 fdef.attr().count(FunctionLibraryDefinition::kIntsOnDeviceAttr) != 0 &
1343 const FunctionDef* fdef = GetAttrImpl(ndef); local
1542 FunctionDef fdef; local
1601 FunctionDef fdef; local
    [all...]
function.h 208 // "fdef" encodes a TF function with some attrs in fdef.signature.attr
229 Status InstantiateFunction(const FunctionDef& fdef, AttrSlice attr_values,
253 // Return a hash of `fdef` that is consistent with FunctionDefsEqual method.
256 uint64 FunctionDefHash(const FunctionDef& fdef);
333 // Adds function definition 'fdef' to this function library.
335 // 'fdef' already exists in this function library.
336 // If 'fdef' is successfully added to the library, it will be accessible
339 Status AddFunctionDef(const FunctionDef& fdef) LOCKS_EXCLUDED(mu_);
348 // Replaces the function corresponding to `func` with `fdef`. Return
448 FunctionDef fdef; member in struct:tensorflow::FunctionLibraryDefinition::FunctionDefAndOpRegistration
    [all...]
  /external/tensorflow/tensorflow/compiler/tf2xla/
test_util.cc 26 const FunctionDef* fdef = library.Find(name); local
27 TF_RET_CHECK(fdef != nullptr);
34 InstantiateFunction(*fdef, AttrSlice(), get_func_sig, &inst));
  /external/tensorflow/tensorflow/compiler/jit/
create_xla_launch_op_test.cc 67 for (const auto& fdef : flib) {
68 *(proto.add_function()) = fdef;
95 FunctionDef fdef = XTimesY(); local
96 (*fdef.mutable_attr())["_XlaCompile"] = BoolAttr(true);
97 Init({fdef});
120 FunctionDef fdef = XTimesY(); local
121 Init({fdef});
133 FunctionDef fdef = XTimesY(); local
134 (*fdef.mutable_attr())["_XlaCompile"] = BoolAttr(false);
135 Init({fdef});
    [all...]
  /external/tensorflow/tensorflow/python/framework/
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...]
function_def_to_graph.py 31 def function_def_to_graph(fdef, input_shapes=None):
41 fdef: FunctionDef.
44 `fdef.signature.input_arg`. If a shape is None, the corresponding input
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_outpu
    [all...]
function.py 297 fdef = function_pb2.FunctionDef()
299 fdef.ParseFromString(compat.as_bytes(proto_data))
300 return fdef
    [all...]
  /external/tensorflow/tensorflow/python/saved_model/
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.functio
    [all...]
  /external/tensorflow/tensorflow/core/util/
dump_graph.h 47 string DumpFunctionDefToFile(const string& name, FunctionDef const& fdef,
dump_graph_test.cc 55 FunctionDef fdef; local
57 string ret = DumpFunctionDefToFile("function", fdef);
dump_graph.cc 129 string DumpFunctionDefToFile(const string& name, FunctionDef const& fdef,
131 return WriteTextProtoToUniqueFile(Env::Default(), name, "FunctionDef", fdef,
  /external/tensorflow/tensorflow/c/
c_api_function_test.cc 229 FunctionDef fdef() { function in class:tensorflow::__anon43995::CApiFunctionTest
230 tensorflow::FunctionDef fdef; local
231 EXPECT_TRUE(GetFunctionDef(func_, &fdef));
232 return fdef;
252 void VerifyFDefNodes(const tensorflow::FunctionDef& fdef,
254 ASSERT_EQ(nodes.size(), fdef.node_def_size())
257 << "] Actual nodes in fdef: " << fdef.DebugString();
258 for (const NodeDef& node_def : fdef.node_def()) {
261 << " in fdef: " << fdef.DebugString()
359 tensorflow::FunctionDef fdef; local
370 tensorflow::FunctionDef fdef; local
952 tensorflow::FunctionDef fdef; local
1508 tensorflow::FunctionDef fdef; local
1516 tensorflow::FunctionDef fdef; local
1524 tensorflow::FunctionDef fdef; local
1611 tensorflow::FunctionDef fdef; local
    [all...]
c_api_function.cc 164 FunctionDef* fdef) {
166 for (const auto& func_attr : fdef->signature().attr()) {
173 NodeDef* node_def = fdef->add_node_def();
249 fdef->mutable_signature()->set_is_stateful(true);
287 OpDef::AttrDef* attr_def = fdef->mutable_signature()->add_attr();
307 const char* description, FunctionDef* fdef) {
313 fdef->mutable_signature()->set_description(description);
337 OpDef::ArgDef* argdef = fdef->mutable_signature()->add_output_arg();
351 OpDef::ArgDef* argdef = fdef->mutable_signature()->add_input_arg();
393 FillFunctionBody(fn_name, node_names, body_nodes, tensor_renaming, fdef));
    [all...]