| /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() argument
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_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()
|
| /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 | 736 const absl::optional<ChannelHandle>& channel_id = absl::nullopt,
743 const absl::optional<ChannelHandle>& channel_id = absl::nullopt,
867 void Send(XlaOp operand, const ChannelHandle& handle);
868 XlaOp SendWithToken(XlaOp operand, XlaOp token, const ChannelHandle& handle);
871 const ChannelHandle& handle);
874 const ChannelHandle& handle);
880 XlaOp Recv(const Shape& shape, const ChannelHandle& handle);
882 const ChannelHandle& handle);
1300 const absl::optional<ChannelHandle>& channel_id,
1305 const absl::optional<ChannelHandle>& channel_id,
[all …]
|
| D | xla_builder.cc | 2717 const absl::optional<ChannelHandle>& channel_id, in AllGather()
2783 const absl::optional<ChannelHandle>& channel_id, in AllReduce()
3142 void XlaBuilder::Send(XlaOp operand, const ChannelHandle& handle) { in Send()
3158 const ChannelHandle& handle) { in SendWithToken()
3160 if (handle.type() != ChannelHandle::DEVICE_TO_DEVICE) { in SendWithToken()
3185 XlaOp XlaBuilder::Recv(const Shape& shape, const ChannelHandle& handle) { in Recv()
3211 const ChannelHandle& handle) { in RecvWithToken()
3213 if (handle.type() != ChannelHandle::DEVICE_TO_DEVICE) { in RecvWithToken()
3240 const ChannelHandle& handle) { in SendToHost()
3258 if (handle.type() != ChannelHandle::DEVICE_TO_HOST) { in SendToHost()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_compiler.cc | 569 py::enum_<ChannelHandle::ChannelType>(m, "ChannelHandle_ChannelType") in BuildXlaCompilerSubmodule()
570 .value("CHANNEL_TYPE_INVALID", ChannelHandle::CHANNEL_TYPE_INVALID) in BuildXlaCompilerSubmodule()
571 .value("DEVICE_TO_DEVICE", ChannelHandle::DEVICE_TO_DEVICE) in BuildXlaCompilerSubmodule()
572 .value("DEVICE_TO_HOST", ChannelHandle::DEVICE_TO_HOST) in BuildXlaCompilerSubmodule()
573 .value("HOST_TO_DEVICE", ChannelHandle::HOST_TO_DEVICE); in BuildXlaCompilerSubmodule()
575 py::class_<ChannelHandle>(m, "ChannelHandle") in BuildXlaCompilerSubmodule()
576 .def_property_readonly("type", &ChannelHandle::type) in BuildXlaCompilerSubmodule()
577 .def_property_readonly("handle", &ChannelHandle::handle) in BuildXlaCompilerSubmodule()
578 .def("__repr__", [](ChannelHandle* h) { return h->DebugString(); }); in BuildXlaCompilerSubmodule()
|
| D | py_client.h | 118 StatusOr<ChannelHandle> CreateChannelHandle() { in CreateChannelHandle()
121 StatusOr<ChannelHandle> CreateDeviceToHostChannelHandle() { in CreateDeviceToHostChannelHandle()
124 StatusOr<ChannelHandle> CreateHostToDeviceChannelHandle() { in CreateHostToDeviceChannelHandle()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.h | 239 Status GetChannelHandle(const string& key, xla::ChannelHandle* channel);
244 xla::ChannelHandle* channel);
249 xla::ChannelHandle* channel);
351 std::unordered_map<string, xla::ChannelHandle> channels_;
|
| D | xla_compiler.cc | 1405 xla::ChannelHandle* channel) { in GetChannelHandle()
1406 auto result = channels_.emplace(key, xla::ChannelHandle()); in GetChannelHandle()
1416 xla::ChannelHandle* channel) { in GetHostToDeviceChannelHandle()
1417 auto result = channels_.emplace(key, xla::ChannelHandle()); in GetHostToDeviceChannelHandle()
1428 xla::ChannelHandle* channel) { in GetDeviceToHostChannelHandle()
1429 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/pjrt/ |
| D | pjrt_client.h | 262 virtual StatusOr<ChannelHandle> CreateChannelHandle() = 0;
263 virtual StatusOr<ChannelHandle> CreateDeviceToHostChannelHandle() = 0;
264 virtual StatusOr<ChannelHandle> CreateHostToDeviceChannelHandle() = 0;
|
| D | pjrt_stream_executor_client.h | 204 StatusOr<ChannelHandle> CreateChannelHandle() override { in CreateChannelHandle()
207 StatusOr<ChannelHandle> CreateDeviceToHostChannelHandle() override { in CreateDeviceToHostChannelHandle()
210 StatusOr<ChannelHandle> CreateHostToDeviceChannelHandle() override { in CreateHostToDeviceChannelHandle()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops_base_structs.td | 67 def ChannelHandle : StructAttr<"ChannelHandle", HLO_Dialect, [
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | host_compute_ops.cc | 162 xla::ChannelHandle channel; in Compile()
222 xla::ChannelHandle channel; in Compile()
427 xla::ChannelHandle channel; in Compile()
479 xla::ChannelHandle channel; in Compile()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.h | 156 mlir::NamedAttribute ConvertChannelHandle(const xla::ChannelHandle& channel);
|