Home
last modified time | relevance | path

Searched refs:fn_args (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/python/util/
Dfunction_utils_test.py42 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))
Dfunction_utils.py39 def fn_args(fn): function
52 args = fn_args(fn.func)
/external/tensorflow/tensorflow/contrib/learn/python/learn/
Dmetric_spec.py67 fn_name, fn_args, candidate_args, canonical_arg, is_required=False): argument
87 matching_args = candidate_args.intersection(fn_args)
97 canonical_arg, fn_name, fn_args, matching_arg)
100 '%s missing from %s(%s).' % (candidate_args, fn_name, fn_args))
Dexperiment.py53 pred_fn_args = function_utils.fn_args(predicate_fn)
/external/tensorflow/tensorflow/contrib/compiler/
Dxla.py662 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/bcc/src/cc/
Dbpf_module.cc350 vector<Type *> fn_args({B.getInt8PtrTy(), PointerType::getUnqual(type)}); in make_reader() local
351 FunctionType *fn_type = FunctionType::get(B.getInt32Ty(), fn_args, /*isVarArg=*/false); in make_reader()
410 vector<Type *> fn_args({B.getInt8PtrTy(), B.getInt64Ty(), PointerType::getUnqual(type)}); in make_writer() local
411 FunctionType *fn_type = FunctionType::get(B.getInt32Ty(), fn_args, /*isVarArg=*/false); in make_writer()
/external/tensorflow/tensorflow/contrib/eager/python/
Dnetwork.py548 args = function_utils.fn_args(layer_func.call)
551 args = function_utils.fn_args(layer_func)
/external/tensorflow/tensorflow/python/tpu/
Dtpu_estimator.py1664 model_fn_args = function_utils.fn_args(self._model_fn)
1762 fn_args = function_utils.fn_args(host_call[0])
1764 if fullargspec.varargs is None and len(host_call[1]) != len(fn_args):
1768 name, len(host_call[1]), len(fn_args)))
2640 input_fn_args = function_utils.fn_args(input_fn)
/external/tensorflow/tensorflow/python/layers/
Dbase.py531 self._call_fn_args = function_utils.fn_args(self.call)
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer.py182 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/
Dmonitored_session.py779 step_fn_arguments = function_utils.fn_args(step_fn)
/external/tensorflow/tensorflow/python/ops/
Dvariable_scope.py547 if ("constraint" in function_utils.fn_args(custom_getter) or