Home
last modified time | relevance | path

Searched refs:remote_func_params (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.cc355 if (op->remote_func_params().has_value() && in MustCompileWithXLA()
356 op->remote_func_params().value().step_id.has_value()) { in MustCompileWithXLA()
577 const absl::optional<EagerRemoteFunctionParams>& remote_func_params, in CreateUnshapedOutput() argument
584 if (remote_func_params.has_value()) { in CreateUnshapedOutput()
585 op_id = remote_func_params.value().op_id; in CreateUnshapedOutput()
618 absl::optional<EagerRemoteFunctionParams> remote_func_params = in AddOrExecuteNode() local
619 op->remote_func_params(); in AddOrExecuteNode()
620 if (kernel->IsCrossProcess() && !remote_func_params.has_value()) { in AddOrExecuteNode()
627 remote_func_params = in AddOrExecuteNode()
643 remote_func_params, &ctx, &retvals[i])); in AddOrExecuteNode()
[all …]
Dkernel_and_device.h135 const absl::optional<EagerRemoteFunctionParams>& remote_func_params,
150 const absl::optional<EagerRemoteFunctionParams>& remote_func_params,
212 const absl::optional<EagerRemoteFunctionParams>& remote_func_params,
219 const absl::optional<EagerRemoteFunctionParams>& remote_func_params, in RunAsync() argument
223 remote_func_params, {})); in RunAsync()
301 const absl::optional<EagerRemoteFunctionParams>& remote_func_params,
308 const absl::optional<EagerRemoteFunctionParams>& remote_func_params,
Dexecute_node.h96 const absl::optional<EagerRemoteFunctionParams>& remote_func_params, in ExecuteNode() argument
105 remote_func_params_(remote_func_params), in ExecuteNode()
152 const absl::optional<EagerRemoteFunctionParams>& remote_func_params, in AsyncExecuteNode() argument
161 remote_func_params_(remote_func_params), in AsyncExecuteNode()
Dkernel_and_device.cc246 const absl::optional<EagerRemoteFunctionParams>& remote_func_params, in Run() argument
325 const absl::optional<EagerRemoteFunctionParams>& remote_func_params, in Run() argument
330 remote_func_params, [&status, &n](const Status& s) { in Run()
342 const absl::optional<EagerRemoteFunctionParams>& remote_func_params, in RunAsync() argument
345 if (remote_func_params.has_value()) { in RunAsync()
346 const EagerRemoteFunctionParams& params = remote_func_params.value(); in RunAsync()
Dexecute.h52 const absl::optional<EagerRemoteFunctionParams>& remote_func_params,
Deager_operation.h144 remote_func_params = absl::nullopt);
188 const absl::optional<EagerRemoteFunctionParams>& remote_func_params() const { in remote_func_params() function
Deager_operation.cc315 const absl::optional<EagerRemoteFunctionParams> remote_func_params) { in Reset() argument
347 remote_func_params_ = remote_func_params; in Reset()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl.cc101 absl::optional<tensorflow::EagerRemoteFunctionParams> remote_func_params = in GetEagerOperationAndNumRetvals() local
105 remote_func_params = {operation.id(), operation.func_step_id()}; in GetEagerOperationAndNumRetvals()
107 remote_func_params = {operation.id(), absl::nullopt}; in GetEagerOperationAndNumRetvals()
111 eager_executor, remote_func_params)); in GetEagerOperationAndNumRetvals()