Home
last modified time | relevance | path

Searched refs:call_kwargs (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/keras/engine/
Dnode.py56 call_kwargs=None, argument
59 call_kwargs = {} if call_kwargs is None else call_kwargs
63 self.is_input = not call_args and not call_kwargs
72 call_kwargs = nest.map_structure(lambda t: t, call_kwargs)
75 self.call_kwargs = call_kwargs
78 self._flat_arguments = nest.flatten((self.call_args, self.call_kwargs))
80 self._single_positional_tensor_passed = (not self.call_kwargs and len(
157 args, kwargs = nest.pack_sequence_as((self.call_args, self.call_kwargs),
164 args, kwargs = self.call_args, self.call_kwargs
Dnode_test.py68 self.assertEqual(node_a.call_kwargs, {})
130 call_kwargs={'x': kwarg_x, 'y': kwarg_y},
137 self.assertEqual({'x': kwarg_x, 'y': kwarg_y}, node.call_kwargs)
Dbase_layer.py2654 node_module.Node(self, call_args=args, call_kwargs=kwargs, outputs=outputs)
/external/rust/crates/grpcio-sys/grpc/tools/run_tests/xds_k8s_test_driver/framework/rpc/
Dgrpc.py59 call_kwargs = dict(wait_for_ready=True, timeout=timeout_sec)
60 self._log_rpc_request(rpc, req, call_kwargs, log_level)
61 return rpc_callable(req, **call_kwargs)
63 def _log_rpc_request(self, rpc, req, call_kwargs, log_level=logging.DEBUG): argument
67 ', '.join({f'{k}={v}' for k, v in call_kwargs.items()}))
/external/tensorflow/tensorflow/python/debug/lib/
Dcheck_numerics_callback_test.py126 (_, call_kwargs) = fake_get_check_numerics_error_message.call_args
127 self.assertEqual(call_kwargs["stack_height_limit"], 123)
128 self.assertEqual(call_kwargs["path_length_limit"], 1200)
407 (_, call_kwargs) = fake_get_check_numerics_error_message.call_args
408 self.assertEqual(call_kwargs["stack_height_limit"], 123)
409 self.assertEqual(call_kwargs["path_length_limit"], 1200)
/external/tensorflow/tensorflow/python/eager/
Dfunction_test.py3718 call_kwargs=lambda: {'c': constant_op.constant(1.0)},
3781 call_kwargs=lambda: {'x': 3},
3790 call_kwargs=None, argument
3806 call_kwargs = call_kwargs() if callable(call_kwargs) else call_kwargs or {}
3810 self.assertIsInstance(call_kwargs, dict)
3819 self.evaluate(conc(*call_args, **call_kwargs))
3832 call_kwargs=lambda: {
3847 call_kwargs=lambda: {'c': constant_op.constant(1)},
3880 call_kwargs=lambda: {'c': constant_op.constant(1)},
3889 call_kwargs=None, argument
[all …]
/external/tensorflow/tensorflow/python/keras/
Dmodels.py119 node.call_kwargs)