Home
last modified time | relevance | path

Searched defs:retvals (Results 1 – 25 of 43) sorted by relevance

12

/external/tensorflow/tensorflow/core/common_runtime/
Dsingle_threaded_executor_test.cc216 std::vector<Tensor> retvals; in TEST_F() local
244 std::vector<Tensor> retvals; in TEST_F() local
274 std::vector<Tensor> retvals; in TEST_F() local
322 std::vector<Tensor> retvals; in TEST_F() local
354 std::vector<Tensor> retvals; in TEST_F() local
/external/tensorflow/tensorflow/core/distributed_runtime/integration_test/
Dc_api_multi_client_test.cc100 TFE_TensorHandle* retvals[1]; in TEST() local
192 TFE_TensorHandle* retvals[1]; in TEST() local
204 TFE_TensorHandle* retvals[1]; in TEST() local
Dc_api_coordination_test.cc156 TFE_TensorHandle* retvals[1]; in TEST() local
232 TFE_TensorHandle* retvals[1]; in TEST() local
381 TFE_TensorHandle* retvals[1]; in TEST() local
473 TFE_TensorHandle* retvals[1]; in TEST_P() local
Dc_api_multi_client_function_test.cc325 TFE_TensorHandle* retvals[1]; in TEST_P() local
/external/tensorflow/tensorflow/c/eager/
Dc_api_distributed_test.cc205 TFE_TensorHandle* retvals[1] = {nullptr}; in TestFunctionWithPackedInput() local
391 TFE_TensorHandle* retvals[1] = {nullptr}; in TEST() local
515 TFE_TensorHandle* retvals[1] = {nullptr}; in TestDistributedFunctionCancellation() local
597 TFE_TensorHandle* retvals[1]; in TestRemoteExecuteDeleteContextWithOutstandingRPC() local
Dc_api_test.cc90 TFE_TensorHandle* retvals[1]; in BM_Execute() local
128 TFE_TensorHandle* retvals[1]; in BM_Execute_Identity() local
438 TFE_TensorHandle* retvals[1]; in TensorHandleSilentCopy() local
574 TFE_TensorHandle* retvals[1]; in TEST() local
757 TFE_TensorHandle* retvals[2] = {nullptr, nullptr}; in Execute_MatMul_CPU() local
801 TFE_TensorHandle* retvals[1] = {nullptr}; in Execute_MatMul_CPU_Runtime_Error() local
866 TFE_TensorHandle* retvals[1] = {nullptr}; in Execute_MatMul_CPU_Type_Error() local
896 TFE_TensorHandle* retvals[1] = {nullptr}; in TEST() local
930 TFE_TensorHandle* retvals[1] = {nullptr}; in ExecuteWithTracing() local
1435 TFE_TensorHandle* retvals[1]; in TEST() local
[all …]
Dc_api_remote_test.cc81 TFE_TensorHandle* retvals[1]; in TestRemoteExecute() local
Dc_api_remote_test_util.cc170 TFE_TensorHandle* retvals[1]; in TestRemoteExecuteSilentCopies() local
Dunified_api_testutil.cc102 int retvals = outputs.size() - null_indices.size(); in RunModel() local
Dc_api_experimental_test.cc335 TFE_TensorHandle* retvals[2] = {nullptr, nullptr}; in Executor_MatMul_CPU() local
461 TFE_TensorHandle* retvals[1]; in TEST() local
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.cc1049 EagerOperation* op, TensorHandle** retvals, int* num_retvals, in GetOrCreateKernelAndDevice()
1309 EagerOperation* op, TensorHandle** retvals) { in AddOrExecuteNode()
1393 Status EagerLocalExecute(EagerOperation* op, TensorHandle** retvals, in EagerLocalExecute()
1484 absl::FixedArray<tensorflow::TensorHandle*> retvals(num_retvals); in MaybePackInputTensor() local
1509 TensorHandle** retvals) { in StoreResourceDtypesAndShapes()
1529 Status EagerRemoteExecute(EagerOperation* op, TensorHandle** retvals, in EagerRemoteExecute()
1700 TensorHandle** retvals, EagerContext* ctx, KernelAndDevice* kernel, in GetKernelOutputs()
1782 Status EagerExecute(EagerOperation* op, TensorHandle** retvals, in EagerExecute()
1832 absl::Span<TensorHandle*> retvals, in EagerKernelExecute()
2046 TensorHandle** retvals, int num_outputs, StatusCallback done) { in EagerKernelExecuteAsync()
[all …]
Dshape_inference.cc31 const gtl::InlinedVector<TensorHandle*, 2>& retvals) { in RunShapeInference()
Dexecute_node.h100 absl::Span<TensorHandle*> retvals, in ExecuteNode()
156 absl::Span<TensorHandle*> retvals, in AsyncExecuteNode()
Dcontext_test.cc279 std::vector<AbstractTensorHandle*> retvals(1); in TEST_F() local
334 std::vector<AbstractTensorHandle*> retvals(1); in TEST_F() local
Dcustom_device_op_handler.cc55 ImmediateExecutionTensorHandle** retvals, in Execute()
Dcore.cc175 Status EagerOperation::Execute(absl::Span<AbstractTensorHandle*> retvals, in Execute()
Dcustom_device_test.cc53 ImmediateExecutionTensorHandle** retvals, in Execute()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_context.h71 const std::vector<XlaExpression>& retvals() { return retvals_; } in retvals() function
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_execute_node.cc31 const gtl::InlinedVector<TensorHandle*, 2>& retvals = retvals_; in RunAsync() local
Dremote_execute_node.h47 absl::Span<TensorHandle*> retvals) in RemoteExecuteNode()
Deager_service_impl.cc174 const std::vector<int32>& output_nums, TensorHandle** retvals, in AddOpRetvalsToResponse()
511 auto* retvals = new absl::FixedArray<TensorHandle*>(*num_retvals); in RunComponentFunction() local
566 absl::FixedArray<tensorflow::TensorHandle*> retvals(num_retvals); in ExecuteOp() local
/external/tensorflow/tensorflow/core/function/
Druntime_client.cc168 std::vector<ImmediateExecutionTensorHandle*> retvals(num_retvals); in CallFunction() local
/external/tensorflow/tensorflow/core/protobuf/tpu/
Dcompile_metadata.proto78 repeated Retval retvals = 2; field
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_xla_computations_pass.cc124 std::vector<Node*> retvals; in RewriteSubgraph() local
/external/tensorflow/tensorflow/c/experimental/gradients/tape/
Dtape_operation.cc197 Status TapeOperation::Execute(absl::Span<AbstractTensorHandle*> retvals, in Execute()

12