| /external/tensorflow/tensorflow/contrib/tensorrt/convert/ |
| convert_graph.h | 37 const std::vector<string>& output_names, size_t max_batch_size,
|
| convert_graph.cc | 114 const std::vector<string>& output_names,
131 // Collect outputs referenced from output_names
132 auto output_name_to_index_map = BuildTensorNameMap(output_names);
200 const std::vector<string>& output_names, size_t max_batch_size,
204 item.fetch = output_names;
241 for (auto node : output_names) {
261 output_names, subgraph_node_ids, max_batch_size,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| graph_runner.h | 47 // Function semantics for `inputs`, `output_names` and `outputs`
58 const std::vector<string>& output_names,
|
| session.cc | 39 const std::vector<string>& output_names,
48 const std::vector<string>& output_names,
|
| session_state.cc | 66 Status TensorStore::SaveTensors(const std::vector<string>& output_names,
70 // Save only the tensors in output_names in the session.
71 for (const string& name : output_names) {
|
| graph_runner.cc | 106 const std::vector<string>& output_names,
148 graph_to_run.get(), input_names, output_names, {} /* target nodes */,
187 outputs->resize(output_names.size());
188 for (size_t i = 0; i < output_names.size(); ++i) {
191 output_names[i], FrameAndIter(0, 0));
|
| direct_session.h | 72 const std::vector<string>& output_names,
79 const std::vector<string>& output_names,
87 const std::vector<string>& output_names,
91 const std::vector<string>& output_names,
234 const std::vector<string>& output_names,
263 const std::vector<string>& output_names,
|
| direct_session_with_tracking_alloc_test.cc | 82 std::vector<string> output_names = {y->name() + ":0"}; local
86 Status s = session->Run(inputs, output_names, target_nodes, &outputs);
192 std::vector<string> output_names = {y->name() + ":0"}; local
202 Status s = session->Run(run_options, inputs, output_names, target_nodes,
276 std::vector<string> output_names = {y->name() + ":0"}; local
281 Status s = session->Run(run_options, inputs, output_names, target_nodes,
334 std::vector<string> output_names = {y->name() + ":0"}; local
337 Status s = session->Run(run_options, inputs, output_names, {}, &outputs,
|
| direct_session.cc | 403 const std::vector<string>& output_names,
407 return Run(RunOptions(), inputs, output_names, target_nodes, outputs,
414 const std::vector<string>& output_names,
421 input_names, output_names, target_names));
438 const std::vector<string>& output_names,
474 TF_RETURN_IF_ERROR(GetOrCreateExecutors(input_tensor_names, output_names,
483 input_tensor_names, output_names, target_nodes, &debugger_state));
641 output_names.size() == executors_and_keys->output_name_to_index.size();
643 // output_names[i] == output_names[j]
[all...] |
| debugger_state_interface.h | 49 // output_names: Names of the fetched Tensors.
54 const std::vector<string>& output_names,
|
| /external/tensorflow/tensorflow/contrib/predictor/ |
| saved_model_predictor.py | 94 output_names):
99 output_names_specified = output_names is not None
102 'input_names and output_names must both be specified or both be '
110 '(input_names AND output_names).'
122 output_names=None,
138 output_names: A dictionary mapping strings to `Tensor`s in the
147 (input_names AND output_names) is specified.
150 signature_def_key, signature_def, input_names, output_names)
162 output_names = {k: v.name for k, v in signature_def.outputs.items()}
167 for k, v in output_names.items()
[all...] |
| /external/tensorflow/tensorflow/contrib/tensorrt/ |
| trt_conversion.i | 71 std::vector<string> output_names,
93 if (!output_names.size()) {
94 out_status = "InvalidArgument;Size of the output_names vector is 0";
101 graph_def, output_names, max_batch_size, max_workspace_size_bytes,
126 std::vector<string> output_names,
|
| /external/tensorflow/tensorflow/core/debug/ |
| debugger_state_impl.cc | 41 const std::vector<string>& output_names,
45 output_names, target_names, debug_urls_);
|
| debugger_state_impl.h | 39 const std::vector<string>& output_names,
|
| /external/tensorflow/tensorflow/python/client/ |
| tf_session_helper.cc | 47 const NameVector& output_names,
88 TF_TensorVector outputs(output_names.size());
98 input_names.size(), const_cast<const char**>(output_names.data()),
99 outputs.data(), output_names.size(),
104 input_names.size(), const_cast<const char**>(output_names.data()),
105 outputs.data(), output_names.size(),
129 for (size_t i = 0; i < output_names.size(); ++i) {
150 PyObject* feed_dict, const NameVector& output_names,
153 TF_Run_wrapper_helper(session, nullptr, run_options, feed_dict, output_names,
162 const NameVector& output_names,
[all...] |
| tf_session_helper.h | 51 // On success, the tensors corresponding to output_names[0,noutputs-1]
58 PyObject* feed_dict, const NameVector& output_names,
71 const NameVector& output_names,
81 // On success, the tensors corresponding to output_names[0,noutputs-1]
88 PyObject* feed_dict, const NameVector& output_names,
176 // `output_names` must be empty or have the same length as `outputs`.
181 const NameVector& output_names, const TF_FunctionOptions* opts,
|
| /external/tensorflow/tensorflow/core/framework/ |
| session_state.h | 74 Status SaveTensors(const std::vector<string>& output_names,
|
| /external/tensorflow/tensorflow/core/public/ |
| session.h | 157 /// fetches are specified by `input_names` and `output_names`. Returns
162 const std::vector<string>& output_names,
168 /// in `output_names`.
172 const std::vector<string>& output_names,
|
| /external/tensorflow/tensorflow/tools/graph_transforms/python/ |
| transform_graph_test.py | 74 output_names = ["add_op"]
77 output_names, transforms)
|
| /external/tensorflow/tensorflow/core/platform/ |
| device_tracer_test.cc | 160 std::vector<string> output_names = {y_ + ":0"}; local
165 Status s = session->Run(inputs, output_names, target_nodes, &outputs);
187 std::vector<string> output_names = {y_ + ":0"}; local
192 Status s = session->Run(inputs, output_names, target_nodes, &outputs);
213 std::vector<string> output_names = {y_ + ":0"}; local
221 Status s = session->Run(run_options, inputs, output_names, target_nodes,
|
| /external/tensorflow/tensorflow/contrib/meta_graph_transform/ |
| meta_graph_transform.py | 70 def _gtt_transforms(graph_def, input_names, output_names, initializer_names,
77 output_names: Names of output nodes.
94 all_output_names = output_names + initializer_names_flat
99 def _freeze_transform(graph_def, output_names, initializer_names, saver_def,
108 output_names: Names of output nodes.
124 graph_def = _freeze_graph_with_def_protos(graph_def, output_names,
185 def _sparsify_gather_transform(graph_def, input_names, output_names,
195 output_names: Names of output nodes.
222 graph_def = _gtt_transforms(graph_def, input_names, output_names,
265 output_names,
[all...] |
| meta_graph_transform_test.py | 78 output_names = ['mul:0']
83 base_meta_graph_def, input_names, output_names, transforms, tags)
107 output_names = ['o1', 'o2']
116 graph_def, input_names, output_names, initializer_names, transforms))
120 graph_def, input_names, output_names + init_nodes, transforms)
126 output_names = ['o1', 'o2']
143 meta_graph_transform._freeze_transform(graph_def, output_names,
153 freeze_mock.assert_called_once_with(graph_def, output_names,
251 output_names = ['o1', 'o2']
289 graph_def, input_names, output_names, initializer_names
[all...] |
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| insert_logging_test.cc | 39 const std::vector<string>& output_names) {
43 TF_ASSERT_OK(session->Run({}, output_names, {}, &outputs));
71 context.output_names = {"add_node4"};
118 context.output_names = {"add_node4"};
167 context.output_names = {"add_node4"};
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| grpc_session.h | 86 const std::vector<string>& output_names,
94 const std::vector<string>& output_names,
|
| /external/tensorflow/tensorflow/python/keras/_impl/keras/engine/ |
| training.py | 158 def _standardize_sample_or_class_weights(x_weight, output_names, weight_type):
163 output_names: List of output names (strings) in the model.
174 return [None for _ in output_names]
175 if len(output_names) == 1:
178 if isinstance(x_weight, dict) and output_names[0] in x_weight:
179 return [x_weight[output_names[0]]]
183 if len(x_weight) != len(output_names):
186 str(len(output_names)) + ' outputs. '
192 for name in output_names:
202 def _standardize_class_weights(class_weight, output_names)
[all...] |
