Searched refs:tensor_info (Results 1 - 16 of 16) sorted by relevance
/external/tensorflow/tensorflow/python/saved_model/ |
H A D | 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...] |
H A D | 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 [all...] |
H A D | 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
|
H A D | 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()
|
H A D | loader_impl.py | 107 asset_tensor_dict[asset_proto.tensor_info.name] = os.path.join(
|
H A D | 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")
|
/external/tensorflow/tensorflow/contrib/saved_model/cc/saved_model/ |
H A D | 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 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 D | 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/contrib/tensorboard/plugins/trace/ |
H A D | 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/ |
H A D | 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)
|
H A D | 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;
|
/external/tensorflow/tensorflow/cc/tools/ |
H A D | 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, 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 D | 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/ |
H A D | loader.cc | 121 {asset_file_def.tensor_info().name(), assets_file_path_tensor});
|
/external/tensorflow/tensorflow/contrib/meta_graph_transform/ |
H A D | meta_graph_transform_test.py | 42 asset_file_def.tensor_info.name = node_name
|
/external/tensorflow/tensorflow/core/grappler/ |
H A D | grappler_item_builder.cc | 265 asset_node_to_value[NodeName(asset_file_def.tensor_info().name())] =
|
Completed in 173 milliseconds