| /external/tensorflow/tensorflow/core/kernels/ |
| D | save_restore_v2_ops.cc | 42 const Tensor& prefix, const Tensor& tensor_names, in ValidateInputs() argument
45 const int num_tensors = static_cast<int>(tensor_names.NumElements()); in ValidateInputs()
51 TensorShapeUtils::IsVector(tensor_names.shape()) && in ValidateInputs()
56 tensor_names.shape().DebugString(), " and ", in ValidateInputs()
59 tensor_names.NumElements() == shape_and_slices.NumElements(), in ValidateInputs()
62 tensor_names.NumElements(), " vs. ", in ValidateInputs()
65 FastBoundsCheck(tensor_names.NumElements() + kFixedInputs, in ValidateInputs()
97 const Tensor& tensor_names = context->input(1); in Compute() local
99 ValidateInputs(true /* is save op */, context, prefix, tensor_names, in Compute()
103 const int num_tensors = static_cast<int>(tensor_names.NumElements()); in Compute()
[all …]
|
| D | restore_v2_op_test.cc | 63 const std::vector<string> tensor_names = { in RunTest() local
113 TensorShape({static_cast<int>(tensor_names.size())}), in RunTest()
114 [&tensor_names](int x) -> string { return tensor_names[x]; }); in RunTest()
118 TensorShape({static_cast<int>(tensor_names.size())}), in RunTest()
201 [&](int x) -> string { return tensor_names[0]; }); in RunTest()
216 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[1]; in RunTest()
228 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[2]; in RunTest()
240 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[3]; in RunTest()
252 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[4]; in RunTest()
264 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[5]; in RunTest()
[all …]
|
| D | restore_op_test.cc | 63 const std::vector<string> tensor_names = { in TEST_F() local
102 TensorShape({static_cast<int>(tensor_names.size())}), in TEST_F()
103 [&tensor_names](int x) -> string { return tensor_names[x]; }); in TEST_F()
194 [&](int x) -> string { return tensor_names[0]; }); in TEST_F()
206 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[1]; in TEST_F()
218 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[2]; in TEST_F()
230 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[3]; in TEST_F()
242 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[4]; in TEST_F()
254 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[5]; in TEST_F()
267 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[6]; in TEST_F()
[all …]
|
| D | save_restore_tensor.h | 67 const Tensor& tensor_names,
|
| D | save_restore_tensor.cc | 318 const Tensor& tensor_names, in RestoreTensorsV2() argument
323 const auto& tensor_names_flat = tensor_names.flat<string>(); in RestoreTensorsV2()
|
| D | remote_fused_graph_execute_utils.cc | 87 const std::vector<string>& tensor_names) { in ContainsSameTensorId() argument
89 for (const string& name : tensor_names) { in ContainsSameTensorId()
|
| /external/tensorflow/tensorflow/cc/tools/ |
| D | freeze_saved_model.cc | 37 std::unordered_set<string>* tensor_names) { in GetTensorNamesFromTensorInfo() argument
42 tensor_names->insert(coo_sparse.values_tensor_name()); in GetTensorNamesFromTensorInfo()
43 tensor_names->insert(coo_sparse.indices_tensor_name()); in GetTensorNamesFromTensorInfo()
44 tensor_names->insert(coo_sparse.dense_shape_tensor_name()); in GetTensorNamesFromTensorInfo()
46 tensor_names->insert(tensor_info.name()); in GetTensorNamesFromTensorInfo()
128 std::vector<string> tensor_names; in GetVariableNameToTensorMap() local
129 tensor_names.reserve(variable_names_set.size()); in GetVariableNameToTensorMap()
136 tensor_names.push_back(node_name + "/Read/ReadVariableOp:0"); in GetVariableNameToTensorMap()
138 tensor_names.push_back(node_name + ":0"); in GetVariableNameToTensorMap()
143 session->Run(/* inputs */ {}, tensor_names, /* targets */ {}, &outputs)); in GetVariableNameToTensorMap()
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | functional_saver.py | 86 tensor_names = []
91 tensor_names.append(spec.name)
96 file_prefix, tensor_names, tensor_slices, tensors)]):
127 tensor_names, tensor_slices, tensor_dtypes = zip(*restore_specs)
130 file_prefix, tensor_names, tensor_slices, tensor_dtypes)
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_Save.pbtxt | 11 name: "tensor_names"
24 The size of `tensor_names` must match the number of tensors in `data`. `data[i]`
25 is written to `filename` with name `tensor_names[i]`.
|
| D | api_def_SaveSlices.pbtxt | 11 name: "tensor_names"
34 have as many elements as `tensor_names`.
|
| D | api_def_SaveV2.pbtxt | 11 name: "tensor_names"
|
| D | api_def_RestoreV2.pbtxt | 10 name: "tensor_names"
|
| /external/tensorflow/tensorflow/python/debug/cli/ |
| D | stepper_cli.py | 531 tensor_names = self._resolve_tensor_names(tensor_name)
532 if not tensor_names:
535 elif len(tensor_names) > 1:
539 tensor_name = tensor_names[0]
574 tensor_names = self._resolve_tensor_names(parsed.tensor_name)
575 if not tensor_names:
578 elif len(tensor_names) > 1:
582 tensor_name = tensor_names[0]
|
| D | analyzer_cli_test.py | 214 tensor_names = []
225 tensor_names.append(items[3])
245 sorted_tensor_names = sorted(tensor_names)
248 tst.assertEqual(sorted_tensor_names, tensor_names)
254 tst.assertIn(tensor_name, tensor_names)
255 index = tensor_names.index(tensor_name)
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | quantize_training.cc | 192 Tensor tensor_names; in ConnectVariablesToSaveOp() local
195 GetNodeAttr(tensor_names_op->attrs(), "value", &tensor_names)); in ConnectVariablesToSaveOp()
199 int tn_size = tensor_names.NumElements(); in ConnectVariablesToSaveOp()
219 FillStringTensor(&new_tensor_names, tensor_names); in ConnectVariablesToSaveOp()
276 Node* tensor_names; in AddRestoreVariableSubgraphs() local
282 .Finalize(graph, &tensor_names)); in AddRestoreVariableSubgraphs()
297 .Input(tensor_names) in AddRestoreVariableSubgraphs()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | io_ops.py | 42 def _save(filename, tensor_names, tensors, tensor_slices=None, name="save"): argument
69 return gen_io_ops.save(filename, tensor_names, tensors, name=name)
71 return gen_io_ops.save_slices(filename, tensor_names, tensor_slices,
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert_saved_model.py | 156 def get_tensors_from_tensor_names(graph, tensor_names): argument
179 for name in tensor_names:
|
| /external/tensorflow/tensorflow/cc/saved_model/testdata/half_plus_two_pbtxt/00000123/ |
| D | saved_model.pbtxt | 331 name: "tensor_names"
356 name: "tensor_names"
1828 name: "save/SaveV2/tensor_names"
1907 input: "save/SaveV2/tensor_names"
2023 name: "save/RestoreV2/tensor_names"
2098 input: "save/RestoreV2/tensor_names"
2161 name: "save/RestoreV2_1/tensor_names"
2236 input: "save/RestoreV2_1/tensor_names"
2299 name: "save/RestoreV2_2/tensor_names"
2374 input: "save/RestoreV2_2/tensor_names"
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| D | monitors.py | 443 def __init__(self, tensor_names, every_n=100, first_n=1): argument
453 if not isinstance(tensor_names, dict):
454 tensor_names = {item: item for item in tensor_names}
455 self._tensor_names = tensor_names
851 monitors.append(PrintTensor(tensor_names={"loss": loss_op.name}))
|
| D | monitors_test.py | 245 self._run_monitor(learn.monitors.PrintTensor(tensor_names=[t.name]))
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/ |
| D | strip_pruning_vars_test.py | 49 def _get_node_names(tensor_names): argument
52 for tensor_name in tensor_names
|
| /external/tensorflow/tensorflow/python/training/ |
| D | saver.py | 105 tensor_names = []
110 tensor_names.append(spec.name)
116 tensor_names=tensor_names,
122 return io_ops.save_v2(filename_tensor, tensor_names, tensor_slices,
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | base.py | 299 tensor_names=[checkpoint_key],
|
| D | util.py | 236 tensor_names=[spec.name],
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 3037 …argspec: "args=[\'prefix\', \'tensor_names\', \'shape_and_slices\', \'dtypes\', \'name\'], varargs…
3153 …argspec: "args=[\'filename\', \'tensor_names\', \'data\', \'name\'], varargs=None, keywords=None, …
3157 …argspec: "args=[\'filename\', \'tensor_names\', \'shapes_and_slices\', \'data\', \'name\'], vararg…
3161 …argspec: "args=[\'prefix\', \'tensor_names\', \'shape_and_slices\', \'tensors\', \'name\'], vararg…
|