Searched refs:graph_fn (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | batch_matmul_test.py | 38 def BuildParams(self, graph_fn, dtype, input_shapes, output_shapes): argument
40 graph_fn=graph_fn,
|
| D | tf_trt_integration_test_base.py | 178 def BuildParams(self, graph_fn, dtype, input_shapes, output_shapes): argument
206 return self.BuildParamsWithMask(graph_fn, dtype, input_shapes,
210 def BuildParamsWithMask(self, graph_fn, dtype, input_shapes, output_shapes, argument
252 graph_fn=graph_fn,
934 outputs = params.graph_fn(*inputs)
961 params.graph_fn, input_signature=params.input_specs)
|
| D | lru_cache_test.py | 50 graph_fn=self.GraphFn,
|
| D | dynamic_input_shapes_test.py | 71 graph_fn=self.GraphFn,
|
| /external/tensorflow/tensorflow/python/autograph/converters/ |
| D | control_flow_test.py | 265 graph_fn = def_function.function(tr, autograph=False)
267 graph_fn(constant_op.constant(0), constant_op.constant(5)))
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | meta_graph_test.py | 1007 def _testExportImportAcrossScopes(self, graph_fn, use_resource): argument
1016 graph_fn(use_resource=use_resource)
1028 graph_fn(use_resource=use_resource)
|
| /external/tensorflow/tensorflow/python/autograph/impl/ |
| D | api_test.py | 833 def graph_fn(): function
838 self.assertAllEqual(self.evaluate(graph_fn()), (3, 2, 1))
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 4139 graph_fn = eager_function.ConcreteFunction(exec_graph)
4141 graph_fn._num_positional_args = 0
4142 graph_fn._arg_keywords = []
4144 outputs = graph_fn()
|
| /external/tensorflow/tensorflow/python/training/ |
| D | saver_test.py | 2209 def _testGradientSerDes(self, graph_fn): argument
2227 output = graph_fn(var)
|