Home
last modified time | relevance | path

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

123

/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()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/
Dtrt_convert_api.cc66 const std::vector<std::string>& output_names, in RunGrappler()
157 const std::vector<std::string>& output_names, in RunTfTrt()
195 const std::vector<std::string>& output_names, in RunSession()
228 const std::vector<std::string>& output_names, in Build()
391 const std::vector<string>& output_names, in ConvertAndBuild()
500 std::vector<std::string> output_names = GetNodeNames(signature.outputs()); in ConvertAndBuild() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dfake_session.cc112 const std::vector<std::string>& output_names, in Run()
123 const std::vector<std::string>& output_names, in Run()
133 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
200 std::vector<string> output_names = {y->name() + ":0"}; in TestHWAccelerator() local
285 std::vector<string> output_names = {y->name() + ":0"}; in TEST() local
344 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.cc468 const std::vector<string>& output_names, in Run()
484 std::vector<string> output_names(callable_options.fetch().begin(), in CreateDebuggerState() local
838 const std::vector<string>& output_names, in Run()
848 const std::vector<string>& output_names, in Run()
958 const std::vector<string>& output_names, in PRunSetup()
1037 const std::vector<string>& output_names, in PRun()
1203 const std::vector<string>& output_names, in RecvPRunOutputs()
/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.py33 def __init__(self, output_names=None): argument
106 def __init__(self, losses, loss_weights=None, output_names=None): argument
295 def __init__(self, metrics=None, weighted_metrics=None, output_names=None, argument
615 def map_to_output_names(y_pred, output_names, struct): argument
Dtraining_utils_v1.py670 def standardize_sample_or_class_weights(x_weight, output_names, weight_type): argument
716 def standardize_class_weights(class_weight, output_names): argument
721 def standardize_sample_weights(sample_weight, output_names): argument
840 output_names, argument
1461 def prepare_loss_functions(loss, output_names): argument
/external/tensorflow/tensorflow/core/profiler/backends/gpu/
Ddevice_tracer_test.cc184 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
208 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
237 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
286 std::vector<string> output_names = {y_ + ":0"}; in TEST_F() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/saved_model/
Dsaved_model.h57 llvm::ArrayRef<llvm::StringRef> output_names; member
Dsaved_model.cc111 llvm::SmallVector<llvm::StringRef, 4> output_names; in MapFunctionSignaturesFromTFSavedModelMLIR() local
/external/tensorflow/tensorflow/lite/
Dsignature_runner.h103 const std::vector<const char*>& output_names() { return output_names_; } in output_names() function
Dsignature_runner_test.cc53 const std::vector<const char*>& output_names = add_runner->output_names(); in TEST() local
/external/armnn/python/pyarmnn/examples/image_classification/
Dtflite_mobilenetv1_quantized.py44 output_names = parser.GetSubgraphOutputTensorNames(graph_id) variable
/external/tensorflow/tensorflow/core/tfrt/saved_model/
Dsaved_model_testutil.cc97 const std::vector<std::string>& output_names, in ComputeCurrentTFResult()
140 const std::vector<std::string>& output_names, in ComputeCurrentTFResult()
/external/tensorflow/tensorflow/core/debug/
Ddebugger_state_impl.cc44 const std::vector<string>& output_names, in PublishDebugMetadata()
/external/tensorflow/tensorflow/python/client/
Dsession_ref.cc257 const std::vector<string>& output_names, in RecordPRunSetup()
280 const std::vector<string>& output_names, in RecordPRun()
451 const std::vector<string>& output_names, in PRunSetup()
460 const std::vector<string>& output_names, in PRun()
Dtf_session_helper.cc62 const NameVector& output_names, in TF_Run_wrapper_helper()
166 PyObject* feed_dict, const NameVector& output_names, in TF_Run_wrapper()
317 const NameVector& output_names, in TF_PRunSetup_wrapper()
333 PyObject* feed_dict, const NameVector& output_names, in TF_PRun_wrapper()
583 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/lite/c/
Dc_api_signature_runner_test.cc58 std::vector<const char*> output_names; in TEST() local
/external/tensorflow/tensorflow/python/
Dpywrap_mlir.py29 output_names=[]): argument
/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dsaved_model_to_tfl_flatbuffer.cc108 auto output_names = output_attr.cast<mlir::StringAttr>().getValue(); in HandleInputOutputArraysWithModule() local

123