| /external/tensorflow/tensorflow/contrib/framework/python/framework/ |
| D | checkpoint_utils.py | 222 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/ |
| D | checkpoint_utils.py | 200 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/ |
| D | feature_column.py | 108 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/ |
| D | feature_column.py | 1088 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 …]
|
| D | feature_column_test.py | 1122 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/ |
| D | tensorflow.feature_column.pbtxt | 29 …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/ |
| D | tensorflow.feature_column.pbtxt | 29 …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/ |
| D | feature_column_v2.py | 823 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 …]
|
| D | feature_column.py | 815 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
|
| D | feature_column_test.py | 4794 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 …]
|
| D | feature_column_v2_test.py | 6248 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',
|