Home
last modified time | relevance | path

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

1234567

/external/tensorflow/tensorflow/core/framework/
Dfunction_testlib.h58 gtl::ArraySlice<FunctionDef> funcs = {});
64 FunctionDef XTimesTwo();
67 FunctionDef TwoDeviceTimesFive();
70 FunctionDef TwoDeviceMult();
73 FunctionDef TwoDeviceInputOutput();
76 FunctionDef FuncWithListInput();
79 FunctionDef FuncWithListOutput();
82 FunctionDef XAddX();
85 FunctionDef XTimesTwoInt32();
88 FunctionDef XTimesFour();
[all …]
Dfunction_testlib.cc32 gtl::ArraySlice<FunctionDef> funcs) { in GDef()
60 FunctionDef NonZero() { in NonZero()
76 FunctionDef IsZero() { in IsZero()
94 FunctionDef RandomUniform() { in RandomUniform()
119 FunctionDef XTimesTwo() { in XTimesTwo()
138 FunctionDef TwoDeviceMult() { in TwoDeviceMult()
178 FunctionDef TwoDeviceInputOutput() { in TwoDeviceInputOutput()
210 FunctionDef FuncWithListInput() { in FuncWithListInput()
228 FunctionDef FuncWithListOutput() { in FuncWithListOutput()
246 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()
Dfunction.h132 static FunctionDef Create(
143 static FunctionDef Create(const string& function_name,
151 static FunctionDef Define(const string& function_name,
158 static FunctionDef Define(gtl::ArraySlice<string> arg_def,
229 Status InstantiateFunction(const FunctionDef& fdef, AttrSlice attr_values,
241 string DebugString(const FunctionDef& func_def);
251 bool FunctionDefsEqual(const FunctionDef& f1, const FunctionDef& f2);
256 uint64 FunctionDefHash(const FunctionDef& fdef);
331 const FunctionDef* Find(const string& func) const LOCKS_EXCLUDED(mu_);
339 Status AddFunctionDef(const FunctionDef& fdef) LOCKS_EXCLUDED(mu_);
[all …]
/external/tensorflow/tensorflow/core/ops/
Dmath_grad.cc25 Status GradForUnaryCwise(FunctionDef* g, std::vector<FDH::Node> nodes) { in GradForUnaryCwise()
43 Status AbsGrad(const AttrSlice& attrs, FunctionDef* g) { in AbsGrad()
53 Status NegGrad(const AttrSlice& attrs, FunctionDef* g) { in NegGrad()
62 Status InvGrad(const AttrSlice& attrs, FunctionDef* g) { in InvGrad()
75 Status SquareGrad(const AttrSlice& attrs, FunctionDef* g) { in SquareGrad()
87 Status SqrtGrad(const AttrSlice& attrs, FunctionDef* g) { in SqrtGrad()
101 Status RsqrtGrad(const AttrSlice& attrs, FunctionDef* g) { in RsqrtGrad()
116 Status ExpGrad(const AttrSlice& attrs, FunctionDef* g) { in ExpGrad()
126 Status Expm1Grad(const AttrSlice& attrs, FunctionDef* g) { in Expm1Grad()
136 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* function, DataType dt);
70 FunctionDef* fd);
74 const FunctionDef& function);
77 bool ContainsFunctionNodeWithOp(StringPiece op, const FunctionDef& function);
81 const FunctionDef& function);
85 int FindFunctionInputWithName(StringPiece name, const FunctionDef& function);
89 int FindFunctionOutputWithName(StringPiece name, const FunctionDef& function);
93 int FindFunctionNodeWithName(StringPiece name, const FunctionDef& function);
97 int FindFunctionNodeWithOp(StringPiece op, 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* function, DataType dt) { in AddFunctionOutputWithUniqueName()
108 FunctionDef* fd) { in AddNode()
126 const FunctionDef& function) { in ContainsFunctionNodeWithName()
130 bool ContainsFunctionNodeWithOp(StringPiece op, const FunctionDef& function) { in ContainsFunctionNodeWithOp()
135 const FunctionDef& function) { in ContainsFunctionOutputWithName()
139 int FindFunctionInputWithName(StringPiece name, const FunctionDef& function) { in FindFunctionInputWithName()
145 int FindFunctionOutputWithName(StringPiece name, const FunctionDef& function) { in FindFunctionOutputWithName()
151 int FindFunctionNodeWithName(StringPiece name, const FunctionDef& function) { in FindFunctionNodeWithName()
157 int FindFunctionNodeWithOp(StringPiece op, const FunctionDef& function) { in FindFunctionNodeWithOp()
[all …]
Dfunction_utils_test.cc45 FunctionDef outer = FunctionDefHelper::Create( in TEST()
57 FunctionDef function = test::function::XTimesTwo(); in TEST()
64 FunctionDef function = test::function::XTimesTwo(); in TEST()
71 FunctionDef function = test::function::XTimesTwo(); in TEST()
78 FunctionDef function = test::function::XTimesTwo(); in TEST()
84 FunctionDef function = test::function::XTimesTwo(); in TEST()
92 FunctionDef function = test::function::XTimesTwo(); in TEST()
100 FunctionDef function = test::function::XTimesTwo(); in TEST()
106 FunctionDef function = test::function::XTimesTwo(); in TEST()
112 FunctionDef function = test::function::XTimesTwo(); in TEST()
[all …]
Dfusion_utils.cc50 string GetOutputNode(const FunctionDef& function, int output_idx) { in GetOutputNode()
56 string& GetMutableOutputNode(FunctionDef* function, int output_idx) { in GetMutableOutputNode()
115 void RenameFunctionNodes(const FunctionDef& first_function, in RenameFunctionNodes()
143 StringCollection GetFunctionInputs(const FunctionDef& function) { in GetFunctionInputs()
250 StringCollection GetFunctionOutputs(const FunctionDef& function) { in GetFunctionOutputs()
260 FunctionDef* CreateFalsePredicate( in CreateFalsePredicate()
297 void MergeNodes(const FunctionDef& first_function, in MergeNodes()
298 const FunctionDef& second_function, FunctionDef* fused_function, in MergeNodes()
376 void LazyConjunctionNodes(const FunctionDef& first_function, in LazyConjunctionNodes()
377 const FunctionDef& second_function, in LazyConjunctionNodes()
[all …]
Dmap_vectorization.cc59 FunctionDef* CreateMapDefunWrapper(const NodeDef& map_node, in CreateMapDefunWrapper()
60 const FunctionDef& orig_func, in CreateMapDefunWrapper()
62 FunctionDef* vectorized_func = library->add_function(); in CreateMapDefunWrapper()
112 FunctionDef* AddVectorizedFunction(const NodeDef& map_node, in AddVectorizedFunction()
113 const FunctionDef& orig_func, in AddVectorizedFunction()
117 FunctionDef* vectorized_func = in AddVectorizedFunction()
122 FunctionDef* result; in AddVectorizedFunction()
192 const FunctionDef& vectorized_func, in AddNewBatchNode()
252 const FunctionDef& vectorized_func, in AddNewMapNode()
325 FunctionDef* branch_func = library->add_function(); in AddBranch()
[all …]
Dvectorization_utils.cc143 Status Vectorize(const FunctionDef& outer_scope,
144 const NodeDef& map_defun_node, FunctionDef** result);
149 Status Initialize(const FunctionDef& outer_scope,
153 Status GetResult(FunctionDef** vectorized_function);
332 Status Vectorization::Vectorize(const FunctionDef& outer_scope, in Vectorize()
334 FunctionDef** result) { in Vectorize()
367 Status Vectorization::Initialize(const FunctionDef& outer_scope, in Initialize()
371 const FunctionDef* map_defun_fn = in Initialize()
616 Status Vectorization::GetResult(FunctionDef** vectorized_function) { in GetResult()
622 FunctionDef* map_defun_fn = lib_->add_function(); in GetResult()
[all …]
Dfilter_fusion.cc38 const FunctionDef& fused_function, in MakeFusedFilterNode()
81 const NodeDef* second_filter_node) -> FunctionDef* { in OptimizeAndCollectStats()
83 const FunctionDef* first_func = in OptimizeAndCollectStats()
86 const FunctionDef* second_func = function_library.Find(fun.func().name()); in OptimizeAndCollectStats()
Dhoist_random_uniform.cc40 const FunctionDef& stateless_function, in MakeStatelessMap()
149 const FunctionDef* MakeLessStatefulFunction(const FunctionDef& map_function, in MakeLessStatefulFunction()
153 FunctionDef* stateless_function = library->add_function(); in MakeLessStatefulFunction()
180 bool CanHoistRandomUniform(const FunctionDef& map_function, in CanHoistRandomUniform()
247 const FunctionDef* func = function_library.Find(fun.func().name()); in OptimizeAndCollectStats()
Dmap_fusion.cc40 const FunctionDef& fused_function, in MakeFusedNode()
103 const NodeDef* map_node) -> FunctionDef* { in OptimizeAndCollectStats()
105 const FunctionDef* parent_func = in OptimizeAndCollectStats()
108 const FunctionDef* func = function_library.Find(fun.func().name()); in OptimizeAndCollectStats()
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.h182 friend Status MakeGrapplerFunctionItem(const FunctionDef&, const AttrSlice&,
212 bool HasParametrizedType(const FunctionDef& func);
217 bool HasParametrizedBody(const FunctionDef& func);
220 bool IsParametrized(const FunctionDef& func);
225 const FunctionDef& func, const AttrSlice& func_instantiation_attr,
232 const FunctionDef& func, const AttrSlice& func_instantiation_attr,
260 Status MakeGrapplerFunctionItem(const FunctionDef& func,
271 Status MakeGrapplerFunctionItem(const FunctionDef& func,
280 FunctionDef* func);
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()
226 FunctionDef func = FunctionDefHelper::Define( in TEST_F()
312 FunctionDef func = FunctionDefHelper::Define( in TEST_F()
395 FunctionDef func = FunctionDefHelper::Create( in TEST_F()
464 FunctionDef func = FunctionDefHelper::Create( in TEST_F()
514 FunctionDef func = FunctionDefHelper::Create( in TEST_F()
579 FunctionDef func = FunctionDefHelper::Define( in TEST_F()
623 FunctionDef func = FunctionDefHelper::Define( in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer_test.cc258 FunctionDef mul_func = FunctionDefHelper::Create( in TEST_F()
264 FunctionDef square_func = FunctionDefHelper::Create( in TEST_F()
271 FunctionDef quadratic_func = FunctionDefHelper::Create( in TEST_F()
330 const FunctionDef* optimized_func_0 = optimized_flib.Find(optimized_0); in TEST_F()
331 const FunctionDef* optimized_func_1 = optimized_flib.Find(optimized_1); in TEST_F()
332 const FunctionDef* optimized_func_2 = optimized_flib.Find(optimized_2); in TEST_F()
360 const std::vector<const FunctionDef*> optimized_funcs = {optimized_func_1, in TEST_F()
364 for (const FunctionDef* optimized_func : optimized_funcs) { in TEST_F()
414 FunctionDef my_mul = FunctionDefHelper::Create( in TEST_F()
423 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()
Dfunction_optimizer_test.cc124 FunctionDef x_times_two = test::function::XTimesTwo(); in TEST_F()
127 FunctionDef my_identity_n = FunctionDefHelper::Create( in TEST_F()
202 FunctionDef x_times_two = FunctionDefHelper::Define( in TEST_F()
299 FunctionDef func = FunctionDefHelper::Create( in TEST_F()
383 FunctionDef func = FunctionDefHelper::Create( in TEST_F()
437 FunctionDef func = FunctionDefHelper::Define( in TEST_F()
474 FunctionDef mul_func = FunctionDefHelper::Create( in TEST_F()
480 FunctionDef square_func = FunctionDefHelper::Create( in TEST_F()
565 FunctionDef func = FunctionDefHelper::Define( in TEST_F()
607 FunctionDef func = FunctionDefHelper::Create( in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/kernels/data/
Dmap_dataset_op_test.cc93 std::vector<FunctionDef> input_func_lib) in GetNextTestParams()
106 std::vector<FunctionDef> func_lib;
161 std::vector<FunctionDef>{test::function::XTimesTwo()}),
163 std::vector<FunctionDef>{test::function::XAddX()}),
166 std::vector<FunctionDef>{test::function::XTimesTwo(),
172 FunctionDef func_def = test::function::XTimesTwo(); in TEST_F()
199 FunctionDef func_def = test::function::XTimesTwo(); in TEST_F()
227 FunctionDef func_def = test::function::XTimesTwo(); in TEST_F()
278 FunctionDef func_def = test::function::XTimesTwo(); in TEST_P()
311 FunctionDef func_def = test::function::XTimesTwo(); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/jit/
Dcreate_xla_launch_op_test.cc39 FunctionDef XTimesY() { in XTimesY()
58 void Init(const std::vector<FunctionDef>& flib) { in Init()
95 FunctionDef fdef = XTimesY(); in TEST_F()
120 FunctionDef fdef = XTimesY(); in TEST_F()
133 FunctionDef fdef = XTimesY(); in TEST_F()

1234567