Searched refs:EagerKernelArgs (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 62 class EagerKernelArgs : public FunctionArgsInterface {
64 EagerKernelArgs() {} in EagerKernelArgs() function
66 explicit EagerKernelArgs(int count) : tensor_args_(count) {} in EagerKernelArgs() function
68 explicit EagerKernelArgs(gtl::InlinedVector<TensorValue, 4>&& tensor_args) in EagerKernelArgs() function
71 ~EagerKernelArgs() override{}; in ~EagerKernelArgs()
132 ScopedStepContainer* step_container, const EagerKernelArgs& inputs,
147 ScopedStepContainer* step_container, const EagerKernelArgs& inputs,
209 ScopedStepContainer* step_container, const EagerKernelArgs& inputs,
216 ScopedStepContainer* step_container, const EagerKernelArgs& inputs, in RunAsync()
298 ScopedStepContainer* step_container, const EagerKernelArgs& inputs,
[all …]
|
| D | kernel_and_device.cc | 53 Status EagerKernelArgs::GetLocalArg(const FunctionArgIndex& index, in GetLocalArg()
68 std::vector<Tensor> EagerKernelArgs::GetLocalTensors() const { in GetLocalTensors()
243 ScopedStepContainer* step_container, const EagerKernelArgs& inputs, in Run()
322 ScopedStepContainer* step_container, const EagerKernelArgs& inputs, in Run()
339 ScopedStepContainer* step_container, const EagerKernelArgs& inputs, in RunAsync()
|
| D | execute_node.h | 50 class ExecuteNodeArgs : public EagerKernelArgs {
52 explicit ExecuteNodeArgs(int count) : EagerKernelArgs(count) {} in ExecuteNodeArgs()
|
| D | execute_node.cc | 126 Status s = EagerKernelArgs::GetLocalArg(index, val); in GetLocalArg()
|
| D | kernel_and_device_test.cc | 140 const EagerKernelArgs args(std::move(inputs)); in BM_KernelAndDeviceRun()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_copy_node.cc | 107 EagerKernelArgs args(1); in RunLocalSend()
195 EagerKernelArgs args; in RunLocalRecv()
|
| D | eager_service_impl_test.cc | 711 class TestExecuteNodeArgs : public EagerKernelArgs {
717 : EagerKernelArgs(std::move(tensor_args)), in TestExecuteNodeArgs()
|