Searched refs:tensor_name_to_dtype (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/contrib/session_bundle/
H A Dbundle_shim.cc51 const std::unordered_map<string, DataType>& tensor_name_to_dtype) {
55 tensor_name_to_dtype, kRegressInputs,
58 tensor_name_to_dtype, kRegressOutputs,
65 const std::unordered_map<string, DataType>& tensor_name_to_dtype) {
69 classification_signature.input().tensor_name(), tensor_name_to_dtype,
72 classification_signature.classes().tensor_name(), tensor_name_to_dtype,
75 classification_signature.scores().tensor_name(), tensor_name_to_dtype,
81 const std::unordered_map<string, DataType>& tensor_name_to_dtype,
107 tensor_name_to_dtype, map_entry.first,
114 tensor_name_to_dtype, map_entr
49 BuildRegressionSignatureDef( const RegressionSignature& regression_signature, const std::unordered_map<string, DataType>& tensor_name_to_dtype) argument
63 BuildClassificationSignatureDef( const ClassificationSignature& classification_signature, const std::unordered_map<string, DataType>& tensor_name_to_dtype) argument
80 MaybeBuildPredictSignatureDef( const std::unordered_map<string, DataType>& tensor_name_to_dtype, MetaGraphDef* meta_graph_def) argument
160 ConvertDefaultSignatureToSignatureDef( const Signatures& signatures, const std::unordered_map<string, DataType>& tensor_name_to_dtype, MetaGraphDef* meta_graph_def) argument
196 ConvertNamedSignaturesToSignatureDef( const Signatures& signatures, const std::unordered_map<string, DataType>& tensor_name_to_dtype, MetaGraphDef* meta_graph_def) argument
243 AddInputToSignatureDef( const string& tensor_name, const std::unordered_map<string, DataType>& tensor_name_to_dtype, const string& input_key, SignatureDef* signature_def) argument
273 AddOutputToSignatureDef( const string& tensor_name, const std::unordered_map<string, DataType>& tensor_name_to_dtype, const string& output_key, SignatureDef* signature_def) argument
304 BuildTensorNameToDtypeMap( const MetaGraphDef& meta_graph_def, std::unordered_map<string, DataType>* tensor_name_to_dtype) argument
326 std::unordered_map<string, DataType> tensor_name_to_dtype; local
[all...]
H A Dbundle_shim.h40 const std::unordered_map<string, DataType>& tensor_name_to_dtype,
48 const std::unordered_map<string, DataType>& tensor_name_to_dtype,
H A Dbundle_shim_test.cc117 std::unordered_map<string, DataType> tensor_name_to_dtype; local
118 tensor_name_to_dtype[tensor_name] = tensorflow::DT_STRING;
120 AddInputToSignatureDef(tensor_name, tensor_name_to_dtype, map_key,
133 std::unordered_map<string, DataType> tensor_name_to_dtype; local
134 tensor_name_to_dtype[tensor_name] = tensorflow::DT_STRING;
136 AddOutputToSignatureDef(tensor_name, tensor_name_to_dtype, map_key,

Completed in 637 milliseconds