Searched refs:EagerRemoteFunctionParams (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/core/common_runtime/eager/ |
D | kernel_and_device.h | 56 struct EagerRemoteFunctionParams { struct 135 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() 301 const absl::optional<EagerRemoteFunctionParams>& remote_func_params, 308 const absl::optional<EagerRemoteFunctionParams>& remote_func_params,
|
D | execute_node.h | 96 const absl::optional<EagerRemoteFunctionParams>& remote_func_params, in ExecuteNode() 140 const absl::optional<EagerRemoteFunctionParams>& remote_func_params_; 152 const absl::optional<EagerRemoteFunctionParams>& remote_func_params, in AsyncExecuteNode() 235 const absl::optional<EagerRemoteFunctionParams> remote_func_params_;
|
D | eager_operation.h | 143 const absl::optional<EagerRemoteFunctionParams> 188 const absl::optional<EagerRemoteFunctionParams>& remote_func_params() const { in remote_func_params() 255 absl::optional<EagerRemoteFunctionParams> remote_func_params_;
|
D | kernel_and_device.cc | 246 const absl::optional<EagerRemoteFunctionParams>& remote_func_params, in Run() 325 const absl::optional<EagerRemoteFunctionParams>& remote_func_params, in Run() 342 const absl::optional<EagerRemoteFunctionParams>& remote_func_params, in RunAsync() 346 const EagerRemoteFunctionParams& params = remote_func_params.value(); in RunAsync()
|
D | execute.h | 52 const absl::optional<EagerRemoteFunctionParams>& remote_func_params,
|
D | execute.cc | 577 const absl::optional<EagerRemoteFunctionParams>& remote_func_params, in CreateUnshapedOutput() 618 absl::optional<EagerRemoteFunctionParams> remote_func_params = in AddOrExecuteNode() 628 EagerRemoteFunctionParams{op_id, /*step_id=*/absl::nullopt}; in AddOrExecuteNode() 990 const absl::optional<EagerRemoteFunctionParams>& remote_func_params) { in GetKernelOutputs() 1109 const absl::optional<EagerRemoteFunctionParams>& remote_func_params, in EagerKernelExecute() 1316 const absl::optional<EagerRemoteFunctionParams>& remote_func_params, in EagerKernelExecuteAsync()
|
D | eager_operation.cc | 315 const absl::optional<EagerRemoteFunctionParams> remote_func_params) { in Reset()
|
/external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
D | eager_service_impl.cc | 101 absl::optional<tensorflow::EagerRemoteFunctionParams> remote_func_params = in GetEagerOperationAndNumRetvals()
|