Home
last modified time | relevance | path

Searched refs:output_names (Results 1 – 25 of 109) sorted by relevance

12345

/external/tflite-support/tensorflow_lite_support/codegen/
Dcode_generator.cc75 std::vector<std::string> output_names; in NameInputsAndOutputs() local
83 output_names.reserve(outputs->size()); in NameInputsAndOutputs()
85 output_names.push_back(NameTensor(*tensor, "output")); in NameInputsAndOutputs()
89 ResolveConflictedInputAndOutputNames(&input_names, &output_names); in NameInputsAndOutputs()
90 return std::make_pair(input_names, output_names); in NameInputsAndOutputs()
150 auto& output_names = *outputs; in ResolveConflictedInputAndOutputNames() local
155 for (const auto& output : output_names) { in ResolveConflictedInputAndOutputNames()
167 for (int i = 0; i < output_names.size(); i++) { in ResolveConflictedInputAndOutputNames()
168 if (io_conflict.find(output_names[i]) != io_conflict.end()) { in ResolveConflictedInputAndOutputNames()
169 output_names[i] = "output_" + output_names[i]; in ResolveConflictedInputAndOutputNames()
[all …]
/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.cc61 const NameVector& output_names, in TF_Run_wrapper_helper() argument
102 TF_TensorVector outputs(output_names.size()); in TF_Run_wrapper_helper()
112 input_names.size(), const_cast<const char**>(output_names.data()), in TF_Run_wrapper_helper()
113 outputs.data(), output_names.size(), in TF_Run_wrapper_helper()
118 input_names.size(), const_cast<const char**>(output_names.data()), in TF_Run_wrapper_helper()
119 outputs.data(), output_names.size(), in TF_Run_wrapper_helper()
143 for (size_t i = 0; i < output_names.size(); ++i) { in TF_Run_wrapper_helper()
165 PyObject* feed_dict, const NameVector& output_names, in TF_Run_wrapper() argument
168 TF_Run_wrapper_helper(session, nullptr, run_options, feed_dict, output_names, in TF_Run_wrapper()
316 const NameVector& output_names, in TF_PRunSetup_wrapper() argument
[all …]
Dsession_ref.cc256 const std::vector<string>& output_names, in RecordPRunSetup() argument
265 for (auto& output : output_names) { in RecordPRunSetup()
271 RUN_WITH_TIMESTAMP(PRunSetup, input_names, output_names, target_nodes, in RecordPRunSetup()
279 const std::vector<string>& output_names, in RecordPRun() argument
294 for (auto& output : output_names) { in RecordPRun()
298 RUN_WITH_TIMESTAMP(PRun, handle, inputs, output_names, outputs); in RecordPRun()
304 tproto->set_name(output_names[i]); in RecordPRun()
450 const std::vector<string>& output_names, in PRunSetup() argument
453 LOG_AND_RUN_OPERATION(PRunSetup, input_names, output_names, target_nodes, in PRunSetup()
459 const std::vector<string>& output_names, in PRun() argument
[all …]
Dtf_session_helper.h61 PyObject* feed_dict, const NameVector& output_names,
98 const NameVector& output_names,
115 PyObject* feed_dict, const NameVector& output_names,
210 const NameVector& output_names,
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlift_variables_test_pass.cc67 const std::vector<std::string>& output_names, in Run() argument
71 return Run(tensorflow::RunOptions(), inputs, output_names, target_nodes, in Run()
77 const std::vector<std::string>& output_names, in Run() argument
81 return Run(run_options, inputs, output_names, target_nodes, outputs, in Run()
87 const std::vector<std::string>& output_names, in Run() argument
93 for (const std::string& output_name : output_names) { in Run()
Dlift_variables_test_pass.h70 const std::vector<std::string>& output_names, in Run() argument
74 return Run(tensorflow::RunOptions(), inputs, output_names, target_nodes, in Run()
80 const std::vector<std::string>& output_names, in Run() argument
84 return Run(run_options, inputs, output_names, target_nodes, outputs, in Run()
90 const std::vector<std::string>& output_names, in Run() argument
96 for (const std::string& output_name : output_names) { in Run()
/external/tensorflow/tensorflow/python/keras/saving/
Dsaving_utils_test.py80 if model.output_names:
81 expected_outputs = {model.output_names[0]: model(inputs)}
105 if model.output_names:
106 expected_outputs = {model.output_names[0]: model(inputs)}
149 if model.output_names:
151 model.output_names[0]: outputs[0],
152 model.output_names[1]: outputs[1]
190 if model.output_names:
191 expected_outputs = {model.output_names[0]: model(inputs)}
330 if model.output_names:
[all …]
Dsaving_utils.py137 output_names = model.output_names # Functional Model.
138 if output_names is None: # Subclassed Model.
140 output_names = compile_utils.create_pseudo_output_names(outputs)
142 return {name: output for name, output in zip(output_names, outputs)}
/external/tensorflow/tensorflow/python/keras/utils/
Dmulti_gpu_utils.py243 for n in model.output_names:
249 output_names = []
250 for n in model.output_names:
254 output_names.append(n)
259 for name, outputs in zip(output_names, all_outputs):
/external/tensorflow/tensorflow/python/keras/engine/
Dcompile_utils.py38 def __init__(self, output_names=None): argument
39 self._output_names = output_names
111 def __init__(self, losses, loss_weights=None, output_names=None): argument
112 super(LossesContainer, self).__init__(output_names=output_names)
287 def __init__(self, metrics=None, weighted_metrics=None, output_names=None): argument
288 super(MetricsContainer, self).__init__(output_names=output_names)
570 def map_to_output_names(y_pred, output_names, struct): argument
603 output_names = output_names or create_pseudo_output_names(y_pred)
605 new_struct = [struct.pop(name, None) for name in output_names]
609 struct.keys(), output_names))
Dtraining_utils_v1.py676 def standardize_sample_or_class_weights(x_weight, output_names, weight_type): argument
693 return [None for _ in output_names]
694 if len(output_names) == 1:
697 if isinstance(x_weight, dict) and output_names[0] in x_weight:
698 return [x_weight[output_names[0]]]
702 if len(x_weight) != len(output_names):
705 str(len(output_names)) + ' outputs. '
710 generic_utils.check_for_unexpected_keys(weight_type, x_weight, output_names)
712 for name in output_names:
722 def standardize_class_weights(class_weight, output_names): argument
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_runner.cc105 const std::vector<string>& output_names, in Run() argument
147 graph_to_run.get(), input_names, output_names, {} /* target nodes */, in Run()
191 outputs->resize(output_names.size()); in Run()
192 for (size_t i = 0; i < output_names.size(); ++i) { in Run()
195 output_names[i], FrameAndIter(0, 0)); in Run()
Ddirect_session_with_tracking_alloc_test.cc88 std::vector<string> output_names = {y->name() + ":0"}; in TEST() local
92 Status s = session->Run(inputs, output_names, target_nodes, &outputs); in TEST()
199 std::vector<string> output_names = {y->name() + ":0"}; in TestHWAccelerator() local
209 Status s = session->Run(run_options, inputs, output_names, target_nodes, in TestHWAccelerator()
283 std::vector<string> output_names = {y->name() + ":0"}; in TEST() local
288 Status s = session->Run(run_options, inputs, output_names, target_nodes, in TEST()
341 std::vector<string> output_names = {y->name() + ":0"}; in TEST() local
344 Status s = session->Run(run_options, inputs, output_names, {}, &outputs, in TEST()
Ddirect_session.cc457 const std::vector<string>& output_names, in Run() argument
461 return Run(RunOptions(), inputs, output_names, target_nodes, outputs, in Run()
473 std::vector<string> output_names(callable_options.fetch().begin(), in CreateDebuggerState() local
480 output_names, target_names)); in CreateDebuggerState()
821 const std::vector<string>& output_names, in Run() argument
825 return Run(run_options, inputs, output_names, target_nodes, outputs, in Run()
831 const std::vector<string>& output_names, in Run() argument
856 TF_RETURN_IF_ERROR(GetOrCreateExecutors(input_tensor_names, output_names, in Run()
908 output_names.size() == executors_and_keys->output_name_to_index.size(); in Run()
913 first_indices.reserve(output_names.size()); in Run()
[all …]
Ddirect_session.h75 const std::vector<string>& output_names,
82 const std::vector<string>& output_names,
90 const NamedTensorList& inputs, const std::vector<string>& output_names,
98 const std::vector<string>& output_names,
102 const std::vector<string>& output_names,
292 const std::vector<string>& output_names,
Dsession_state.cc70 Status TensorStore::SaveTensors(const std::vector<string>& output_names, in SaveTensors() argument
75 for (const string& name : output_names) { in SaveTensors()
/external/tensorflow/tensorflow/core/profiler/internal/gpu/
Ddevice_tracer_test.cc177 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
182 Status s = session->Run(inputs, output_names, target_nodes, &outputs); in TEST_F()
204 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
209 Status s = session->Run(inputs, output_names, target_nodes, &outputs); in TEST_F()
229 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
237 Status s = session->Run(run_options, inputs, output_names, target_nodes, in TEST_F()
257 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
262 Status s = session->Run(inputs, output_names, target_nodes, &outputs); in TEST_F()
/external/tensorflow/tensorflow/core/framework/
Dgraph_to_functiondef.cc298 const std::vector<string>& output_names, FunctionDef* fdef) { in GraphToFunctionDefHelper() argument
360 body_nodes, inputs, outputs, output_names, in GraphToFunctionDefHelper()
374 const std::vector<string>& output_names, in GraphToFunctionDef() argument
378 if (!output_names.empty()) { in GraphToFunctionDef()
379 DCHECK_EQ(output_names.size(), outputs.size()); in GraphToFunctionDef()
413 if (!output_names.empty()) { in GraphToFunctionDef()
414 TF_RETURN_IF_ERROR(node_names.UseOutputName(output_names[i])); in GraphToFunctionDef()
415 argdef->set_name(output_names[i]); in GraphToFunctionDef()
600 const std::vector<std::string>& output_names, in GraphToFunctionDef() argument
603 output_names, fdef); in GraphToFunctionDef()
Drun_handler_test.cc657 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
665 Status s = session->Run(run_options, inputs, output_names, target_nodes, in TEST_F()
690 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
691 auto fn = [&session, output_names, run_options]() { in TEST_F()
696 Status s = session->Run(run_options, inputs, output_names, {}, &outputs, in TEST_F()
721 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
732 Status s = session->Run(run_options, inputs, output_names, target_nodes, in TEST_F()
754 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
755 auto fn = [&session, output_names]() { in TEST_F()
765 Status s = session->Run(run_options, inputs, output_names, {}, &outputs, in TEST_F()
Dgraph_to_functiondef.h38 const std::vector<string>& output_names,
60 const std::vector<std::string>& output_names,
/external/tensorflow/tensorflow/tools/graph_transforms/
Dinsert_logging_test.cc39 const std::vector<string>& output_names) { in CheckGraphCanRun() argument
43 TF_ASSERT_OK(session->Run({}, output_names, {}, &outputs)); in CheckGraphCanRun()
71 context.output_names = {"add_node4"}; in TestInsertLogging()
118 context.output_names = {"add_node4"}; in TestInsertLoggingByOpType()
167 context.output_names = {"add_node4"}; in TestInsertLoggingByPrefix()
/external/tensorflow/tensorflow/c/
Dc_api_function.cc154 int noutputs, const TF_Output* outputs, const char* const* output_names, in TF_GraphToFunctionWithControlOutputs() argument
175 if (output_names) { in TF_GraphToFunctionWithControlOutputs()
178 output_names_vec.push_back(string(output_names[i])); in TF_GraphToFunctionWithControlOutputs()
187 control_output_names_vec.push_back(string(output_names[i])); in TF_GraphToFunctionWithControlOutputs()
230 const char* const* output_names, in TF_GraphToFunction() argument
235 inputs, noutputs, outputs, output_names, 0, nullptr, nullptr, opts, in TF_GraphToFunction()
/external/tensorflow/tensorflow/core/debug/
Ddebugger_state_impl.cc44 const std::vector<string>& output_names, in PublishDebugMetadata() argument
48 output_names, target_names, debug_urls_); in PublishDebugMetadata()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_activity_listener_test.cc102 std::vector<std::string> output_names = {std::string("add_4:0")}; in TEST_F() local
117 TF_ASSERT_OK(session->Run(inputs_2x2, output_names, /*target_node_names=*/{}, in TEST_F()
177 TF_ASSERT_OK(session->Run(inputs_3x3, output_names, in TEST_F()
/external/tensorflow/tensorflow/tools/graph_transforms/python/
Dtransform_graph_test.py74 output_names = ["add_op"]
77 output_names, transforms)

12345