| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | channel_tracker.cc | 34 StatusOr<ChannelHandle> ChannelTracker::NewChannel( in NewChannel()
35 ChannelHandle::ChannelType type) { in NewChannel()
36 if (type != ChannelHandle::DEVICE_TO_DEVICE && in NewChannel()
37 type != ChannelHandle::HOST_TO_DEVICE && in NewChannel()
38 type != ChannelHandle::DEVICE_TO_HOST) { in NewChannel()
44 ChannelHandle new_handle = AllocateHandle(type); in NewChannel()
56 Status ChannelTracker::RegisterSend(const ChannelHandle& handle) { in RegisterSend()
61 Status ChannelTracker::RegisterRecv(const ChannelHandle& handle) { in RegisterRecv()
66 ChannelHandle ChannelTracker::AllocateHandle(ChannelHandle::ChannelType type) { in AllocateHandle()
68 ChannelHandle result; in AllocateHandle()
[all …]
|
| D | channel_tracker.h | 52 ChannelHandle::ChannelType type;
57 StatusOr<ChannelHandle> NewChannel(ChannelHandle::ChannelType type);
61 Status RegisterSend(const ChannelHandle& handle);
65 Status RegisterRecv(const ChannelHandle& handle);
70 ChannelHandle AllocateHandle(ChannelHandle::ChannelType type)
73 Status RegisterSendInternal(const ChannelHandle& handle)
76 Status RegisterRecvInternal(const ChannelHandle& handle)
|
| /external/libchrome/ipc/ |
| D | ipc_channel_handle.h | 22 struct ChannelHandle { struct
23 ChannelHandle() {} in ChannelHandle() function
24 explicit ChannelHandle(const base::FileDescriptor& s) : socket(s) {} in ChannelHandle() function
29 struct ChannelHandle { argument
30 ChannelHandle() {}
31 ChannelHandle(mojo::MessagePipeHandle h) : mojo_handle(h) {}
|
| D | ipc_channel_factory.cc | 17 ChannelHandle handle, in PlatformChannelFactory()
38 ChannelHandle handle_;
49 const ChannelHandle& handle, in Create()
|
| D | ipc_channel.h | 175 const IPC::ChannelHandle& channel_handle, in COMPONENT_EXPORT()
180 const IPC::ChannelHandle& channel_handle, in COMPONENT_EXPORT()
185 const IPC::ChannelHandle& channel_handle, in COMPONENT_EXPORT()
|
| D | ipc_channel_common.cc | 32 const IPC::ChannelHandle& channel_handle, in CreateClient()
48 const IPC::ChannelHandle& channel_handle, in CreateServer()
|
| D | ipc_channel_nacl.h | 40 ChannelNacl(const IPC::ChannelHandle& channel_handle,
57 bool CreatePipe(const IPC::ChannelHandle& channel_handle);
|
| D | ipc_channel_nacl.cc | 130 ChannelNacl::ChannelNacl(const IPC::ChannelHandle& channel_handle, in ChannelNacl()
250 const IPC::ChannelHandle& channel_handle) { in CreatePipe()
390 const IPC::ChannelHandle& channel_handle, in Create()
|
| D | ipc_channel_factory.h | 27 const ChannelHandle& handle, in COMPONENT_EXPORT()
|
| D | ipc_channel_proxy.h | 96 const IPC::ChannelHandle& channel_handle, in COMPONENT_EXPORT()
120 void Init(const IPC::ChannelHandle& channel_handle, in COMPONENT_EXPORT()
|
| D | ipc_message_utils_unittest.cc | 102 IPC::ChannelHandle channel_handle(message_pipe.handle0.release()); in TEST()
108 IPC::ChannelHandle result_handle; in TEST()
|
| D | ipc_sync_channel.h | 83 const IPC::ChannelHandle& channel_handle, in COMPONENT_EXPORT()
|
| D | ipc_channel_proxy.cc | 395 const IPC::ChannelHandle& channel_handle, in Create()
442 void ChannelProxy::Init(const IPC::ChannelHandle& channel_handle, in Init()
|
| D | ipc_message_utils.h | 60 struct ChannelHandle;
980 struct COMPONENT_EXPORT(IPC) ParamTraits<IPC::ChannelHandle> {
981 typedef ChannelHandle param_type;
|
| D | ipc_message_utils.cc | 1289 void ParamTraits<IPC::ChannelHandle>::Write(base::Pickle* m, in Write()
1298 bool ParamTraits<IPC::ChannelHandle>::Read(const base::Pickle* m, in Read()
1308 void ParamTraits<IPC::ChannelHandle>::Log(const param_type& p, in Log()
|
| D | ipc_sync_channel.cc | 535 const IPC::ChannelHandle& channel_handle, in Create()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | client.h | 218 StatusOr<ChannelHandle> CreateChannelHandle();
222 StatusOr<ChannelHandle> CreateHostToDeviceChannelHandle();
223 StatusOr<ChannelHandle> CreateDeviceToHostChannelHandle();
235 StatusOr<ChannelHandle> CreateChannelHandleByType(
236 ChannelHandle::ChannelType type);
|
| D | client.cc | 487 StatusOr<ChannelHandle> Client::CreateChannelHandleByType( in CreateChannelHandleByType()
488 ChannelHandle::ChannelType type) { in CreateChannelHandleByType()
504 StatusOr<ChannelHandle> Client::CreateChannelHandle() { in CreateChannelHandle()
505 return CreateChannelHandleByType(ChannelHandle::DEVICE_TO_DEVICE); in CreateChannelHandle()
508 StatusOr<ChannelHandle> Client::CreateHostToDeviceChannelHandle() { in CreateHostToDeviceChannelHandle()
509 return CreateChannelHandleByType(ChannelHandle::HOST_TO_DEVICE); in CreateHostToDeviceChannelHandle()
512 StatusOr<ChannelHandle> Client::CreateDeviceToHostChannelHandle() { in CreateDeviceToHostChannelHandle()
513 return CreateChannelHandleByType(ChannelHandle::DEVICE_TO_HOST); in CreateDeviceToHostChannelHandle()
|
| D | xla_builder.h | 472 const absl::optional<ChannelHandle>& channel_id = absl::nullopt);
551 void Send(const XlaOp& operand, const ChannelHandle& handle);
553 const ChannelHandle& handle);
556 const Shape& shape_with_layout, const ChannelHandle& handle);
559 const ChannelHandle& handle);
565 XlaOp Recv(const Shape& shape, const ChannelHandle& handle);
567 const ChannelHandle& handle);
889 const absl::optional<ChannelHandle>& channel_id);
975 friend void Send(const XlaOp& operand, const ChannelHandle& handle);
977 const ChannelHandle& handle);
[all …]
|
| D | xla_builder.cc | 2137 const absl::optional<ChannelHandle>& channel_id) { in CrossReplicaSum()
2310 void XlaBuilder::Send(const XlaOp& operand, const ChannelHandle& handle) { in Send()
2326 const ChannelHandle& handle) { in SendWithToken()
2328 if (handle.type() != ChannelHandle::DEVICE_TO_DEVICE) { in SendWithToken()
2353 XlaOp XlaBuilder::Recv(const Shape& shape, const ChannelHandle& handle) { in Recv()
2379 const ChannelHandle& handle) { in RecvWithToken()
2381 if (handle.type() != ChannelHandle::DEVICE_TO_DEVICE) { in RecvWithToken()
2408 const ChannelHandle& handle) { in SendToHost()
2426 if (handle.type() != ChannelHandle::DEVICE_TO_HOST) { in SendToHost()
2454 const ChannelHandle& handle) { in RecvFromHost()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.h | 367 Status GetChannelHandle(const string& key, xla::ChannelHandle* channel);
372 xla::ChannelHandle* channel);
377 xla::ChannelHandle* channel);
478 std::unordered_map<string, xla::ChannelHandle> channels_;
|
| D | xla_compiler.cc | 1179 xla::ChannelHandle* channel) { in GetChannelHandle()
1180 auto result = channels_.emplace(key, xla::ChannelHandle()); in GetChannelHandle()
1190 xla::ChannelHandle* channel) { in GetHostToDeviceChannelHandle()
1191 auto result = channels_.emplace(key, xla::ChannelHandle()); in GetHostToDeviceChannelHandle()
1202 xla::ChannelHandle* channel) { in GetDeviceToHostChannelHandle()
1203 auto result = channels_.emplace(key, xla::ChannelHandle()); in GetDeviceToHostChannelHandle()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | sendrecv_ops.cc | 46 xla::ChannelHandle channel; in Compile()
77 xla::ChannelHandle channel; in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla.proto | 374 ChannelHandle.ChannelType channel_type = 1;
378 ChannelHandle channel = 1;
|
| D | xla_data.proto | 319 message ChannelHandle { message
|