| /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 | 230 const Tensor& tensor_names,
235 const auto& tensor_names_flat = tensor_names.flat<string>();
|
| remote_fused_graph_execute_utils.cc | 85 const std::vector<string>& tensor_names) {
87 for (const string& name : tensor_names) {
[all...] |
| /external/tensorflow/tensorflow/cc/tools/ |
| freeze_saved_model.cc | 36 std::unordered_set<string>* tensor_names) {
41 tensor_names->insert(coo_sparse.values_tensor_name());
42 tensor_names->insert(coo_sparse.indices_tensor_name());
43 tensor_names->insert(coo_sparse.dense_shape_tensor_name());
45 tensor_names->insert(tensor_info.name());
120 std::vector<string> tensor_names; local
124 tensor_names.push_back(node_name + ":0");
128 session->Run(/* inputs */ {}, tensor_names, /* targets */ {}, &outputs));
|
| /external/tensorflow/tensorflow/python/debug/cli/ |
| stepper_cli.py | 507 tensor_names = self._resolve_tensor_names(tensor_name)
508 if not tensor_names:
511 elif len(tensor_names) > 1:
515 tensor_name = tensor_names[0]
550 tensor_names = self._resolve_tensor_names(parsed.tensor_name)
551 if not tensor_names:
554 elif len(tensor_names) > 1:
558 tensor_name = tensor_names[0]
|
| analyzer_cli_test.py | 213 tensor_names = []
224 tensor_names.append(items[3])
244 sorted_tensor_names = sorted(tensor_names)
247 tst.assertEqual(sorted_tensor_names, tensor_names)
253 tst.assertIn(tensor_name, tensor_names)
254 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);
265 strings::StrCat(new_restore_op_name, "/tensor_names");
269 // Construct the tensor_names input with the variable name.
270 Node* tensor_names; local
276 .Finalize(graph, &tensor_names));
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| io_ops.py | 87 def _save(filename, tensor_names, tensors, tensor_slices=None, name="save"):
99 tensor_names: a list of strings.
107 The length of tensors should match the size of tensor_names and of
114 return gen_io_ops._save(filename, tensor_names, tensors, name=name)
116 return gen_io_ops._save_slices(filename, tensor_names, tensor_slices,
|
| /external/tensorflow/tensorflow/contrib/tensorrt/convert/ |
| convert_graph.cc | 102 const std::vector<string>& tensor_names) {
104 for (string const& tensor_name : tensor_names) {
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| monitors.py | 427 def __init__(self, tensor_names, every_n=100, first_n=1):
431 tensor_names: `dict` of tag to tensor names or
437 if not isinstance(tensor_names, dict):
438 tensor_names = {item: item for item in tensor_names}
439 self._tensor_names = tensor_names
801 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 | 223 tensor_names = []
228 tensor_names.append(spec.name)
234 tensor_names=tensor_names,
240 return io_ops.save_v2(filename_tensor, tensor_names, tensor_slices,
819 # Store the tensor values to the tensor_names.
[all...] |
| checkpointable.py | 181 tensor_names=[checkpoint_key],
|
| /external/tensorflow/tensorflow/contrib/eager/python/ |
| checkpointable_utils.py | 418 tensor_names=[_OBJECT_GRAPH_PROTO_KEY],
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| control_flow_ops_py_test.py | 88 tensor_names = []
93 tensor_names.append(t.name)
94 return tensor_names
[all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
