Home
last modified time | relevance | path

Searched refs:FunctionDef (Results 1 – 25 of 291) sorted by relevance

12345678910>>...12

/external/tensorflow/tensorflow/core/framework/
Dfunction_testlib.h66 gtl::ArraySlice<FunctionDef> funcs = {});
72 FunctionDef XTimesTwo();
75 FunctionDef TwoDeviceTimesFive();
78 FunctionDef TwoDeviceMult();
81 FunctionDef TwoDeviceInputOutput();
84 FunctionDef FuncWithListInput();
87 FunctionDef FuncWithListOutput();
90 FunctionDef XAddX();
93 FunctionDef XAddY();
96 FunctionDef XTimesTwoInt32();
[all …]
Dfunction_testlib.cc32 gtl::ArraySlice<FunctionDef> funcs) { in GDef()
61 FunctionDef NonZero() { in NonZero()
77 FunctionDef IsZero() { in IsZero()
95 FunctionDef RandomUniform() { in RandomUniform()
121 FunctionDef XTimesTwo() { in XTimesTwo()
140 FunctionDef TwoDeviceMult() { in TwoDeviceMult()
180 FunctionDef TwoDeviceInputOutput() { in TwoDeviceInputOutput()
212 FunctionDef FuncWithListInput() { in FuncWithListInput()
230 FunctionDef FuncWithListOutput() { in FuncWithListOutput()
248 FunctionDef XAddX() { in XAddX()
[all …]
Dgraph_to_functiondef_test.cc32 FunctionDef RemoveDebugInfo(const FunctionDef& def) { in RemoveDebugInfo()
33 FunctionDef copy = def; in RemoveDebugInfo()
40 bool EqualFunctionDef(const FunctionDef& a, const FunctionDef& b, in EqualFunctionDef()
72 FunctionDef fdef; in TEST()
75 FunctionDef fdef_expected = FunctionDefHelper::Create( in TEST()
109 FunctionDef fdef; in TEST()
112 FunctionDef fdef_expected = FunctionDefHelper::Create( in TEST()
148 FunctionDef fdef; in TEST()
151 FunctionDef fdef_expected = in TEST()
Dgraph_to_functiondef.h41 const char* description, FunctionDef* fdef);
54 FunctionDef* fdef);
57 FunctionDef* fdef);
61 FunctionDef* fdef);
64 const string& function_name, AttrSlice attrs, const FunctionDef* fdef,
Dfunction.h163 static FunctionDef Create(
174 static FunctionDef Create(const std::string& function_name,
182 static FunctionDef Define(const std::string& function_name,
189 static FunctionDef Define(gtl::ArraySlice<string> arg_def,
260 Status InstantiateFunction(const FunctionDef& fdef, AttrSlice attr_values,
272 std::string DebugString(const FunctionDef& func_def);
282 bool FunctionDefsEqual(const FunctionDef& f1, const FunctionDef& f2);
287 uint64 FunctionDefHash(const FunctionDef& fdef);
417 const FunctionDef* Find(const std::string& func) const TF_LOCKS_EXCLUDED(mu_);
428 Status AddFunctionDef(const FunctionDef& fdef,
[all …]
/external/tensorflow/tensorflow/core/ops/
Dmath_grad.cc28 Status GradForUnaryCwise(FunctionDef* g, std::vector<FDH::Node> nodes) { in GradForUnaryCwise()
46 Status AbsGrad(const AttrSlice& attrs, FunctionDef* g) { in AbsGrad()
56 Status NegGrad(const AttrSlice& attrs, FunctionDef* g) { in NegGrad()
65 Status InvGrad(const AttrSlice& attrs, FunctionDef* g) { in InvGrad()
78 Status SquareGrad(const AttrSlice& attrs, FunctionDef* g) { in SquareGrad()
90 Status SqrtGrad(const AttrSlice& attrs, FunctionDef* g) { in SqrtGrad()
104 Status RsqrtGrad(const AttrSlice& attrs, FunctionDef* g) { in RsqrtGrad()
119 Status ExpGrad(const AttrSlice& attrs, FunctionDef* g) { in ExpGrad()
129 Status Expm1Grad(const AttrSlice& attrs, FunctionDef* g) { in Expm1Grad()
139 Status LogGrad(const AttrSlice& attrs, FunctionDef* g) { in LogGrad()
[all …]
Darray_grad.cc33 Status ReshapeGrad(const AttrSlice& attrs, FunctionDef* g) { in ReshapeGrad()
54 Status SqueezeGrad(const AttrSlice& attrs, FunctionDef* g) { in SqueezeGrad()
73 Status IdentityGrad(const AttrSlice& attrs, FunctionDef* g) { in IdentityGrad()
92 Status PackGrad(const AttrSlice& attrs, FunctionDef* g) { in PackGrad()
118 Status UnpackGrad(const AttrSlice& attrs, FunctionDef* g) { in UnpackGrad()
142 Status ConcatGradHelper(const AttrSlice& attrs, FunctionDef* g, in ConcatGradHelper()
216 Status ConcatGrad(const AttrSlice& attrs, FunctionDef* g) { in ConcatGrad()
220 Status ConcatGradV2(const AttrSlice& attrs, FunctionDef* g) { in ConcatGradV2()
227 Status SplitGrad(const AttrSlice& attrs, FunctionDef* g) { in SplitGrad()
247 Status SplitVGrad(const AttrSlice& attrs, FunctionDef* g) { in SplitVGrad()
[all …]
Dnn_grad.cc25 Status SoftmaxGrad(const AttrSlice& attrs, FunctionDef* g) { in SoftmaxGrad()
50 Status LogSoftmaxGrad(const AttrSlice& attrs, FunctionDef* g) { in LogSoftmaxGrad()
75 Status ReluGrad(const AttrSlice& attrs, FunctionDef* g) { in ReluGrad()
93 Status Relu6Grad(const AttrSlice& attrs, FunctionDef* g) { in Relu6Grad()
111 Status CrossEntropyGrad(const AttrSlice& attrs, FunctionDef* g) { in CrossEntropyGrad()
140 Status Conv2DGrad(const AttrSlice& attrs, FunctionDef* g) { in Conv2DGrad()
176 Status MaxPoolGrad(const AttrSlice& attrs, FunctionDef* g) { in MaxPoolGrad()
207 Status AvgPoolGrad(const AttrSlice& attrs, FunctionDef* g) { in AvgPoolGrad()
233 Status MaxPoolGradGrad(const AttrSlice& attrs, FunctionDef* g) { in MaxPoolGradGrad()
264 Status BiasAddGrad(const AttrSlice& attrs, FunctionDef* g) { in BiasAddGrad()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dfunction_utils.h58 void ReplaceReferences(const string& from, const string& to, FunctionDef* func);
64 FunctionDef* fdef, DataType dtype);
67 OpDef_ArgDef* AddFunctionInput(const string& name, FunctionDef* fdef,
74 FunctionDef* fd);
78 const FunctionDef& function);
81 bool ContainsFunctionNodeWithOp(StringPiece op, const FunctionDef& function);
85 const FunctionDef& function);
89 int FindFunctionInputWithName(StringPiece name, const FunctionDef& function);
93 int FindFunctionOutputWithName(StringPiece name, const FunctionDef& function);
97 int FindFunctionNodeWithName(StringPiece name, const FunctionDef& function);
[all …]
Dfusion_utils.h59 const FunctionDef& first_function, const FunctionDef& second_function,
60 FunctionDef* fused_function, FunctionDefLibrary* library)>;
62 void MergeNodes(const FunctionDef& first_function,
63 const FunctionDef& second_function, FunctionDef* fused_function,
114 void LazyConjunctionNodes(const FunctionDef& first_function,
115 const FunctionDef& second_function,
116 FunctionDef* fused_function,
125 FunctionDef* FuseFunctions(
126 const FunctionDef& first_function, const FunctionDef& second_function,
Dsplit_utils_test.cc70 void RunFunction(const FunctionDef& func, in RunFunction()
106 void RunSplitFunctions(const FunctionDef& first, const FunctionDef& second, in RunSplitFunctions()
153 void CheckSplitFunctions(const FunctionDef& orig, const FunctionDef& first, in CheckSplitFunctions()
154 const FunctionDef& second, in CheckSplitFunctions()
171 std::vector<string> GetNodeNames(const FunctionDef& function) { in GetNodeNames()
179 const NodeDef& GetNode(const std::string& name, const FunctionDef& function) { in GetNode()
189 FunctionDef orig = FunctionDefHelper::Create( in TEST()
207 FunctionDef first = results.first_function; in TEST()
208 FunctionDef second = results.second_function; in TEST()
234 FunctionDef orig = FunctionDefHelper::Create( in TEST()
[all …]
Dfunction_utils.cc76 FunctionDef* func) { in ReplaceReferences()
91 FunctionDef* fdef, DataType dtype) { in AddFunctionOutputWithUniqueName()
105 OpDef_ArgDef* AddFunctionInput(const string& name, FunctionDef* fdef, in AddFunctionInput()
117 FunctionDef* fd) { in AddNode()
135 const FunctionDef& function) { in ContainsFunctionNodeWithName()
139 bool ContainsFunctionNodeWithOp(StringPiece op, const FunctionDef& function) { in ContainsFunctionNodeWithOp()
144 const FunctionDef& function) { in ContainsFunctionOutputWithName()
148 int FindFunctionInputWithName(StringPiece name, const FunctionDef& function) { in FindFunctionInputWithName()
154 int FindFunctionOutputWithName(StringPiece name, const FunctionDef& function) { in FindFunctionOutputWithName()
160 int FindFunctionNodeWithName(StringPiece name, const FunctionDef& function) { in FindFunctionNodeWithName()
[all …]
Dfunction_utils_test.cc45 FunctionDef outer = FunctionDefHelper::Create( in TEST()
57 FunctionDef function = test::function::XTimesTwo(); in TEST()
64 FunctionDef fdef; in TEST()
76 FunctionDef function = test::function::XTimesTwo(); in TEST()
83 FunctionDef function = test::function::XTimesTwo(); in TEST()
90 FunctionDef function = test::function::XTimesTwo(); in TEST()
96 FunctionDef function = test::function::XTimesTwo(); in TEST()
104 FunctionDef function = test::function::XTimesTwo(); in TEST()
112 FunctionDef function = test::function::XTimesTwo(); in TEST()
118 FunctionDef function = test::function::XTimesTwo(); in TEST()
[all …]
Dfusion_utils.cc51 string GetOutputNode(const FunctionDef& function, int output_idx) { in GetOutputNode()
57 string& GetMutableOutputNode(FunctionDef* function, int output_idx) { in GetMutableOutputNode()
116 void RenameFunctionNodes(const FunctionDef& first_function, in RenameFunctionNodes()
144 StringCollection GetFunctionInputs(const FunctionDef& function) { in GetFunctionInputs()
251 StringCollection GetFunctionOutputs(const FunctionDef& function) { in GetFunctionOutputs()
261 FunctionDef* CreateFalsePredicate( in CreateFalsePredicate()
298 void MergeNodes(const FunctionDef& first_function, in MergeNodes()
299 const FunctionDef& second_function, FunctionDef* fused_function, in MergeNodes()
377 void LazyConjunctionNodes(const FunctionDef& first_function, in LazyConjunctionNodes()
378 const FunctionDef& second_function, in LazyConjunctionNodes()
[all …]
Dmap_and_filter_fusion.cc46 const FunctionDef& fused_function, in MakeFusedNode()
88 const FunctionDef& fused_map_func, in MakeFilterNode()
101 FunctionDef* func = library->add_function(); in MakeFilterNode()
121 const FunctionDef& fused_map_func, MutableGraphView* graph, in MakeMapNode()
141 FunctionDef* func = library->add_function(); in MakeMapNode()
192 const NodeDef* filter_node) -> FunctionDef* { in OptimizeAndCollectStats()
194 const FunctionDef* map_func = in OptimizeAndCollectStats()
197 const FunctionDef* filter_func = function_library.Find(fun.func().name()); in OptimizeAndCollectStats()
Dsplit_utils.h31 FunctionDef first_function;
32 FunctionDef second_function;
68 const FunctionDef& function,
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.h82 const std::vector<const FunctionDef::ArgAttrs*>& arg_attr() const;
91 friend Status MakeGrapplerFunctionItem(const FunctionDef&, const AttrSlice&,
102 std::vector<const FunctionDef::ArgAttrs*> arg_attr,
114 std::vector<const FunctionDef::ArgAttrs*> arg_attr_;
125 bool HasParametrizedType(const FunctionDef& func);
130 bool HasParametrizedBody(const FunctionDef& func);
133 bool IsParametrized(const FunctionDef& func);
138 const FunctionDef& func, const AttrSlice& func_instantiation_attr,
145 const FunctionDef& func, const AttrSlice& func_instantiation_attr,
164 Status MakeGrapplerFunctionItem(const FunctionDef& func,
[all …]
Dfunctions_test.cc39 FunctionDef parametrized_func = FunctionDefHelper::Create( in TEST_F()
46 FunctionDef non_parametrized_func = FunctionDefHelper::Create( in TEST_F()
63 FunctionDef func = FunctionDefHelper::Create( in TEST_F()
110 FunctionDef func = FunctionDefHelper::Define( in TEST_F()
195 FunctionDef func = FunctionDefHelper::Define( in TEST_F()
282 FunctionDef func = FunctionDefHelper::Create( in TEST_F()
353 FunctionDef func = FunctionDefHelper::Create( in TEST_F()
406 FunctionDef func = FunctionDefHelper::Define( in TEST_F()
450 FunctionDef func = FunctionDefHelper::Define( in TEST_F()
478 FunctionDef func = FunctionDefHelper::Create( in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/ir/
Ddialect.td42 FunctionDef.attr will prepand with "tf" prefix
43 FunctionDef.signature.name <-> "sym_name"
44 FunctionDef.signature.description <-> "description"
45 FunctionDef.signature.is_stateful <-> "is_stateful"
46 FunctionDef.signature.gradient <-> "gradient"
47 FunctionDef.resource_arg_unique_id <-> "resource_arg_unique_ids_keys"
48 FunctionDef.resource_arg_unique_id <-> "resource_arg_unique_ids_values"
51 FunctionDef.signature.input_arg.name <-> "tfg.name"
52 FunctionDef.signature.input_arg.description <-> "tfg.description"
53 FunctionDef.signature.input_arg.handle_data <-> "tfg.handle_data"
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer_test.cc281 FunctionDef mul_func = FunctionDefHelper::Create( in TEST_F()
287 FunctionDef square_func = FunctionDefHelper::Create( in TEST_F()
294 FunctionDef quadratic_func = FunctionDefHelper::Create( in TEST_F()
353 const FunctionDef* optimized_func_0 = optimized_flib.Find(optimized_0); in TEST_F()
354 const FunctionDef* optimized_func_1 = optimized_flib.Find(optimized_1); in TEST_F()
355 const FunctionDef* optimized_func_2 = optimized_flib.Find(optimized_2); in TEST_F()
383 const std::vector<const FunctionDef*> optimized_funcs = {optimized_func_1, in TEST_F()
387 for (const FunctionDef* optimized_func : optimized_funcs) { in TEST_F()
430 FunctionDef my_mul = FunctionDefHelper::Create( in TEST_F()
439 FunctionDef my_fwd = FunctionDefHelper::Create( in TEST_F()
[all …]
Dfunction_api_info.cc29 Status FunctionApiInfo::Init(const FunctionDef& function_def) { in Init()
102 bool IsSameSignature(const FunctionDef& f1, const FunctionDef& f2, in IsSameSignature()
124 const std::vector<const FunctionDef*>& equiv_funcs, in ValidateSignature()
146 const std::unordered_map<string, std::vector<const FunctionDef*>>& in ValidateSignatures()
158 std::unordered_map<string, std::vector<const FunctionDef*>> infer_funcs; in Init()
159 std::unordered_map<string, std::vector<const FunctionDef*>> fwd_funcs; in Init()
160 std::unordered_map<string, std::vector<const FunctionDef*>> bwd_funcs; in Init()
/external/tensorflow/tensorflow/c/experimental/saved_model/core/
Dtf_concrete_function_loading_test.cc57 FunctionDef FuncDefWithNumInputsOutputs(int num_inputs, int num_outputs) { in FuncDefWithNumInputsOutputs()
58 FunctionDef func; in FuncDefWithNumInputsOutputs()
79 FunctionDef func = FuncDefWithNumInputsOutputs(2, 0); in TEST_F()
100 FunctionDef func = FuncDefWithNumInputsOutputs(3, 0); in TEST_F()
122 FunctionDef func = FuncDefWithNumInputsOutputs(2, 0); in TEST_F()
144 FunctionDef func = FuncDefWithNumInputsOutputs(3, 0); in TEST_F()
168 FunctionDef func = FuncDefWithNumInputsOutputs(4, 0); in TEST_F()
191 FunctionDef func = FuncDefWithNumInputsOutputs(0, 2); in TEST_F()
211 FunctionDef func = FuncDefWithNumInputsOutputs(1, 2); in TEST_F()
232 FunctionDef func = FuncDefWithNumInputsOutputs(3, 3); in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/data/
Dhash_utils_test.cc38 uint64 GetHash(const FunctionDefLibrary& library, const FunctionDef& fn) { in GetHash()
52 Status CheckEqual(const FunctionDefLibrary& library, const FunctionDef& fn1, in CheckEqual()
53 const FunctionDef& fn2) { in CheckEqual()
91 FunctionDef* f1 = fl.add_function(); in TEST_F()
99 FunctionDef* f2 = fl.add_function(); in TEST_F()
114 FunctionDef* f1 = fl.add_function(); in TEST_F()
122 FunctionDef* f2 = fl.add_function(); in TEST_F()
140 FunctionDef* f1 = fl.add_function(); in TEST_F()
148 FunctionDef* f2 = fl.add_function(); in TEST_F()
483 FunctionDef* f1 = fl1->add_function(); in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/function/
Druntime_client_test.cc85 FunctionDef MakeNullaryFunction() { in MakeNullaryFunction()
86 FunctionDef fd; in MakeNullaryFunction()
116 FunctionDef MakeUnaryFunction() { in MakeUnaryFunction()
117 FunctionDef fd; in MakeUnaryFunction()
139 FunctionDef MakeBinaryFunction() { in MakeBinaryFunction()
140 FunctionDef fd; in MakeBinaryFunction()
201 StatusOr<FunctionDef> fdef_ret = rt.GetFunctionProto("NullaryFunction"); in TEST()
204 FunctionDef fdef = *fdef_ret; in TEST()
/external/tensorflow/tensorflow/cc/experimental/libtf/tests/
Dfunction_test.cc38 using tensorflow::FunctionDef;
79 FunctionDef SquareFunc() { in SquareFunc()
99 FunctionDef AddFunc() { in AddFunc()
119 FunctionDef IdentityNFunc() { in IdentityNFunc()
153 FunctionDef fdef = SquareFunc(); in TEST_P()
171 FunctionDef fdef = AddFunc(); in TEST_P()
194 FunctionDef fdef = IdentityNFunc(); in TEST_P()
217 FunctionDef fdef = SquareFunc(); in TEST_P()
239 FunctionDef fdef = IdentityNFunc(); in TEST_P()
264 FunctionDef fdef = AddFunc(); in TEST_P()

12345678910>>...12