| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| map_and_filter_fusion.cc | 38 const FunctionDef& fused_function,
124 const NodeDef* filter_node) -> FunctionDef* {
126 const FunctionDef* map_func =
129 const FunctionDef* filter_func = function_library.Find(fun.func().name());
|
| map_fusion.cc | 40 const FunctionDef& fused_function,
103 const NodeDef* map_node) -> FunctionDef* {
105 const FunctionDef* parent_func =
108 const FunctionDef* func = function_library.Find(fun.func().name());
|
| hoist_random_uniform.cc | 40 const FunctionDef& stateless_function,
149 const FunctionDef* MakeLessStatefulFunction(const FunctionDef& map_function,
153 FunctionDef* stateless_function = library->add_function();
180 bool CanHoistRandomUniform(const FunctionDef& map_function,
247 const FunctionDef* func = function_library.Find(fun.func().name());
|
| map_vectorization.cc | 57 // Returns a FunctionDef containing a MapDefun op that wraps the original
59 FunctionDef* CreateMapDefunWrapper(const NodeDef& map_node,
60 const FunctionDef& orig_func,
62 FunctionDef* vectorized_func = library->add_function();
112 FunctionDef* AddVectorizedFunction(const NodeDef& map_node,
113 const FunctionDef& orig_func,
117 FunctionDef* vectorized_func =
122 FunctionDef* result;
192 const FunctionDef& vectorized_func,
252 const FunctionDef& vectorized_func
[all...] |
| vectorization_utils.cc | 141 // the conversion between FunctionDef -> Graph -> FunctionDef failed anywhere
143 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,
334 FunctionDef** result) {
367 Status Vectorization::Initialize(const FunctionDef& outer_scope,
371 const FunctionDef* map_defun_fn
[all...] |
| map_vectorization_test.cc | 58 FunctionDef* AddMapFn(MutableGraphView* graph) {
59 FunctionDef* map_fn = graph->graph()->mutable_library()->add_function();
175 void CheckBranch(const FunctionDef& function, gtl::ArraySlice<string> ops) {
181 const FunctionDef* GetFunction(const GraphDef& graph,
214 const FunctionDef* branch_0 =
220 const FunctionDef* branch_1 =
231 const FunctionDef* function = GetFunction(output, function_name);
383 const FunctionDef* branch_0 =
389 const FunctionDef* branch_1 =
|
| graph_utils_test.cc | 119 FunctionDef* new_function = library.add_function();
153 FunctionDef* new_function = library.add_function();
211 FunctionDef* new_function = library.add_function();
214 FunctionDef* other_function = library.add_function();
|
| rebatch.cc | 206 const FunctionDef* fdef = flib->Find(func_name);
215 for (const FunctionDef& func_def :
222 // Convert optimized graph back to FunctionDef.
223 FunctionDef optimized_func;
227 // Replace optimized function with a new FunctionDef.
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| grappler_test.h | 70 void CompareFunctions(FunctionDef want, FunctionDef got) const;
|
| functions_test.cc | 39 FunctionDef parametrized_func = FunctionDefHelper::Create(
46 FunctionDef non_parametrized_func = FunctionDefHelper::Create(
63 FunctionDef func = FunctionDefHelper::Create(
226 FunctionDef func = FunctionDefHelper::Define(
312 FunctionDef func = FunctionDefHelper::Define(
395 FunctionDef func = FunctionDefHelper::Create(
464 FunctionDef func = FunctionDefHelper::Create(
514 FunctionDef func = FunctionDefHelper::Create(
579 FunctionDef func = FunctionDefHelper::Define(
623 FunctionDef func = FunctionDefHelper::Define
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| meta_optimizer_test.cc | 258 FunctionDef mul_func = FunctionDefHelper::Create(
264 FunctionDef square_func = FunctionDefHelper::Create(
271 FunctionDef quadratic_func = FunctionDefHelper::Create(
330 const FunctionDef* optimized_func_0 = optimized_flib.Find(optimized_0);
331 const FunctionDef* optimized_func_1 = optimized_flib.Find(optimized_1);
332 const FunctionDef* optimized_func_2 = optimized_flib.Find(optimized_2);
360 const std::vector<const FunctionDef*> optimized_funcs = {optimized_func_1,
364 for (const FunctionDef* optimized_func : optimized_funcs) {
414 FunctionDef my_mul = FunctionDefHelper::Create(
423 FunctionDef my_fwd = FunctionDefHelper::Create
[all...] |
| function_api_info.h | 42 Status Init(const FunctionDef& function_def);
|
| implementation_selector_test.cc | 141 FunctionDef boost_1 = FDH::Create(
152 FunctionDef boost_1_gradient = FDH::Create(
164 FunctionDef boost_2_func = FDH::Create(
177 FunctionDef boost_2_gradient = FDH::Create(
|
| function_optimizer_test.cc | 124 FunctionDef x_times_two = test::function::XTimesTwo();
127 FunctionDef my_identity_n = FunctionDefHelper::Create(
202 FunctionDef x_times_two = FunctionDefHelper::Define(
299 FunctionDef func = FunctionDefHelper::Create(
383 FunctionDef func = FunctionDefHelper::Create(
437 FunctionDef func = FunctionDefHelper::Define(
474 FunctionDef mul_func = FunctionDefHelper::Create(
480 FunctionDef square_func = FunctionDefHelper::Create(
565 FunctionDef func = FunctionDefHelper::Define(
607 FunctionDef func = FunctionDefHelper::Create
[all...] |
| function_optimizer.cc | 84 bool AttrIsTrue(const FunctionDef& func, const string& attr) {
88 bool MarkedSpecialized(const FunctionDef& func) {
92 bool MarkedNoInline(const FunctionDef& func) {
104 // Check if func_node.op() matches the name in FunctionDef signature.
105 bool IsDirectFunctionCall(const FunctionDef& func, const NodeDef& func_node) {
110 // FunctionDef signature.
111 bool IsIndirectFunctionCall(const FunctionDef& func, const NodeDef& func_node) {
117 AttrSlice FunctionInstantiationAttributes(const FunctionDef& func,
158 // FunctionDef is somewhat similar to function template in C++, given all the
162 // Function specialization instantiates a parametrized FunctionDef into
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| function_testlib.cc | 47 FunctionDef FindDevice() {
100 FunctionDef BlockingOpFn() {
|
| optimization_registry_test.cc | 73 const FunctionDef* GetFunctionDef(const string& func) const {
|
| shape_refiner.h | 214 Status InferShapesForFunction(const tensorflow::FunctionDef* function_def,
298 std::unordered_map<const FunctionDef*, std::unique_ptr<const Graph>>
|
| /external/tensorflow/tensorflow/core/framework/ |
| graph_def_util.cc | 151 FunctionDef* func_def = graph_def->mutable_library()->mutable_function(f);
169 std::unordered_map<string, const FunctionDef*> name_to_function;
178 std::vector<const FunctionDef*> functions_to_process; // A subset of used_ops
194 const FunctionDef* fun = functions_to_process.back();
|
| function_test.cc | 1033 FunctionDef fdef = test::function::XTimesTwo();
[all...] |
| function.cc | 541 string Print(const FunctionDef& fdef) {
682 Status InstantiateFunction(const FunctionDef& fdef, AttrSlice attr_values,
766 string DebugString(const FunctionDef& func_def) { return Print(func_def); }
801 std::map<string, AttrValue> GetSetAttrs(const FunctionDef& fdef) {
813 bool FunctionDefsEqual(const FunctionDef& f1, const FunctionDef& f2) {
842 uint64 FunctionDefHash(const FunctionDef& fdef) {
[all...] |
| /external/tensorflow/tensorflow/core/util/ |
| dump_graph_test.cc | 55 FunctionDef fdef;
|
| /external/tensorflow/tensorflow/python/autograph/pyct/ |
| pretty_printer_test.py | 39 node = ast.FunctionDef(
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| extract_outside_compilation_pass_test.cc | 285 FunctionDef *xla_fdef = fdl.add_function();
413 FunctionDef *xla_fdef = fdl.add_function();
461 FunctionDef *xla_fdef = fdl.add_function();
550 FunctionDef *true_fn_fdef = fdl.add_function();
566 FunctionDef *false_fn_fdef = fdl.add_function();
584 FunctionDef *xla_fdef = fdl.add_function();
631 const FunctionDef *true_def = fld.Find("oc_then_branch_host_if_if");
643 const FunctionDef *false_def = fld.Find("oc_else_branch_host_if_if");
710 FunctionDef *cond_fn_fdef = fdl.add_function();
726 FunctionDef *body_fn_fdef = fdl.add_function()
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| context.h | 139 const FunctionDef* FindFunctionDef(const string& name);
151 Status AddFunctionDef(const FunctionDef& fdef);
248 Status MaybeRegisterFunctionRemotely(const FunctionDef& fdef);
|
