Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/kernels/
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 …]
/external/tensorflow/tensorflow/cc/client/
Dclient_session.cc115 std::vector<string> output_tensor_names; in Run() local
141 std::vector<string> output_tensor_names; in Run() local
/external/tensorflow/tensorflow/python/client/
Dsession_ref.cc121 const std::vector<string>& output_tensor_names, in RecordRun()
130 const std::vector<string>& output_tensor_names, in RecordRun()
411 const std::vector<string>& output_tensor_names, in Run()
420 const std::vector<string>& output_tensor_names, in Run()
/external/tensorflow/tensorflow/core/common_runtime/
Dsession.cc39 const std::vector<string>& output_tensor_names, in Run()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
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()
/external/tensorflow/tensorflow/cc/saved_model/
Dloader.cc157 const std::vector<string>& output_tensor_names, in RunOnce()
325 const std::vector<string>& output_tensor_names, in Run()
350 const std::vector<string>& output_tensor_names, in Run()
/external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/
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
/external/tensorflow/tensorflow/core/public/
Dsession.h182 const std::vector<std::string>& output_tensor_names, in Run()
/external/tensorflow/tensorflow/core/tfrt/saved_model/
Dsaved_model.cc919 absl::Span<const std::string> output_tensor_names, in RunByTensorNames()
1079 absl::Span<const std::string> output_tensor_names, in GetOrCreateLoadingResult()
/external/tflite-support/tensorflow_lite_support/codegen/
Dandroid_java_generator.cc183 std::vector<std::string> output_tensor_names = std::move(names.second); in CreateModelInfo() local
/external/tensorflow/tensorflow/tools/benchmark/
Dbenchmark_model.cc123 std::vector<string> output_tensor_names; in GetOutputShapes() local
/external/tensorflow/tensorflow/c/
Dc_api.cc435 const std::vector<string>& output_tensor_names, TF_Tensor** c_outputs, in TF_Run_Helper()