Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.h78 std::vector<Tensor> GetLocalTensors() const override;
Dkernel_and_device.cc68 std::vector<Tensor> EagerKernelArgs::GetLocalTensors() const { in GetLocalTensors() function in tensorflow::EagerKernelArgs
/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime.h49 virtual std::vector<Tensor> GetLocalTensors() const = 0;
Dprocess_function_library_runtime_test.cc831 std::vector<Tensor> GetLocalTensors() const override { return {}; } in GetLocalTensors() function in tensorflow::__anon96a4e55d0111::TestFunctionPackedArgs
Dprocess_function_library_runtime.cc1651 const std::vector<Tensor> local_inputs = args.GetLocalTensors(); in Run()