Home
last modified time | relevance | path

Searched defs:output_tensor_names (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dsharding_util_ops_test.cc52 const std::vector<std::string>& output_tensor_names, in RunGraph()
136 std::vector<std::string>* output_tensor_names) { in CreateSplitTensorGraph()
163 std::vector<std::string>* output_tensor_names) { in CreateSplitResourceGraph()
218 std::vector<std::string> output_tensor_names; in TEST_P() local
236 std::vector<std::string> output_tensor_names; in TEST_P() local
254 std::vector<std::string> output_tensor_names; in TEST_P() local
272 std::vector<std::string> output_tensor_names; in TEST_P() local
289 std::vector<std::string> output_tensor_names; in TEST_P() local
307 std::vector<std::string> output_tensor_names; in TEST_P() local
324 std::vector<std::string> output_tensor_names; in TEST_P() local
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/client/
H A Dsession_ref.cc122 const std::vector<string>& output_tensor_names, in RecordRun()
131 const std::vector<string>& output_tensor_names, in RecordRun()
412 const std::vector<string>& output_tensor_names, in Run()
421 const std::vector<string>& output_tensor_names, in Run()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
H A Dgrpc_session.cc203 const std::vector<string>& output_tensor_names, in RunHelper()
289 const std::vector<string>& output_tensor_names, in Run()
298 const std::vector<string>& output_tensor_names, in Run()
/aosp_15_r20/external/tensorflow/tensorflow/cc/saved_model/
H A Dloader.cc176 const std::vector<string>& output_tensor_names, in RunOnce()
347 const std::vector<string>& output_tensor_names, in Run()
372 const std::vector<string>& output_tensor_names, in Run()
/aosp_15_r20/external/tensorflow/tensorflow/cc/client/
H A Dclient_session.cc115 std::vector<string> output_tensor_names; in Run() local
141 std::vector<string> output_tensor_names; in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/
H A Dmodel_handler.py230 def output_tensor_names(self) -> Sequence[str]: member in _ModelHandlerBase
286 def output_tensor_names(self) -> Sequence[str]: member in ModelHandlerV1
347 def output_tensor_names(self): member in ModelHandlerV2
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/graph_executor/
H A Dgraph_executor.cc345 absl::Span<const std::string> output_tensor_names, in Run()
556 absl::Span<const std::string> output_tensor_names, in GetOrCreateLoadedClientGraph()
/aosp_15_r20/external/federated-compute/fcp/client/engine/
H A Dtflite_wrapper.h38 std::vector<std::string> output_tensor_names; member
H A Dtf_wrapper.cc148 const std::vector<std::string>& output_tensor_names, in Run()
H A Dsimple_plan_engine.cc164 const std::vector<std::string>& output_tensor_names, in RunTensorFlowInternal()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dsession.cc39 const std::vector<string>& output_tensor_names, in Run()
/aosp_15_r20/external/tensorflow/tensorflow/core/public/
H A Dsession.h182 const std::vector<std::string>& output_tensor_names, in Run()
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/codegen/
H A Dandroid_java_generator.cc183 std::vector<std::string> output_tensor_names = std::move(names.second); in CreateModelInfo() local
/aosp_15_r20/external/tensorflow/tensorflow/tools/benchmark/
H A Dbenchmark_model.cc127 std::vector<string> output_tensor_names; in GetOutputShapes() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/saved_model/
H A Dsaved_model.cc877 absl::Span<const std::string> output_tensor_names, in RunByTensorNames()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/saved_model/tests/
H A Dsaved_model_test.cc466 std::vector<std::string> output_tensor_names{"result22", "result1", in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dc_api.cc414 const std::vector<string>& output_tensor_names, TF_Tensor** c_outputs, in TF_Run_Helper()