| /external/tensorflow/tensorflow/core/kernels/ |
| save_restore_v2_ops.cc | 42 const Tensor& prefix, const Tensor& tensor_names,
45 const int num_tensors = static_cast<int>(tensor_names.NumElements());
51 TensorShapeUtils::IsVector(tensor_names.shape()) &&
54 "Input tensor_names and shape_and_slices "
56 tensor_names.shape().DebugString(), " and ",
59 tensor_names.NumElements() == shape_and_slices.NumElements(),
60 errors::InvalidArgument("tensor_names and shape_and_slices "
62 tensor_names.NumElements(), " vs. ",
65 FastBoundsCheck(tensor_names.NumElements() + kFixedInputs,
97 const Tensor& tensor_names = context->input(1) variable
151 const Tensor& tensor_names = context->input(1); variable
[all...] |
| save_restore_tensor.h | 64 // * "tensor_names" and "shape_and_slices" shaped {N}, both DT_STRING.
67 const Tensor& tensor_names,
|
| restore_v2_op_test.cc | 53 .Input(FakeInput()) // tensor_names
63 const std::vector<string> tensor_names = { local
76 .Input(FakeInput()) // tensor_names
86 .Input(FakeInput()) // tensor_names
113 TensorShape({static_cast<int>(tensor_names.size())}),
114 [&tensor_names](int x) -> string { return tensor_names[x]; });
118 TensorShape({static_cast<int>(tensor_names.size())}),
201 [&](int x) -> string { return tensor_names[0]; });
216 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[1]
[all...] |
| restore_op_test.cc | 63 const std::vector<string> tensor_names = { local
102 TensorShape({static_cast<int>(tensor_names.size())}),
103 [&tensor_names](int x) -> string { return tensor_names[x]; });
194 [&](int x) -> string { return tensor_names[0]; });
206 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[1];
218 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[2];
230 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[3];
242 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[4];
254 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[5]
[all...] |
| save_restore_tensor.cc | 318 const Tensor& tensor_names,
323 const auto& tensor_names_flat = tensor_names.flat<string>();
|
| remote_fused_graph_execute_utils.cc | 87 const std::vector<string>& tensor_names) {
89 for (const string& name : tensor_names) {
[all...] |
| /external/tensorflow/tensorflow/cc/tools/ |
| freeze_saved_model.cc | 37 std::unordered_set<string>* tensor_names) {
42 tensor_names->insert(coo_sparse.values_tensor_name());
43 tensor_names->insert(coo_sparse.indices_tensor_name());
44 tensor_names->insert(coo_sparse.dense_shape_tensor_name());
46 tensor_names->insert(tensor_info.name());
128 std::vector<string> tensor_names; local
129 tensor_names.reserve(variable_names_set.size());
136 tensor_names.push_back(node_name + "/Read/ReadVariableOp:0");
138 tensor_names.push_back(node_name + ":0");
143 session->Run(/* inputs */ {}, tensor_names, /* targets */ {}, &outputs))
[all...] |
| /external/tensorflow/tensorflow/python/training/saving/ |
| 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/python/debug/cli/ |
| 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]
|
| 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)
[all...] |
| /external/tensorflow/tensorflow/core/graph/ |
| quantize_training.cc | 191 // Get the tensor_names and shape_and_slices tensors from the const op.
192 Tensor tensor_names; local
195 GetNodeAttr(tensor_names_op->attrs(), "value", &tensor_names));
199 int tn_size = tensor_names.NumElements();
205 // The first three inputs are prefix, tensor_names, and shapes_and_slices.
219 FillStringTensor(&new_tensor_names, tensor_names);
271 strings::StrCat(new_restore_op_name, "/tensor_names");
275 // Construct the tensor_names input with the variable name.
276 Node* tensor_names; local
282 .Finalize(graph, &tensor_names));
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| io_ops.py | 42 def _save(filename, tensor_names, tensors, tensor_slices=None, name="save"):
54 tensor_names: a list of strings.
62 The length of tensors should match the size of tensor_names and of
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/ |
| convert_saved_model.py | 156 def get_tensors_from_tensor_names(graph, tensor_names):
157 """Gets the Tensors associated with the `tensor_names` in the provided graph.
161 tensor_names: List of strings that represent names of tensors in the graph.
168 tensor_names contains an invalid tensor name.
176 # Get the tensors associated with tensor_names.
179 for name in tensor_names:
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/ |
| strip_pruning_vars_test.py | 49 def _get_node_names(tensor_names):
52 for tensor_name in tensor_names
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| monitors.py | 443 def __init__(self, tensor_names, every_n=100, first_n=1):
447 tensor_names: `dict` of tag to tensor names or
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}))
[all...] |
| monitors_test.py | 245 self._run_monitor(learn.monitors.PrintTensor(tensor_names=[t.name]))
[all...] |
| /external/tensorflow/tensorflow/python/training/ |
| 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,
526 # Store the tensor values to the tensor_names.
[all...] |
| /external/tensorflow/tensorflow/python/training/tracking/ |
| base.py | 299 tensor_names=[checkpoint_key],
[all...] |
| util.py | 236 tensor_names=[spec.name],
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| control_flow_ops_py_test.py | 100 tensor_names = []
105 tensor_names.append(t.name)
106 return tensor_names
[all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
