HomeSort by relevance Sort by last modified time
    Searched refs:tensor_name_to_dtype (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/contrib/session_bundle/
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_entry.first
326 std::unordered_map<string, DataType> tensor_name_to_dtype; local
    [all...]
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,
bundle_shim_test.cc 121 std::unordered_map<string, DataType> tensor_name_to_dtype; local
122 tensor_name_to_dtype[tensor_name] = tensorflow::DT_STRING;
124 AddInputToSignatureDef(tensor_name, tensor_name_to_dtype, map_key,
137 std::unordered_map<string, DataType> tensor_name_to_dtype; local
138 tensor_name_to_dtype[tensor_name] = tensorflow::DT_STRING;
140 AddOutputToSignatureDef(tensor_name, tensor_name_to_dtype, map_key,

Completed in 114 milliseconds