Home
last modified time | relevance | path

Searched defs:output_names (Results 1 – 25 of 38) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dmlir_roundtrip_flags.cc37 std::vector<string> output_names = absl::StrSplit(array_names, ','); in ParseOutputArrayInfo() local
41 Status ParseOutputArrayInfo(const std::vector<string>& output_names, in ParseOutputArrayInfo()
/external/tensorflow/tensorflow/python/keras/engine/
Dcompile_utils.py37 def __init__(self, losses, loss_weights=None, output_names=None): argument
199 def __init__(self, metrics=None, weighted_metrics=None, output_names=None): argument
465 def map_to_output_names(y_pred, output_names, struct): argument
Dtraining_utils.py586 def standardize_sample_or_class_weights(x_weight, output_names, weight_type): argument
632 def standardize_class_weights(class_weight, output_names): argument
637 def standardize_sample_weights(sample_weight, output_names): argument
816 output_names, argument
1464 def prepare_loss_functions(loss, output_names): argument
/external/tensorflow/tensorflow/core/common_runtime/
Dsession.cc48 const std::vector<string>& output_names, in PRunSetup()
57 const std::vector<string>& output_names, in PRun()
Ddirect_session_with_tracking_alloc_test.cc88 std::vector<string> output_names = {y->name() + ":0"}; in TEST() local
208 std::vector<string> output_names = {y->name() + ":0"}; in TestHWAccelerator() local
292 std::vector<string> output_names = {y->name() + ":0"}; in TEST() local
350 std::vector<string> output_names = {y->name() + ":0"}; in TEST() local
Dsession_state.cc70 Status TensorStore::SaveTensors(const std::vector<string>& output_names, in SaveTensors()
Ddirect_session.cc454 const std::vector<string>& output_names, in Run()
470 std::vector<string> output_names(callable_options.fetch().begin(), in CreateDebuggerState() local
807 const std::vector<string>& output_names, in Run()
817 const std::vector<string>& output_names, in Run()
927 const std::vector<string>& output_names, in PRunSetup()
1006 const std::vector<string>& output_names, in PRun()
1172 const std::vector<string>& output_names, in RecvPRunOutputs()
Dgraph_runner.cc103 const std::vector<string>& output_names, in Run()
Ddirect_session_test.cc141 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
212 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
244 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
505 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
563 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
633 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
717 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
783 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
/external/tensorflow/tensorflow/core/profiler/internal/gpu/
Ddevice_tracer_test.cc180 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
208 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
233 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
262 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
/external/tensorflow/tensorflow/core/debug/
Ddebugger_state_impl.cc44 const std::vector<string>& output_names, in PublishDebugMetadata()
/external/tensorflow/tensorflow/core/framework/
Drun_handler_test.cc152 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
185 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
Dgraph_to_functiondef.cc304 const std::vector<string>& output_names, FunctionDef* fdef) { in GraphToFunctionDefHelper()
380 const std::vector<string>& output_names, in GraphToFunctionDef()
596 const std::vector<std::string>& output_names, in GraphToFunctionDef()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.h37 const std::vector<string>* output_names = nullptr; member
Dconvert_graph_test.cc161 const std::vector<string> output_names{"output"}; in RunConvertAfterShape() local
/external/tensorflow/tensorflow/python/client/
Dsession_ref.cc255 const std::vector<string>& output_names, in RecordPRunSetup()
278 const std::vector<string>& output_names, in RecordPRun()
449 const std::vector<string>& output_names, in PRunSetup()
458 const std::vector<string>& output_names, in PRun()
Dtf_session_helper.cc61 const NameVector& output_names, in TF_Run_wrapper_helper()
165 PyObject* feed_dict, const NameVector& output_names, in TF_Run_wrapper()
321 const NameVector& output_names, in TF_PRunSetup_wrapper()
337 PyObject* feed_dict, const NameVector& output_names, in TF_PRun_wrapper()
587 const NameVector& output_names, in TF_GraphToFunction_wrapper()
/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
/external/tensorflow/tensorflow/c/
Dc_api_function.cc154 int noutputs, const TF_Output* outputs, const char* const* output_names, in TF_GraphToFunctionWithControlOutputs()
224 const char* const* output_names, in TF_GraphToFunction()
Dc_api.cc482 std::vector<string> output_names(noutputs); in TF_Run() local
505 std::vector<string> output_names(noutputs); in TF_PRunSetup() local
541 std::vector<string> output_names(noutputs); in TF_PRun() local
2292 std::vector<string> output_names(noutputs); in TF_SessionRun() local
2325 std::vector<string> output_names(noutputs); in TF_SessionPRunSetup() local
2374 std::vector<string> output_names(noutputs); in TF_SessionPRun() local
/external/tensorflow/tensorflow/tools/graph_transforms/
Dtransform_utils.h224 std::vector<string> output_names; member
Dinsert_logging_test.cc39 const std::vector<string>& output_names) { in CheckGraphCanRun()
Dquantize_nodes_test.cc63 const std::vector<string>& output_names, in TestTransformedVersusFloatGraph()
101 const std::vector<string>& output_names) { in TestQuantizedVersusFloatGraph()
121 const std::vector<string>& output_names, float range_min, in TestGraphWithInputRange()
145 const std::vector<string>& output_names, float range_min, float range_max, in TestGraphWithFallbackRange()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session.cc316 const std::vector<string>& output_names, in PRunSetup()
342 const std::vector<string>& output_names, in PRun()
/external/tensorflow/tensorflow/cc/saved_model/
Dloader.cc408 const std::vector<string>& output_names, in PRunSetup()
416 const std::vector<string>& output_names, in PRun()

12