Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlift_variables_test_pass.h70 const std::vector<std::string>& output_names, in Run()
80 const std::vector<std::string>& output_names, in Run()
90 const std::vector<std::string>& output_names, in Run()
Dlift_variables_test_pass.cc67 const std::vector<std::string>& output_names, in Run()
77 const std::vector<std::string>& output_names, in Run()
87 const std::vector<std::string>& output_names, in Run()
/external/tensorflow/tensorflow/core/common_runtime/
Dsession.cc47 const std::vector<string>& output_names, in PRunSetup()
56 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
199 std::vector<string> output_names = {y->name() + ":0"}; in TestHWAccelerator() local
283 std::vector<string> output_names = {y->name() + ":0"}; in TEST() local
341 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.cc457 const std::vector<string>& output_names, in Run()
473 std::vector<string> output_names(callable_options.fetch().begin(), in CreateDebuggerState() local
821 const std::vector<string>& output_names, in Run()
831 const std::vector<string>& output_names, in Run()
941 const std::vector<string>& output_names, in PRunSetup()
1020 const std::vector<string>& output_names, in PRun()
1186 const std::vector<string>& output_names, in RecvPRunOutputs()
Dgraph_runner.cc105 const std::vector<string>& output_names, in Run()
/external/tflite-support/tensorflow_lite_support/codegen/
Dcode_generator.cc75 std::vector<std::string> output_names; in NameInputsAndOutputs() local
150 auto& output_names = *outputs; in ResolveConflictedInputAndOutputNames() local
/external/tensorflow/tensorflow/python/keras/engine/
Dcompile_utils.py38 def __init__(self, output_names=None): argument
111 def __init__(self, losses, loss_weights=None, output_names=None): argument
287 def __init__(self, metrics=None, weighted_metrics=None, output_names=None): argument
570 def map_to_output_names(y_pred, output_names, struct): argument
Dtraining_utils_v1.py676 def standardize_sample_or_class_weights(x_weight, output_names, weight_type): argument
722 def standardize_class_weights(class_weight, output_names): argument
727 def standardize_sample_weights(sample_weight, output_names): argument
846 output_names, argument
1458 def prepare_loss_functions(loss, output_names): argument
/external/tensorflow/tensorflow/core/profiler/internal/gpu/
Ddevice_tracer_test.cc177 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
204 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
229 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
257 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/python/client/
Dsession_ref.cc256 const std::vector<string>& output_names, in RecordPRunSetup()
279 const std::vector<string>& output_names, in RecordPRun()
450 const std::vector<string>& output_names, in PRunSetup()
459 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()
316 const NameVector& output_names, in TF_PRunSetup_wrapper()
332 PyObject* feed_dict, const NameVector& output_names, in TF_PRun_wrapper()
582 const NameVector& output_names, in TF_GraphToFunction_wrapper()
/external/tensorflow/tensorflow/core/framework/
Dgraph_to_functiondef.cc298 const std::vector<string>& output_names, FunctionDef* fdef) { in GraphToFunctionDefHelper()
374 const std::vector<string>& output_names, in GraphToFunctionDef()
600 const std::vector<std::string>& output_names, in GraphToFunctionDef()
Drun_handler_test.cc657 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
690 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
721 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
754 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.h35 const std::vector<string>* output_names = nullptr; member
Dconvert_graph_test.cc163 const std::vector<string> output_names{"output"}; in RunConvertAfterShape() local
/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dsaved_model_to_tfl_flatbuffer.cc102 auto output_names = output_attr.cast<mlir::StringAttr>().getValue(); in HandleInputOutputArraysWithModule() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dmlir_roundtrip_flags.cc43 Status ParseOutputArrayInfo(const std::vector<string>& output_names, in ParseOutputArrayInfo()
/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()
230 const char* const* output_names, in TF_GraphToFunction()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dinsert_logging_test.cc39 const std::vector<string>& output_names) { in CheckGraphCanRun()
Dtransform_utils.h224 std::vector<string> output_names; member
/external/tensorflow/tensorflow/cc/saved_model/
Dloader.cc352 const std::vector<string>& output_names, in PRunSetup()
360 const std::vector<string>& output_names, in PRun()

12