| /external/tensorflow/tensorflow/core/framework/ |
| function_testlib.h | 58 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...] |
| graph_to_functiondef.h | 25 // Converts 'graph' to a FunctionDef 'fdef', with name 'name'.
28 FunctionDef* fdef);
|
| function_testlib.cc | 32 gtl::ArraySlice<FunctionDef> funcs) {
60 FunctionDef NonZero() {
76 FunctionDef IsZero() {
94 FunctionDef RandomUniform() {
119 FunctionDef XTimesTwo() {
138 FunctionDef TwoDeviceMult() {
178 FunctionDef TwoDeviceInputOutput() {
210 FunctionDef FuncWithListInput() {
228 FunctionDef FuncWithListOutput() {
246 FunctionDef XAddX()
[all...] |
| graph_to_functiondef_test.cc | 32 FunctionDef RemoveDebugInfo(const FunctionDef& def) {
33 FunctionDef copy = def;
40 bool EqualFunctionDef(const FunctionDef& a, const FunctionDef& b,
72 FunctionDef fdef;
75 FunctionDef fdef_expected = FunctionDefHelper::Create(
109 FunctionDef fdef;
112 FunctionDef fdef_expected = FunctionDefHelper::Create(
|
| function.h | 52 // FunctionDef proto.
54 // FunctionDef my_func = FunctionDefHelper::Create(
68 // FunctionDef my_func = FunctionDefHelper::Define(
110 // Node is used to construct FunctionDef.Node using initialization
126 // Creates a FunctionDef from the given parameters. Node inputs must use
132 static FunctionDef Create(
139 // Creates a FunctionDef from the given parameters. Node inputs must use
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
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| vectorization_utils.h | 27 // Given a MapDefun node (`map_defun_node`) in a FunctionDef (`outer_scope`)
89 Status VectorizeMapDefun(const FunctionDef& outer_scope,
91 FunctionDef** result);
|
| function_utils.h | 36 // Describes a FunctionDef input tensor. In FunctionDefs, input tensor strings
58 void ReplaceReferences(const string& from, const string& to, FunctionDef* func);
64 FunctionDef* function, DataType dt);
66 // Adds a node to a FunctionDef.
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)
[all...] |
| fusion_utils.h | 59 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
[all...] |
| vectorization_utils_test.cc | 50 Status WrapFunctionWithMapDefun(const FunctionDef& inner, FunctionDef* result) {
96 Status WrapAndVectorize(const FunctionDef& fn, FunctionDefLibrary* lib,
97 FunctionDef** result) {
98 FunctionDef outer;
118 string GetRetval(const FunctionDef& function_def, int index) {
159 FunctionDef inner = FunctionDefHelper::Create(
167 FunctionDef* vectorized;
234 FunctionDef inner = FunctionDefHelper::Create(
247 FunctionDef* vectorized
[all...] |
| function_utils.cc | 74 // FunctionDef -> GraphDef conversion isn't really necessary in this case.
76 FunctionDef* func) {
91 FunctionDef* function, DataType dt) {
108 FunctionDef* fd) {
126 const FunctionDef& function) {
130 bool ContainsFunctionNodeWithOp(StringPiece op, const FunctionDef& function) {
135 const FunctionDef& function) {
139 int FindFunctionInputWithName(StringPiece name, const FunctionDef& function) {
145 int FindFunctionOutputWithName(StringPiece name, const FunctionDef& function) {
151 int FindFunctionNodeWithName(StringPiece name, const FunctionDef& function)
[all...] |
| filter_fusion.cc | 38 const FunctionDef& fused_function,
81 const NodeDef* second_filter_node) -> FunctionDef* {
83 const FunctionDef* first_func =
86 const FunctionDef* second_func = function_library.Find(fun.func().name());
|
| function_utils_test.cc | 45 FunctionDef outer = FunctionDefHelper::Create(
57 FunctionDef function = test::function::XTimesTwo();
64 FunctionDef function = test::function::XTimesTwo();
71 FunctionDef function = test::function::XTimesTwo();
78 FunctionDef function = test::function::XTimesTwo();
84 FunctionDef function = test::function::XTimesTwo();
92 FunctionDef function = test::function::XTimesTwo();
100 FunctionDef function = test::function::XTimesTwo();
106 FunctionDef function = test::function::XTimesTwo();
112 FunctionDef function = test::function::XTimesTwo()
[all...] |
| fusion_utils.cc | 50 string GetOutputNode(const FunctionDef& function, int output_idx) {
56 string& GetMutableOutputNode(FunctionDef* function, int output_idx) {
115 void RenameFunctionNodes(const FunctionDef& first_function,
143 StringCollection GetFunctionInputs(const FunctionDef& function) {
250 StringCollection GetFunctionOutputs(const FunctionDef& function) {
260 FunctionDef* CreateFalsePredicate(
297 void MergeNodes(const FunctionDef& first_function,
298 const FunctionDef& second_function, FunctionDef* fused_function,
376 void LazyConjunctionNodes(const FunctionDef& first_function
[all...] |
| /external/tensorflow/tensorflow/core/ops/ |
| math_grad.cc | 25 Status GradForUnaryCwise(FunctionDef* g, std::vector<FDH::Node> nodes) {
43 Status AbsGrad(const AttrSlice& attrs, FunctionDef* g) {
53 Status NegGrad(const AttrSlice& attrs, FunctionDef* g) {
62 Status InvGrad(const AttrSlice& attrs, FunctionDef* g) {
75 Status SquareGrad(const AttrSlice& attrs, FunctionDef* g) {
87 Status SqrtGrad(const AttrSlice& attrs, FunctionDef* g) {
101 Status RsqrtGrad(const AttrSlice& attrs, FunctionDef* g) {
116 Status ExpGrad(const AttrSlice& attrs, FunctionDef* g) {
126 Status Expm1Grad(const AttrSlice& attrs, FunctionDef* g) {
136 Status LogGrad(const AttrSlice& attrs, FunctionDef* g)
[all...] |
| nn_grad.cc | 25 Status SoftmaxGrad(const AttrSlice& attrs, FunctionDef* g) {
50 Status LogSoftmaxGrad(const AttrSlice& attrs, FunctionDef* g) {
75 Status ReluGrad(const AttrSlice& attrs, FunctionDef* g) {
93 Status Relu6Grad(const AttrSlice& attrs, FunctionDef* g) {
111 Status CrossEntropyGrad(const AttrSlice& attrs, FunctionDef* g) {
140 Status Conv2DGrad(const AttrSlice& attrs, FunctionDef* g) {
176 Status MaxPoolGrad(const AttrSlice& attrs, FunctionDef* g) {
207 Status AvgPoolGrad(const AttrSlice& attrs, FunctionDef* g) {
233 Status MaxPoolGradGrad(const AttrSlice& attrs, FunctionDef* g) {
264 Status BiasAddGrad(const AttrSlice& attrs, FunctionDef* g)
[all...] |
| array_grad.cc | 33 Status ReshapeGrad(const AttrSlice& attrs, FunctionDef* g) {
54 Status SqueezeGrad(const AttrSlice& attrs, FunctionDef* g) {
73 Status IdentityGrad(const AttrSlice& attrs, FunctionDef* g) {
92 Status PackGrad(const AttrSlice& attrs, FunctionDef* g) {
118 Status UnpackGrad(const AttrSlice& attrs, FunctionDef* g) {
142 Status ConcatGradHelper(const AttrSlice& attrs, FunctionDef* g,
216 Status ConcatGrad(const AttrSlice& attrs, FunctionDef* g) {
220 Status ConcatGradV2(const AttrSlice& attrs, FunctionDef* g) {
227 Status SplitGrad(const AttrSlice& attrs, FunctionDef* g) {
247 Status SplitVGrad(const AttrSlice& attrs, FunctionDef* g)
[all...] |
| functional_grad.cc | 24 Status MapAccumulateGrad(const AttrSlice& attrs, FunctionDef* ret) {
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| function_testlib.h | 26 FunctionDef FindDevice();
42 FunctionDef BlockingOpFn();
|
| function.h | 76 FunctionDef fdef;
85 FunctionBody(const FunctionDef& f, DataTypeSlice arg_types,
167 // a) control returns (`control_ret` field in FunctionDef)
168 // b) data returns (`ret` field in FunctionDef)
221 // side-effects that must always execute (see `control_ret` in FunctionDef).
284 // Instantiates FunctionDef into a graph. Set *fbody to point to the
285 // FunctionBody that holds the instantiated FunctionDef.
287 const FunctionDef& fdef, const AttrSlice& attrs,
|
| /external/tensorflow/tensorflow/core/util/ |
| dump_graph.h | 45 // Similar to DumpGraphDefToFile, but dumps a function as a FunctionDef text
47 string DumpFunctionDefToFile(const string& name, FunctionDef const& fdef,
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| functions.h | 83 // FunctionDef uses different connectivity encoding for the function body nodes,
84 // then a GraphDef (see function.proto for details). Input name in FunctionDef
86 // GraphDef), we need to expand it when converting from FunctionDef to GraphDef,
94 // Expands input encoded in FunctionDef format (name[:output][:position]) into
99 // Updates Node inputs from FunctionDef to GraphDef format.
105 // FunctionDef built from GrapplerFunctionItem is always specialized to its
110 // FunctionDef input format (name[:output][:position]) using registered input
115 // Updates Node inputs from GraphDef to FunctionDef format.
182 friend Status MakeGrapplerFunctionItem(const FunctionDef&, const AttrSlice&,
212 bool HasParametrizedType(const FunctionDef& func)
[all...] |
| /external/tensorflow/tensorflow/compiler/jit/ |
| create_xla_launch_op_test.cc | 38 // Create a FunctionDef that takes one resource and one regular param
39 FunctionDef XTimesY() {
58 void Init(const std::vector<FunctionDef>& flib) {
95 FunctionDef fdef = XTimesY();
120 FunctionDef fdef = XTimesY();
133 FunctionDef fdef = XTimesY();
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| test_util.cc | 26 const FunctionDef* fdef = library.Find(name);
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| function_api_info.cc | 29 Status FunctionApiInfo::Init(const FunctionDef& function_def) {
102 bool IsSameSignature(const FunctionDef& f1, const FunctionDef& f2,
124 const std::vector<const FunctionDef*>& equiv_funcs,
146 const std::unordered_map<string, std::vector<const FunctionDef*>>&
158 std::unordered_map<string, std::vector<const FunctionDef*>> infer_funcs;
159 std::unordered_map<string, std::vector<const FunctionDef*>> fwd_funcs;
160 std::unordered_map<string, std::vector<const FunctionDef*>> bwd_funcs;
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| map_dataset_op_test.cc | 93 std::vector<FunctionDef> input_func_lib)
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();
199 FunctionDef func_def = test::function::XTimesTwo();
227 FunctionDef func_def = test::function::XTimesTwo();
278 FunctionDef func_def = test::function::XTimesTwo();
311 FunctionDef func_def = test::function::XTimesTwo()
[all...] |
