Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.h88 typedef absl::variant<Tensor, TensorShape> EagerKernelRet; typedef
133 std::vector<EagerKernelRet>* outputs,
148 std::vector<EagerKernelRet>* outputs,
210 std::vector<EagerKernelRet>* outputs,
217 std::vector<EagerKernelRet>* outputs, in RunAsync()
299 std::vector<EagerKernelRet>* outputs,
306 std::vector<EagerKernelRet>* outputs,
Dkernel_and_device.cc244 std::vector<EagerKernelRet>* outputs, in Run()
323 std::vector<EagerKernelRet>* outputs, in Run()
340 std::vector<EagerKernelRet>* outputs, in RunAsync()
Dkernel_and_device_test.cc129 std::vector<EagerKernelRet> outputs; in BM_KernelAndDeviceRun()
Dexecute.cc988 std::vector<EagerKernelRet>* outputs, int num_outputs, in GetKernelOutputs()
993 EagerKernelRet& ret = (*outputs)[i]; in GetKernelOutputs()
1026 EagerKernelRet& ret = (*outputs)[i]; in GetKernelOutputs()
1116 std::vector<EagerKernelRet> outputs(1); in EagerKernelExecute()
1321 auto outputs = std::make_shared<std::vector<EagerKernelRet>>(1); in EagerKernelExecuteAsync()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_copy_node.cc196 std::vector<EagerKernelRet> rets; in RunLocalRecv()