HomeSort by relevance Sort by last modified time
    Searched defs:tensor_names (Results 1 - 5 of 5) sorted by null

  /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/core/kernels/
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...]
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...]
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...]
  /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...]

Completed in 98 milliseconds