| /external/tensorflow/tensorflow/python/util/ |
| D | function_utils_test.py | 24 from tensorflow.python.util import function_utils
42 self.assertEqual(('a', 'b'), function_utils.fn_args(fn))
51 self.assertEqual(('a', 'b'), function_utils.fn_args(Foo()))
60 self.assertEqual(('a', 'b'), function_utils.fn_args(Foo().bar))
72 self.assertEqual(('a',), function_utils.fn_args(wrapped_fn))
84 self.assertEqual(('a',), function_utils.fn_args(wrapped_fn))
101 self.assertEqual(('a',), function_utils.fn_args(double_wrapped_fn))
115 self.assertEqual(('a',), function_utils.fn_args(double_wrapped_fn))
132 self.assertEqual(('a',), function_utils.fn_args(double_wrapped_fn))
143 self.assertTrue(function_utils.has_kwargs(fn_has_kwargs))
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | vectorization_utils_test.cc | 171 !function_utils::ContainsFunctionNodeWithOp("MapDefun", *vectorized)); in TEST()
251 function_utils::ContainsFunctionNodeWithOp("MapDefun", *vectorized)); in TEST()
253 function_utils::FindFunctionNodeWithOp("MapDefun", *vectorized)); in TEST()
256 ASSERT_TRUE(function_utils::ContainsFunctionNodeWithOp("Cast", *vectorized)); in TEST()
258 function_utils::FindFunctionNodeWithOp("Cast", *vectorized)); in TEST()
325 !function_utils::ContainsFunctionNodeWithOp("MapDefun", *vectorized)); in TEST()
327 function_utils::FindFunctionNodeWithOp("Cast", *vectorized)); in TEST()
406 !function_utils::ContainsFunctionNodeWithOp("MapDefun", *vectorized)); in TEST()
408 function_utils::FindFunctionNodeWithOp("Cast", *vectorized)); in TEST()
411 function_utils::FindFunctionNodeWithOp("Unpack", *vectorized)); in TEST()
[all …]
|
| D | BUILD | 100 ":function_utils",
119 ":function_utils",
131 name = "function_utils",
132 srcs = ["function_utils.cc"],
134 "function_utils.h",
150 ":function_utils",
219 ":function_utils",
467 ":function_utils",
503 ":function_utils",
525 ":function_utils",
[all …]
|
| D | fusion_utils_test.cc | 114 function_utils::ContainsFunctionNodeWithOp("Equal", *fused_function)); in TEST()
116 function_utils::FindFunctionNodeWithOp("Equal", *fused_function)); in TEST()
|
| D | hoist_random_uniform.cc | 164 function_utils::FindFunctionNodeWithOp("RandomUniform", in MakeLessStatefulFunction()
194 if (!function_utils::IsNodeStateful(library, node, true)) { in CanHoistRandomUniform()
|
| D | function_utils.h | 32 namespace function_utils {
|
| D | map_parallelization.cc | 75 if (function_utils::IsFunctionStateful(function_library, *function, true)) in OptimizeAndCollectStats()
|
| D | map_vectorization.cc | 72 function_utils::SetUniqueFunctionNodeName(map_defun_node->op(), in CreateMapDefunWrapper()
349 function_utils::SetUniqueFunctionNodeName(node->name(), branch_func, in AddBranch()
499 if (function_utils::IsFunctionStateful(function_library, *map_fn)) { in FindMapAndBatchPattern()
|
| D | function_utils.cc | 26 namespace function_utils { namespace
|
| D | function_utils_test.cc | 27 namespace function_utils { namespace
|
| D | fusion_utils.cc | 411 function_utils::SetUniqueFunctionNodeName("cond", fused_function, if_node); in LazyConjunctionNodes()
|
| /external/tensorflow/tensorflow/contrib/compiler/ |
| D | xla.py | 36 from tensorflow.python.util import function_utils
662 eval_metric_fn_args = function_utils.fn_args(eval_metric_fn)
675 model_fn_args = function_utils.fn_args(self._model_fn)
722 eval_metric_fn_args = function_utils.fn_args(eval_metric_fn)
|
| /external/swiftshader/third_party/SPIRV-Tools/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/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/python/ops/ |
| D | functional_ops.py | 40 from tensorflow.python.util import function_utils
814 config = function_utils.get_disabled_rewriter_config()
847 s=function_utils.get_disabled_rewriter_config())
|
| D | variable_scope.py | 45 from tensorflow.python.util import function_utils
547 if ("constraint" in function_utils.fn_args(custom_getter) or
548 function_utils.has_kwargs(custom_getter)):
|
| /external/tensorflow/tensorflow/contrib/eager/python/ |
| D | network.py | 35 from tensorflow.python.util import function_utils
548 args = function_utils.fn_args(layer_func.call)
551 args = function_utils.fn_args(layer_func)
|
| /external/tensorflow/tensorflow/python/layers/ |
| D | base.py | 30 from tensorflow.python.util import function_utils
531 self._call_fn_args = function_utils.fn_args(self.call)
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| D | experiment.py | 47 from tensorflow.python.util import function_utils
53 pred_fn_args = function_utils.fn_args(predicate_fn)
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | function.py | 40 from tensorflow.python.util import function_utils
374 base_func_name = function_utils.get_func_name(self._func)
876 name = function_utils.get_func_name(func)
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | graph_rewrite_dataset.cc | 105 tensorflow::grappler::function_utils::SetUniqueFunctionNodeName( in AddFakeSinks()
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_layer.py | 57 from tensorflow.python.util import function_utils
182 self._call_fn_args = function_utils.fn_args(self.call)
563 self._call_fn_args = function_utils.fn_args(self.call)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | monitored_session.py | 47 from tensorflow.python.util import function_utils
779 step_fn_arguments = function_utils.fn_args(step_fn)
|