Searched refs:tensor_info (Results 1 - 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/python/saved_model/
H A Dutils_impl.py43 tensor_info = meta_graph_pb2.TensorInfo(
47 tensor_info.coo_sparse.values_tensor_name = tensor.values.name
48 tensor_info.coo_sparse.indices_tensor_name = tensor.indices.name
49 tensor_info.coo_sparse.dense_shape_tensor_name = tensor.dense_shape.name
51 tensor_info.name = tensor.name
52 return tensor_info
56 def get_tensor_from_tensor_info(tensor_info, graph=None, import_scope=None):
60 tensor_info: A TensorInfo proto describing a Tensor or SparseTensor.
63 import_scope: If not None, names in `tensor_info` are prefixed with this
67 The Tensor or SparseTensor in `graph` described by `tensor_info`
[all...]
H A Dutils_test.py56 tensor_info = utils.build_tensor_info(expected)
57 actual = utils.get_tensor_from_tensor_info(tensor_info)
63 tensor_info = utils.build_tensor_info(expected)
64 actual = utils.get_tensor_from_tensor_info(tensor_info)
73 tensor_info = utils.build_tensor_info(expected)
76 actual = utils.get_tensor_from_tensor_info(tensor_info,
87 tensor_info = utils.build_tensor_info(unscoped)
88 self.assertEqual("bar/x:0", tensor_info.name)
96 actual = utils.get_tensor_from_tensor_info(tensor_info,
102 tensor_info
[all...]
H A Dbuilder_impl.py193 def _validate_tensor_info(self, tensor_info):
199 tensor_info: `TensorInfo` protocol buffer to validate.
205 if tensor_info is None:
209 if not tensor_info.name:
212 "field set: %s" % tensor_info)
213 if tensor_info.dtype is types_pb2.DT_INVALID:
216 "field set: %s" % tensor_info)
517 asset_proto.tensor_info.name = asset_tensor.name
H A Dsignature_def_utils_impl.py274 key: tensor_shape.TensorShape(tensor_info.tensor_shape)
275 for key, tensor_info in tensor_info_dict.items()
289 key: dtypes.DType(tensor_info.dtype)
290 for key, tensor_info in tensor_info_dict.items()
H A Dloader_impl.py107 asset_tensor_dict[asset_proto.tensor_info.name] = os.path.join(
H A Dsaved_model_test.py95 self.assertEqual(expected_asset_tensor_name, asset.tensor_info.name)
97 def _validate_inputs_tensor_info(self, builder, tensor_info):
102 "foo_inputs": tensor_info
110 def _validate_outputs_tensor_info(self, builder, tensor_info):
115 dict(), {"foo_outputs": tensor_info}, "foo")
/external/tensorflow/tensorflow/contrib/saved_model/cc/saved_model/
H A Dsignature_def_utils.cc47 const TensorInfo** tensor_info) {
49 tensor_info_key, tensor_info);
54 const TensorInfo** tensor_info) {
56 tensor_info_key, tensor_info);
61 const TensorInfo* tensor_info; local
63 FindInputTensorInfoByKey(signature_def, tensor_info_key, &tensor_info));
64 *name = tensor_info->name();
70 const TensorInfo* tensor_info; local
72 FindOutputTensorInfoByKey(signature_def, tensor_info_key, &tensor_info));
73 *name = tensor_info
45 FindInputTensorInfoByKey(const SignatureDef& signature_def, const string& tensor_info_key, const TensorInfo** tensor_info) argument
52 FindOutputTensorInfoByKey(const SignatureDef& signature_def, const string& tensor_info_key, const TensorInfo** tensor_info) argument
[all...]
H A Dsignature_def_utils.h40 // NotFound and leaves *tensor_info unchanged. NOTE: The output TensorInfo*
45 const TensorInfo** tensor_info);
48 // NotFound and leaves *tensor_info unchanged. NOTE: The output TensorInfo*
53 const TensorInfo** tensor_info);
/external/tensorflow/tensorflow/contrib/tensorboard/plugins/trace/
H A Dtrace.py138 tensor_info = info.add()
143 tensor_info.shape.extend(shape)
144 tensor_info.dtype = t.dtype.name
145 tensor_info.num_bytes_per_elem = t.dtype.size
148 tensor_info.consumers.append(c.name)
/external/tensorflow/tensorflow/contrib/session_bundle/
H A 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)
H A Dbundle_shim.cc258 TensorInfo tensor_info; local
259 tensor_info.set_name(tensor_name);
261 tensor_info.set_dtype(it_tensor_info->second);
269 (*signature_def->mutable_inputs())[input_key] = tensor_info;
288 TensorInfo tensor_info; local
289 tensor_info.set_name(tensor_name);
291 tensor_info.set_dtype(it_tensor_info->second);
299 (*signature_def->mutable_outputs())[output_key] = tensor_info;
/external/tensorflow/tensorflow/cc/tools/
H A Dfreeze_saved_model.cc33 // Gets tensor names from tensor_info and inserts them into the set of tensor
35 void GetTensorNamesFromTensorInfo(const TensorInfo& tensor_info, argument
37 if (tensor_info.has_coo_sparse()) {
40 const TensorInfo_CooSparse& coo_sparse = tensor_info.coo_sparse();
45 tensor_names->insert(tensor_info.name());
/external/tensorflow/tensorflow/python/tools/
H A Dsaved_model_cli.py139 print('inputs[\'%s\'] tensor_info:' % input_key)
144 print('outputs[\'%s\'] tensor_info:' % output_key)
151 def _print_tensor_info(tensor_info):
152 """Prints details of the given tensor_info.
155 tensor_info: TensorInfo object to be printed.
159 for (key, value) in types_pb2.DataType.items()}[tensor_info.dtype])
161 if tensor_info.tensor_shape.unknown_rank:
164 dims = [str(dim.size) for dim in tensor_info.tensor_shape.dim]
168 print(' name: ' + tensor_info.name)
/external/tensorflow/tensorflow/cc/saved_model/
H A Dloader.cc121 {asset_file_def.tensor_info().name(), assets_file_path_tensor});
/external/tensorflow/tensorflow/contrib/meta_graph_transform/
H A Dmeta_graph_transform_test.py42 asset_file_def.tensor_info.name = node_name
/external/tensorflow/tensorflow/core/grappler/
H A Dgrappler_item_builder.cc265 asset_node_to_value[NodeName(asset_file_def.tensor_info().name())] =

Completed in 173 milliseconds