Home
last modified time | relevance | path

Searched refs:graph_function (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/eager/
Dfunction.py1269 graph_function, args, kwargs = self._maybe_define_function(args, kwargs)
1270 return graph_function._filtered_call(args, kwargs) # pylint: disable=protected-access
1295 graph_function, _, _ = self._maybe_define_function(args, kwargs)
1296 return graph_function
1300 graph_function = self._get_concrete_function_internal_garbage_collected(
1306 graph_function._garbage_collector.release() # pylint: disable=protected-access
1307 return graph_function
1338 graph_function, args, kwargs = self._maybe_define_function(args, kwargs)
1343 captured = frozenset(graph_function.graph.internal_captures)
1357 graph_function._num_positional_args = allowed_positional
[all …]
Dfunction_test.py613 graph_function = function.Function(
616 self.assertAllEqual(graph_function(), 1.0)
618 self.assertAllEqual(graph_function(), 1.0)
628 graph_function = function.Function(
631 self.assertAllEqual(graph_function(), 1.0)
633 self.assertAllEqual(graph_function(), 1.0)
2099 graph_function = foo.get_concrete_function(v)
2100 self.assertLen(graph_function.inputs, 1)
2101 self.assertEmpty(graph_function.captured_inputs)
2103 self.assertEqual(float(graph_function(v)), 1.0)
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dvariable_scope_test.py48 def run_inside_wrap_function_in_eager_mode(graph_function): argument
63 wrapped = wrap_function.wrap_function(graph_function, [self])
68 graph_function(self)
/external/tensorflow/tensorflow/contrib/distribute/python/
Dmirrored_strategy_multigpu_test.py1323 graph_function = per_replica_graph_functions.get(device=device)
1325 set(graph_function.graph.variables))