Home
last modified time | relevance | path

Searched refs:tensor_name_in_ckpt (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/contrib/framework/python/framework/
Dcheckpoint_utils.py222 for tensor_name_in_ckpt, current_var_or_name in six.iteritems(assignment_map):
240 if tensor_name_in_ckpt not in variable_map:
242 tensor_name_in_ckpt, checkpoint_dir, variable_map
247 variable_map[tensor_name_in_ckpt]):
252 tensor_name_in_ckpt, str(variable_map[tensor_name_in_ckpt])
257 _set_variable_or_list_initializer(var, filepattern, tensor_name_in_ckpt)
259 var_name, checkpoint_dir, tensor_name_in_ckpt
266 if not tensor_name_in_ckpt.endswith("/"):
270 scopes, tensor_name_in_ckpt))
286 if tensor_name_in_ckpt != "/":
[all …]
/external/tensorflow/tensorflow/python/training/
Dcheckpoint_utils.py200 for tensor_name_in_ckpt, current_var_or_name in sorted(
218 if tensor_name_in_ckpt not in variable_map:
220 tensor_name_in_ckpt, ckpt_dir_or_file, variable_map
225 variable_map[tensor_name_in_ckpt]):
230 tensor_name_in_ckpt, str(variable_map[tensor_name_in_ckpt])
235 _set_variable_or_list_initializer(var, ckpt_file, tensor_name_in_ckpt)
237 var_name, ckpt_dir_or_file, tensor_name_in_ckpt)
243 if not tensor_name_in_ckpt.endswith("/"):
247 scopes, tensor_name_in_ckpt))
263 if tensor_name_in_ckpt != "/":
[all …]
/external/tensorflow/tensorflow/python/tpu/
Dfeature_column.py108 tensor_name_in_ckpt=None,
139 tensor_name_in_ckpt=None,
158 tensor_name_in_ckpt=None,
214 tensor_name_in_ckpt=None, argument
227 tensor_name_in_ckpt=tensor_name_in_ckpt,
237 tensor_name_in_ckpt=None, argument
312 tensor_name_in_ckpt=None, argument
323 tensor_name_in_ckpt=tensor_name_in_ckpt,
334 tensor_name_in_ckpt=None, argument
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dfeature_column.py1088 tensor_name_in_ckpt=None, argument
1098 if (ckpt_to_load_from is None) != (tensor_name_in_ckpt is None):
1111 tensor_name_in_ckpt,
1161 return self.ckpt_to_load_from, self.tensor_name_in_ckpt
1323 tensor_name_in_ckpt=None, argument
1360 ckpt_to_load_from, tensor_name_in_ckpt,
1370 tensor_name_in_ckpt=None, argument
1434 ckpt_to_load_from, tensor_name_in_ckpt,
1477 ckpt_to_load_from, tensor_name_in_ckpt,
2285 tensor_name_in_ckpt=None): argument
[all …]
Dfeature_column_test.py1122 tensor_name_in_ckpt=("run_1/object_in_image_embedding/"
1189 tensor_name_in_ckpt=("run_1/col_1_X_col_2_X_col_3/"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.feature_column.pbtxt29 …mension\', \'combiner\', \'initializer\', \'ckpt_to_load_from\', \'tensor_name_in_ckpt\', \'max_no…
65 …\', \'shared_embedding_collection_name\', \'ckpt_to_load_from\', \'tensor_name_in_ckpt\', \'max_no…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.feature_column.pbtxt29 …mension\', \'combiner\', \'initializer\', \'ckpt_to_load_from\', \'tensor_name_in_ckpt\', \'max_no…
73 …\', \'shared_embedding_collection_name\', \'ckpt_to_load_from\', \'tensor_name_in_ckpt\', \'max_no…
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_v2.py823 tensor_name_in_ckpt=None, argument
899 if (ckpt_to_load_from is None) != (tensor_name_in_ckpt is None):
917 tensor_name_in_ckpt=tensor_name_in_ckpt,
929 tensor_name_in_ckpt=None, argument
1034 if (ckpt_to_load_from is None) != (tensor_name_in_ckpt is None):
1088 tensor_name_in_ckpt=tensor_name_in_ckpt,
1102 tensor_name_in_ckpt=None, argument
1206 if (ckpt_to_load_from is None) != (tensor_name_in_ckpt is None):
1249 dimension, initializer, ckpt_to_load_from, tensor_name_in_ckpt,
3059 self.tensor_name_in_ckpt: to_restore
[all …]
Dfeature_column.py815 tensor_name_in_ckpt=None, argument
891 if (ckpt_to_load_from is None) != (tensor_name_in_ckpt is None):
920 tensor_name_in_ckpt=tensor_name_in_ckpt,
2440 self.tensor_name_in_ckpt: to_restore
2583 self.tensor_name_in_ckpt: to_restore
Dfeature_column_test.py4794 self.assertIsNone(embedding_column.tensor_name_in_ckpt)
4816 tensor_name_in_ckpt='my_ckpt_tensor',
4823 self.assertEqual('my_ckpt_tensor', embedding_column.tensor_name_in_ckpt)
4845 tensor_name_in_ckpt='my_ckpt_tensor',
4858 self.assertEqual('my_ckpt_tensor', embedding_column.tensor_name_in_ckpt)
5184 tensor_name_in_ckpt=ckpt_tensor)
5508 self.assertIsNone(embedding_column_a.tensor_name_in_ckpt)
5509 self.assertIsNone(embedding_column_b.tensor_name_in_ckpt)
5545 tensor_name_in_ckpt='my_ckpt_tensor',
5560 self.assertEqual('my_ckpt_tensor', embedding_column_a.tensor_name_in_ckpt)
[all …]
Dfeature_column_v2_test.py6248 self.assertIsNone(embedding_column.tensor_name_in_ckpt)
6277 tensor_name_in_ckpt='my_ckpt_tensor',
6284 self.assertEqual('my_ckpt_tensor', embedding_column.tensor_name_in_ckpt)
6304 tensor_name_in_ckpt='my_ckpt_tensor',
6317 self.assertEqual('my_ckpt_tensor', embedding_column.tensor_name_in_ckpt)
6700 tensor_name_in_ckpt=ckpt_tensor)
7312 tensor_name_in_ckpt='my_ckpt_tensor',
7344 tensor_name_in_ckpt='my_ckpt_tensor',