Home
last modified time | relevance | path

Searched refs:graph_func (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/
Dmodel_handler.py142 graph_func = imported.signatures[saved_model_signature_key]
143 return convert_to_constants.convert_variables_to_constants_v2(graph_func)
335 def graph_func(self): member in ModelHandlerV2
336 graph_func = load_graph_func(
340 return convert_to_constants.convert_variables_to_constants_v2(graph_func)
344 return [tensor.name for tensor in self.graph_func.inputs]
348 return [tensor.name for tensor in self.graph_func.outputs]
356 for tensor in self.graph_func.inputs
369 self.graph_func(*inputs)
373 outputs = self.graph_func(*inputs)
[all …]
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert_test.py886 graph_func = saved_model_loaded.signatures[
891 output = graph_func(input1=np_input1, input2=np_input2)["output_0"]