Searched refs:tensor_name_to_dtype (Results 1 - 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/contrib/session_bundle/ |
H A D | bundle_shim.cc | 51 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 D | bundle_shim.h | 40 const std::unordered_map<string, DataType>& tensor_name_to_dtype, 48 const std::unordered_map<string, DataType>& tensor_name_to_dtype,
|
H A D | bundle_shim_test.cc | 117 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