Home
last modified time | relevance | path

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

12345678910

/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()
120 FunctionDef XTimesTwo() { in XTimesTwo()
139 FunctionDef TwoDeviceMult() { in TwoDeviceMult()
179 FunctionDef TwoDeviceInputOutput() { in TwoDeviceInputOutput()
211 FunctionDef FuncWithListInput() { in FuncWithListInput()
229 FunctionDef FuncWithListOutput() { in FuncWithListOutput()
247 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.h145 static FunctionDef Create(
156 static FunctionDef Create(const std::string& function_name,
164 static FunctionDef Define(const std::string& function_name,
171 static FunctionDef Define(gtl::ArraySlice<string> arg_def,
242 Status InstantiateFunction(const FunctionDef& fdef, AttrSlice attr_values,
254 std::string DebugString(const FunctionDef& func_def);
264 bool FunctionDefsEqual(const FunctionDef& f1, const FunctionDef& f2);
269 uint64 FunctionDefHash(const FunctionDef& fdef);
399 const FunctionDef* Find(const std::string& func) const TF_LOCKS_EXCLUDED(mu_);
410 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 …]
Dvectorization_utils_test.cc50 Status WrapFunctionWithMapDefun(const FunctionDef& inner, FunctionDef* result) { in WrapFunctionWithMapDefun()
96 Status WrapAndVectorize(const FunctionDef& fn, FunctionDefLibrary* lib, in WrapAndVectorize()
97 FunctionDef** result) { in WrapAndVectorize()
98 FunctionDef outer; in WrapAndVectorize()
118 string GetRetval(const FunctionDef& function_def, int index) { in GetRetval()
159 FunctionDef inner = FunctionDefHelper::Create( in TEST()
167 FunctionDef* vectorized; in TEST()
234 FunctionDef inner = FunctionDefHelper::Create( in TEST()
247 FunctionDef* vectorized; in TEST()
263 const FunctionDef* map_defun_fn = in TEST()
[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,
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()
103 FunctionDef* func = library->add_function(); in MakeFilterNode()
123 const FunctionDef& fused_map_func, MutableGraphView* graph, in MakeMapNode()
145 FunctionDef* func = library->add_function(); in MakeMapNode()
196 const NodeDef* filter_node) -> FunctionDef* { in OptimizeAndCollectStats()
198 const FunctionDef* map_func = in OptimizeAndCollectStats()
201 const FunctionDef* filter_func = function_library.Find(fun.func().name()); in OptimizeAndCollectStats()
Dmap_vectorization.cc62 FunctionDef* CreateMapDefunWrapper(const NodeDef& map_node, in CreateMapDefunWrapper()
63 const FunctionDef& orig_func, in CreateMapDefunWrapper()
65 FunctionDef* vectorized_func = library->add_function(); in CreateMapDefunWrapper()
116 FunctionDef* AddVectorizedFunction(const NodeDef& map_node, in AddVectorizedFunction()
117 const FunctionDef& orig_func, in AddVectorizedFunction()
121 FunctionDef* vectorized_func = in AddVectorizedFunction()
126 FunctionDef* result; in AddVectorizedFunction()
196 const FunctionDef& vectorized_func, in AddNewBatchNode()
256 const FunctionDef& vectorized_func, in AddNewMapNode()
345 FunctionDef* branch_func = library->add_function(); in AddBranch()
[all …]
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.h79 const std::vector<const FunctionDef::ArgAttrs*>& arg_attr() const;
88 friend Status MakeGrapplerFunctionItem(const FunctionDef&, const AttrSlice&,
99 std::vector<const FunctionDef::ArgAttrs*> arg_attr,
111 std::vector<const FunctionDef::ArgAttrs*> arg_attr_;
122 bool HasParametrizedType(const FunctionDef& func);
127 bool HasParametrizedBody(const FunctionDef& func);
130 bool IsParametrized(const FunctionDef& func);
135 const FunctionDef& func, const AttrSlice& func_instantiation_attr,
142 const FunctionDef& func, const AttrSlice& func_instantiation_attr,
161 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/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/grappler/optimizers/
Dmeta_optimizer_test.cc263 FunctionDef mul_func = FunctionDefHelper::Create( in TEST_F()
269 FunctionDef square_func = FunctionDefHelper::Create( in TEST_F()
276 FunctionDef quadratic_func = FunctionDefHelper::Create( in TEST_F()
335 const FunctionDef* optimized_func_0 = optimized_flib.Find(optimized_0); in TEST_F()
336 const FunctionDef* optimized_func_1 = optimized_flib.Find(optimized_1); in TEST_F()
337 const FunctionDef* optimized_func_2 = optimized_flib.Find(optimized_2); in TEST_F()
365 const std::vector<const FunctionDef*> optimized_funcs = {optimized_func_1, in TEST_F()
369 for (const FunctionDef* optimized_func : optimized_funcs) { in TEST_F()
412 FunctionDef my_mul = FunctionDefHelper::Create( in TEST_F()
421 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/core/common_runtime/
Dfunction_def_utils.h28 class FunctionDef; variable
34 Status FunctionDefToBodyHelper(const FunctionDef& fdef, const AttrSlice& attrs,
42 const FunctionDef& fdef, const AttrSlice& attrs,
Disolate_placer_inspection_required_ops_pass_test.cc91 FunctionDef func = test::function::ResourceIdentity(); in TEST()
131 FunctionDef func = test::function::ResourceIdentity(); in TEST()
171 FunctionDef func = test::function::Swap(); in TEST()
219 FunctionDef func = test::function::Swap(); in TEST()
266 FunctionDef func = test::function::Swap(); in TEST()
333 FunctionDef func = test::function::Swap(); in TEST()
397 FunctionDef func = test::function::ResourceIdentity(); in TEST()
421 FunctionDef func = test::function::ReadResourceVariable(); in TEST()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_kernel_creator_test.cc42 FunctionDef XTimesY() { in XTimesY()
61 void Init(const std::vector<FunctionDef>& flib) { in Init()
99 FunctionDef fdef = XTimesY(); in TEST_F()
128 FunctionDef fdef = XTimesY(); in TEST_F()
144 FunctionDef fdef = XTimesY(); in TEST_F()
/external/tensorflow/tensorflow/core/kernels/data/
Dhash_utils_test.cc37 uint64 GetHash(const FunctionDefLibrary& library, const FunctionDef& fn) { in GetHash()
51 Status CheckEqual(const FunctionDefLibrary& library, const FunctionDef& fn1, in CheckEqual()
52 const FunctionDef& fn2) { in CheckEqual()
88 FunctionDef* f1 = fl.add_function(); in TEST_F()
96 FunctionDef* f2 = fl.add_function(); in TEST_F()
111 FunctionDef* f1 = fl.add_function(); in TEST_F()
119 FunctionDef* f2 = fl.add_function(); in TEST_F()
137 FunctionDef* f1 = fl.add_function(); in TEST_F()
145 FunctionDef* f2 = fl.add_function(); in TEST_F()
480 FunctionDef* f1 = fl1->add_function(); in TEST_F()
[all …]

12345678910