Searched refs:remote_func_params (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 209 const absl::optional<EagerRemoteFunctionParams>& remote_func_params) { in Run() argument
211 remote_func_params); in Run()
219 const absl::optional<EagerRemoteFunctionParams>& remote_func_params) { in Run() argument
221 remote_func_params); in Run()
240 const absl::optional<EagerRemoteFunctionParams>& remote_func_params) { in Run() argument
309 const absl::optional<EagerRemoteFunctionParams>& remote_func_params) { in Run() argument
311 if (remote_func_params.has_value()) { in Run()
312 const EagerRemoteFunctionParams& params = remote_func_params.value(); in Run()
|
| D | kernel_and_device.h | 124 const absl::optional<EagerRemoteFunctionParams>& remote_func_params) = 0;
129 const absl::optional<EagerRemoteFunctionParams>& remote_func_params) = 0;
191 remote_func_params) override;
197 remote_func_params) override;
271 remote_func_params) override;
276 remote_func_params) override;
|
| D | execute_node.h | 76 const absl::optional<EagerRemoteFunctionParams>& remote_func_params, in ExecuteNode() argument
84 remote_func_params_(remote_func_params), in ExecuteNode()
|
| D | eager_operation.h | 52 remote_func_params = absl::nullopt);
110 const absl::optional<EagerRemoteFunctionParams>& remote_func_params() const { in remote_func_params() function
|
| D | eager_operation.cc | 26 const absl::optional<EagerRemoteFunctionParams> remote_func_params) { in Reset() argument
49 remote_func_params_ = remote_func_params; in Reset()
|
| D | execute.h | 52 const absl::optional<EagerRemoteFunctionParams>& remote_func_params,
|
| D | execute.cc | 286 if (op->remote_func_params().has_value() && in MustCompileWithXLA()
287 op->remote_func_params().value().step_id.has_value()) { in MustCompileWithXLA()
568 &ctx, op->Inputs(), op->remote_func_params(), std::move(kernel), in EagerLocalExecute()
577 ExecuteNode node(&ctx, op->Inputs(), op->remote_func_params(), in EagerLocalExecute()
946 const absl::optional<EagerRemoteFunctionParams>& remote_func_params, in EagerKernelExecute() argument
965 remote_func_params)); in EagerKernelExecute()
968 cancellation_manager, remote_func_params)); in EagerKernelExecute()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 346 absl::optional<tensorflow::EagerRemoteFunctionParams> remote_func_params = in ExecuteOp() local
350 remote_func_params = {operation.id(), operation.func_step_id()}; in ExecuteOp()
352 remote_func_params = {operation.id(), absl::nullopt}; in ExecuteOp()
357 eager_executor, remote_func_params)); in ExecuteOp()
|