Home
last modified time | relevance | path

Searched refs:tensor_info (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/contrib/saved_model/cc/saved_model/
Dsignature_def_utils.cc43 const TensorInfo* tensor_info; in VerifyTensorInfoForKeyInMap() local
44 const Status& status = FindInProtobufMap("", map, key, &tensor_info); in VerifyTensorInfoForKeyInMap()
48 if (tensor_info->dtype() != correct_dtype) { in VerifyTensorInfoForKeyInMap()
95 const TensorInfo& tensor_info = output.second; in IsValidClassificationSignature() local
97 if (tensor_info.dtype() != DT_STRING) { in IsValidClassificationSignature()
101 if (tensor_info.dtype() != DT_FLOAT) { in IsValidClassificationSignature()
122 const TensorInfo** tensor_info) { in FindInputTensorInfoByKey() argument
124 tensor_info_key, tensor_info); in FindInputTensorInfoByKey()
129 const TensorInfo** tensor_info) { in FindOutputTensorInfoByKey() argument
131 tensor_info_key, tensor_info); in FindOutputTensorInfoByKey()
[all …]
Dsignature_def_utils.h45 const TensorInfo** tensor_info);
53 const TensorInfo** tensor_info);
/external/tensorflow/tensorflow/python/saved_model/
Dutils_impl.py68 tensor_info = meta_graph_pb2.TensorInfo(
72 tensor_info.coo_sparse.values_tensor_name = tensor.values.name
73 tensor_info.coo_sparse.indices_tensor_name = tensor.indices.name
74 tensor_info.coo_sparse.dense_shape_tensor_name = tensor.dense_shape.name
76 tensor_info.name = tensor.name
77 return tensor_info
122 def get_tensor_from_tensor_info(tensor_info, graph=None, import_scope=None): argument
143 encoding = tensor_info.WhichOneof("encoding")
145 return _get_tensor(tensor_info.name)
148 _get_tensor(tensor_info.coo_sparse.indices_tensor_name),
[all …]
Dutils_test.py94 tensor_info = utils.build_tensor_info(expected)
95 actual = utils.get_tensor_from_tensor_info(tensor_info)
102 tensor_info = utils.build_tensor_info(expected)
103 actual = utils.get_tensor_from_tensor_info(tensor_info)
112 tensor_info = utils.build_tensor_info(expected)
115 actual = utils.get_tensor_from_tensor_info(tensor_info,
126 tensor_info = utils.build_tensor_info(unscoped)
127 self.assertEqual("bar/x:0", tensor_info.name)
135 actual = utils.get_tensor_from_tensor_info(tensor_info,
142 tensor_info = utils.build_tensor_info(expected)
[all …]
Dbuilder_impl.py154 def _validate_tensor_info(self, tensor_info): argument
167 if tensor_info is None:
171 if tensor_info.WhichOneof("encoding") is None:
176 % tensor_info)
177 if tensor_info.dtype is types_pb2.DT_INVALID:
180 "field set: %s" % tensor_info)
744 asset_proto.tensor_info.name = asset_tensor.name
778 asset_proto.tensor_info.name = asset_tensor.name
Dsignature_def_utils_impl.py383 tensor_info = signature_def.outputs[key]
388 tensor_info, import_scope=import_scope)
Dsaved_model_test.py101 asset_file_def[asset_id].tensor_info.name)
103 def _validate_inputs_tensor_info_fail(self, builder, tensor_info): argument
108 "foo_inputs": tensor_info
116 def _validate_inputs_tensor_info_accept(self, builder, tensor_info): argument
121 "foo_inputs": tensor_info
127 def _validate_outputs_tensor_info_fail(self, builder, tensor_info): argument
132 dict(), {"foo_outputs": tensor_info}, "foo")
139 def _validate_outputs_tensor_info_accept(self, builder, tensor_info): argument
144 dict(), {"foo_outputs": tensor_info}, "foo")
1352 self.assertEqual(expected_asset_tensor_name, asset.tensor_info.name)
Dloader_impl.py124 tensor_name = asset_proto.tensor_info.name
Dsave.py467 asset_def.tensor_info.name = asset_path_initializer.name
/external/tensorflow/tensorflow/lite/python/
Dconvert_saved_model.py31 def _log_tensor_details(tensor_info): argument
33 for key in tensor_info:
34 val = tensor_info[key]
105 def gather_names(tensor_info): argument
106 return [tensor_info[key].name for key in tensor_info]
/external/tensorflow/tensorflow/contrib/session_bundle/
Dbundle_shim.cc258 TensorInfo tensor_info; in AddInputToSignatureDef() local
259 tensor_info.set_name(tensor_name); in AddInputToSignatureDef()
261 tensor_info.set_dtype(it_tensor_info->second); in AddInputToSignatureDef()
269 (*signature_def->mutable_inputs())[input_key] = tensor_info; in AddInputToSignatureDef()
288 TensorInfo tensor_info; in AddOutputToSignatureDef() local
289 tensor_info.set_name(tensor_name); in AddOutputToSignatureDef()
291 tensor_info.set_dtype(it_tensor_info->second); in AddOutputToSignatureDef()
299 (*signature_def->mutable_outputs())[output_key] = tensor_info; in AddOutputToSignatureDef()
Dbundle_shim.py48 tensor_info = meta_graph_pb2.TensorInfo(name=tensor_name)
49 signature_def.inputs[map_key].CopyFrom(tensor_info)
65 tensor_info = meta_graph_pb2.TensorInfo(name=tensor_name)
66 signature_def.outputs[map_key].CopyFrom(tensor_info)
/external/tensorflow/tensorflow/cc/tools/
Dfreeze_saved_model.cc36 void GetTensorNamesFromTensorInfo(const TensorInfo& tensor_info, in GetTensorNamesFromTensorInfo() argument
38 if (tensor_info.has_coo_sparse()) { in GetTensorNamesFromTensorInfo()
41 const TensorInfo_CooSparse& coo_sparse = tensor_info.coo_sparse(); in GetTensorNamesFromTensorInfo()
46 tensor_names->insert(tensor_info.name()); in GetTensorNamesFromTensorInfo()
/external/tensorflow/tensorflow/python/tools/
Dsaved_model_cli.py158 def _print_tensor_info(tensor_info, indent=0): argument
171 for (key, value) in types_pb2.DataType.items()}[tensor_info.dtype])
173 if tensor_info.tensor_shape.unknown_rank:
176 dims = [str(dim.size) for dim in tensor_info.tensor_shape.dim]
180 in_print(' name: ' + tensor_info.name)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.saved_model.utils.pbtxt9 …argspec: "args=[\'tensor_info\', \'graph\', \'import_scope\'], varargs=None, keywords=None, defaul…
Dtensorflow.saved_model.pbtxt157 …argspec: "args=[\'tensor_info\', \'graph\', \'import_scope\'], varargs=None, keywords=None, defaul…
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py226 def _gather_names(tensor_info): argument
228 return set([tensor_info[key].name.split(":")[0] for key in tensor_info])
/external/tensorflow/tensorflow/cc/saved_model/
Dloader.cc92 {asset_file_def.tensor_info().name(), assets_file_path_tensor}); in AddAssetsTensorsToInputs()
/external/tensorflow/tensorflow/core/protobuf/
Dsaved_object_graph.proto87 // `AssetFileDef.tensor_info`, MUST be ignored.
Dmeta_graph.proto321 TensorInfo tensor_info = 1; field
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder.cc397 asset_node_to_value[NodeName(asset_file_def.tensor_info().name())] = in GrapplerItemFromMetaGraphDef()
/external/tensorflow/tensorflow/contrib/meta_graph_transform/
Dmeta_graph_transform_test.py42 asset_file_def.tensor_info.name = node_name