Searched refs:tensor_info_key (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/saved_model/cc/saved_model/ |
| D | signature_def_utils.cc | 121 const string& tensor_info_key, in FindInputTensorInfoByKey() argument
124 tensor_info_key, tensor_info); in FindInputTensorInfoByKey()
128 const string& tensor_info_key, in FindOutputTensorInfoByKey() argument
131 tensor_info_key, tensor_info); in FindOutputTensorInfoByKey()
135 const string& tensor_info_key, string* name) { in FindInputTensorNameByKey() argument
138 FindInputTensorInfoByKey(signature_def, tensor_info_key, &tensor_info)); in FindInputTensorNameByKey()
144 const string& tensor_info_key, string* name) { in FindOutputTensorNameByKey() argument
147 FindOutputTensorInfoByKey(signature_def, tensor_info_key, &tensor_info)); in FindOutputTensorNameByKey()
|
| D | signature_def_utils.h | 44 const string& tensor_info_key,
52 const string& tensor_info_key,
59 const string& tensor_info_key, string* name);
65 const string& tensor_info_key, string* name);
|