Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/predictor/
Dcore_estimator_predictor.py30 serving_input_receiver, estimator, output_key=None): argument
32 if output_key is None:
33 output_key = signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY
40 export_output = export_outputs.get(output_key)
43 export_outputs.keys(), output_key))
53 output_key=None, argument
72 serving_input_receiver, estimator, output_key)
Dcore_estimator_predictor_test.py59 for output_key, op in KEYS_AND_OPS:
67 output_key=output_key)
68 output_tensor_name = predictor.fetch_tensors[output_key].name
71 output_key,
73 output = predictor({'x': x, 'y': y})[output_key]
78 output_key, output, x, y))
Dpredictor_factories.py73 output_key=None, argument
103 output_key=output_key,
/external/tensorflow/tensorflow/python/saved_model/model_utils/
Dexport_utils.py101 for output_key, export_output in export_outputs.items():
102 signature_name = '{}'.format(output_key or 'None')
116 for output_key, export_output in export_outputs.items():
117 signature_name = '{}:{}'.format(receiver_name or 'None', output_key or
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/
Dmodel.py756 for output_key in outputs:
757 accumulator = output_accumulators_dict[output_key]
758 output_value = output_values[output_key]
819 outputs_dict = {output_key: _stack_and_transpose(output)
820 for output_key, output
Dsaved_model_utils.py50 output_key: graph.as_graph_element(output_value.name)
51 for output_key, output_value in signature.outputs.items()
Dhead_test.py435 fetches = {output_key: graph.as_graph_element(output_value.name)
436 for output_key, output_value
470 fetches = {output_key: graph.as_graph_element(output_value.name)
471 for output_key, output_value
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_runner.cc192 const string& output_key = in Run() local
196 TF_RETURN_IF_ERROR(Rendezvous::ParseKey(output_key, &parsed)); in Run()
Ddirect_session.cc1098 const string& output_key = it->second; in RecvPRunOutputs() local
1103 s = Rendezvous::ParseKey(output_key, &parsed); in RecvPRunOutputs()
/external/tensorflow/tensorflow/contrib/session_bundle/
Dbundle_shim.cc276 const string& output_key, SignatureDef* signature_def) { in AddOutputToSignatureDef() argument
299 (*signature_def->mutable_outputs())[output_key] = tensor_info; in AddOutputToSignatureDef()
/external/tensorflow/tensorflow/contrib/learn/python/learn/utils/
Dsaved_model_export_utils.py262 signature_def_map = {('%s:%s' % (input_key, output_key or 'None')):
266 for output_key, (problem_type,
/external/tensorflow/tensorflow/python/tools/
Dsaved_model_cli.py150 for output_key, output_tensor in sorted(outputs_tensor_info.items()):
151 in_print(' outputs[\'%s\'] tensor_info:' % output_key)