Home
last modified time | relevance | path

Searched refs:GetLocalTensors (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dkernel_and_device.cc69 std::vector<Tensor> EagerKernelArgs::GetLocalTensors() const { in GetLocalTensors() function in tensorflow::EagerKernelArgs
443 s.Update(pflr_->RunSync(*opts, handle_, inputs.GetLocalTensors(), &rets)); in Run()
H A Dkernel_and_device.h84 std::vector<Tensor> GetLocalTensors() const override;
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dprocess_function_library_runtime.h50 virtual std::vector<Tensor> GetLocalTensors() const = 0;
H A Dprocess_function_library_runtime_test.cc880 std::vector<Tensor> GetLocalTensors() const override { return {}; } in GetLocalTensors() function in tensorflow::__anon3261924c0111::TestFunctionPackedArgs
H A Dprocess_function_library_runtime.cc1966 const std::vector<Tensor> local_inputs = args.GetLocalTensors(); in Run()