| /external/tensorflow/tensorflow/python/util/ |
| D | function_utils_test.py | 20 from tensorflow.python.util import function_utils
38 self.assertEqual(('a', 'b'), function_utils.fn_args(fn))
47 self.assertEqual(('a', 'b'), function_utils.fn_args(Foo()))
56 self.assertEqual(('a', 'b'), function_utils.fn_args(Foo().bar))
65 self.assertEqual((), function_utils.fn_args(Foo().bar))
77 self.assertEqual(('a',), function_utils.fn_args(wrapped_fn))
89 self.assertEqual(('a',), function_utils.fn_args(wrapped_fn))
106 self.assertEqual(('a',), function_utils.fn_args(double_wrapped_fn))
120 self.assertEqual(('a',), function_utils.fn_args(double_wrapped_fn))
137 self.assertEqual(('a',), function_utils.fn_args(double_wrapped_fn))
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | noop_elimination.cc | 89 auto input = function_utils::FunctionDefTensorDesc(ret_val); in IsOutputIdentityOfInput()
93 while (function_utils::ContainsFunctionNodeWithName(input.node_name, fdef)) { in IsOutputIdentityOfInput()
94 int idx = function_utils::FindFunctionNodeWithName(input.node_name, fdef); in IsOutputIdentityOfInput()
101 input = function_utils::FunctionDefTensorDesc(node.input(0)); in IsOutputIdentityOfInput()
126 if (function_utils::IsFunctionStateful(function_library, *fdef)) return false; in IsMapIdentity()
|
| D | BUILD | 131 ":function_utils",
318 ":function_utils",
358 ":function_utils",
378 ":function_utils",
390 name = "function_utils",
391 srcs = ["function_utils.cc"],
393 "function_utils.h",
410 ":function_utils",
482 ":function_utils",
503 ":function_utils",
[all …]
|
| D | slack_test.cc | 131 function_utils::AddNode("TensorDataset", "TensorDataset", {"args_0"}, in TEST()
135 NodeDef *prefetch_node = function_utils::AddNode( in TEST()
144 NodeDef *identity_node = function_utils::AddNode( in TEST()
|
| D | fusion_utils_test.cc | 114 function_utils::ContainsFunctionNodeWithOp("Equal", *fused_function)); in TEST()
116 function_utils::FindFunctionNodeWithOp("Equal", *fused_function)); in TEST()
|
| D | function_utils.h | 32 namespace function_utils {
|
| D | filter_parallelization.cc | 87 if (function_utils::IsFunctionStateful(function_library, *function, true)) { in OptimizeAndCollectStats()
|
| D | map_parallelization.cc | 88 if (function_utils::IsFunctionStateful(function_library, *function, true)) in OptimizeAndCollectStats()
|
| D | function_utils.cc | 26 namespace function_utils { namespace
|
| D | function_utils_test.cc | 27 namespace function_utils { namespace
|
| D | split_utils_test.cc | 180 int i = function_utils::FindFunctionNodeWithName(name, function); in GetNode()
|
| D | fusion_utils.cc | 412 function_utils::SetUniqueFunctionNodeName("cond", fused_function, if_node); in LazyConjunctionNodes()
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | structured_function.py | 28 from tensorflow.python.util import function_utils
157 function_utils.get_func_name(func)])
|
| /external/angle/third_party/spirv-tools/src/test/opt/dominator_tree/ |
| D | CMakeLists.txt | 17 SRCS ../function_utils.h
|
| /external/deqp-deps/SPIRV-Tools/test/opt/dominator_tree/ |
| D | CMakeLists.txt | 17 SRCS ../function_utils.h
|
| /external/swiftshader/third_party/SPIRV-Tools/test/opt/dominator_tree/ |
| D | CMakeLists.txt | 17 SRCS ../function_utils.h
|
| /external/swiftshader/third_party/SPIRV-Tools/test/opt/loop_optimizations/ |
| D | CMakeLists.txt | 17 SRCS ../function_utils.h
|
| /external/angle/third_party/spirv-tools/src/test/opt/loop_optimizations/ |
| D | CMakeLists.txt | 17 SRCS ../function_utils.h
|
| /external/deqp-deps/SPIRV-Tools/test/opt/loop_optimizations/ |
| D | CMakeLists.txt | 17 SRCS ../function_utils.h
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | BUILD | 195 "function_utils.h",
520 ":function_utils",
790 ":function_utils",
858 name = "function_utils",
859 srcs = ["function_utils.cc"],
860 hdrs = ["function_utils.h"],
915 ":function_utils",
953 ":function_utils",
1018 ":function_utils",
1135 ":function_utils",
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | function.py | 36 from tensorflow.python.util import function_utils
444 base_func_name = function_utils.get_func_name(self._func)
978 name = function_utils.get_func_name(func)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | functional_ops.py | 38 from tensorflow.python.util import function_utils
1166 config = function_utils.get_disabled_rewriter_config()
|
| D | variable_scope.py | 38 from tensorflow.python.util import function_utils
577 if ("constraint" in function_utils.fn_args(custom_getter) or
578 function_utils.has_kwargs(custom_getter)):
|
| /external/tensorflow/tensorflow/core/data/ |
| D | rewrite_utils.cc | 84 tensorflow::grappler::function_utils::SetUniqueFunctionNodeName( in AddFakeSinks()
|
| /external/tensorflow/tensorflow/python/training/ |
| D | monitored_session.py | 40 from tensorflow.python.util import function_utils
823 step_fn_arguments = function_utils.fn_args(step_fn)
|