| /external/tensorflow/tensorflow/contrib/saved_model/cc/saved_model/ |
| signature_def_utils.cc | 47 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->name()
[all...] |
| signature_def_utils.h | 40 // 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/python/saved_model/ |
| utils_impl.py | 43 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...] |
| utils_test.py | 56 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 = utils.build_tensor_info(expected
[all...] |
| builder_impl.py | 193 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
|
| signature_def_utils_impl.py | 274 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()
|
| loader_impl.py | 107 asset_tensor_dict[asset_proto.tensor_info.name] = os.path.join(
|
| saved_model_test.py | 95 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")
[all...] |
| /external/tensorflow/tensorflow/contrib/tensorboard/plugins/trace/ |
| trace.py | 138 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/ |
| bundle_shim.cc | 258 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;
|
| bundle_shim.py | 48 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/ |
| freeze_saved_model.cc | 33 // Gets tensor names from tensor_info and inserts them into the set of tensor
35 void GetTensorNamesFromTensorInfo(const TensorInfo& tensor_info,
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/ |
| saved_model_cli.py | 139 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/ |
| loader.cc | 121 {asset_file_def.tensor_info().name(), assets_file_path_tensor});
|
| /external/tensorflow/tensorflow/core/grappler/ |
| grappler_item_builder.cc | 265 asset_node_to_value[NodeName(asset_file_def.tensor_info().name())] =
|
| /external/tensorflow/tensorflow/contrib/meta_graph_transform/ |
| meta_graph_transform_test.py | 42 asset_file_def.tensor_info.name = node_name
|
