| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/ |
| D | direct_tpu_driver.cc | 196 : driver_(driver), driver_fn_(driver_fn) {} in DirectTpuLinearizer()
202 driver_fn_->TpuDriver_ComputeLinearizedBytesFromShape(driver_, shape_); in ComputeLinearizedBytesFromShape()
212 driver_fn_->TpuDriver_LinearizeShape(driver_, dst, src, shape_); in LinearizeShape()
224 driver_fn_->TpuDriver_DelinearizeShape(driver_, dst, src, shape_); in DelinearizeShape()
232 ::TpuDriver* driver_;
250 driver_ = driver_fn_.TpuDriver_Open("local://"); in DirectTpuDriver()
256 driver_ = driver_fn_.TpuDriver_Open("local://"); in DirectTpuDriver()
260 ~DirectTpuDriver() override { driver_fn_.TpuDriver_Close(driver_); } in ~DirectTpuDriver()
263 ::TpuSystemInfo* info = driver_fn_.TpuDriver_QuerySystemInfo(driver_); in QuerySystemInfo()
269 auto tpu_status = driver_fn_.TpuDriver_Reset(driver_); in Reset()
[all …]
|
| D | recording_tpu_driver.cc | 132 : driver_(std::move(driver)), in RecordingTpuDriver()
152 driver_->QuerySystemInfo(system_info); in QuerySystemInfo()
155 Status Reset() override { return driver_->Reset(); } in Reset()
164 driver_->Allocate(core_id, region, num_bytes, unwrapped_wait_for); in Allocate()
187 auto handle = driver_->Allocate(core_id, region, shape, unwrapped_wait_for); in Allocate()
221 auto handle = driver_->AllocateTuple(core_id, region, unwrapped_children, in AllocateTuple()
250 auto event = driver_->Deallocate(std::move(recording_handle->handle_), in Deallocate()
274 std::make_shared<RecordingEvent>(driver_->TransferToDevice( in TransferToDevice()
302 std::make_shared<RecordingEvent>(driver_->TransferFromDevice( in TransferFromDevice()
325 std::make_shared<RecordingEvent>(driver_->TransferFromDeviceToDevice( in TransferFromDeviceToDevice()
[all …]
|
| D | pod_tpu_driver.cc | 49 : driver_(driver), operation_id_(operation_id) {} in PodEvent()
60 PodTpuDriver* driver_; member in tpu_driver::__anon6b014a980111::PodEvent
166 : driver_(driver), in PodBufferHandle()
170 event_(std::make_shared<PodEvent>(driver_, operation_id_)), in PodBufferHandle()
181 PodTpuDriver* driver_; member in tpu_driver::__anon6b014a980111::PodBufferHandle
192 : driver_(driver), in PodCompiledProgramHandle()
194 event_(std::make_shared<PodEvent>(driver_, operation_id_)) {} in PodCompiledProgramHandle()
203 PodTpuDriver* driver_; member in tpu_driver::__anon6b014a980111::PodCompiledProgramHandle
212 : driver_(driver), in PodLoadedProgramHandle()
215 event_(std::make_shared<PodEvent>(driver_, operation_id_)) {} in PodLoadedProgramHandle()
[all …]
|
| D | grpc_tpu_driver.cc | 276 GrpcTpuDriver* driver_; member in tpu_driver::__anon4073f9c10111::GrpcTpuStream
461 driver_(driver), in GrpcTpuStream()
496 auto operation_id = driver_->NewOperationId(); in InitializeRequest()
869 if (grpc_handle->id().client_id != driver_->client_id()) { in LoadProgram()
906 if (program_handle->id().client_id != driver_->client_id()) { in ExecuteProgram()
917 if (grpc_handle->id().client_id != driver_->client_id()) { in ExecuteProgram()
927 if (grpc_handle->id().client_id != driver_->client_id()) { in ExecuteProgram()
|
| /external/harfbuzz_ng/src/ |
| D | hb-kern.hh | 41 hb_kern_machine_t (const Driver &driver_, in hb_kern_machine_t()
43 driver (driver_), in hb_kern_machine_t()
|
| /external/libchrome/ipc/ |
| D | ipc_channel_mojo_unittest.cc | 726 &driver_); in OnChannelConnected()
732 driver_->ExpectValue(i); in OnChannelConnected()
735 driver_->RequestQuit(base::BindOnce( in OnChannelConnected()
745 IPC::mojom::SimpleTestDriverAssociatedPtr driver_; member in __anon81a6c5020111::ListenerSendingAssociatedMessages
1212 void set_driver(IPC::mojom::SimpleTestDriver* driver) { driver_ = driver; } in set_driver()
1233 DCHECK(driver_); in RequestValue()
1234 EXPECT_TRUE(driver_->RequestValue(&response)); in RequestValue()
1246 DCHECK(driver_); in OnMessageReceived()
1247 EXPECT_TRUE(driver_->RequestValue(&response)); in OnMessageReceived()
1271 IPC::mojom::SimpleTestDriver* driver_ = nullptr; member in __anon81a6c5020111::SimpleTestClientImpl
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.h | 129 tpu_driver::TpuDriver* driver() { return driver_.get(); } in driver()
135 std::unique_ptr<tpu_driver::TpuDriver> driver_; variable
|
| D | tpu_client.cc | 94 driver_(std::move(driver)), in PyTpuClient()
|
| /external/wpa_supplicant_8/wpa_supplicant/ |
| D | todo.txt | 13 driver_*.c to do the mapping for drivers that cannot handle non-zero keyidx
|
| /external/wpa_supplicant_8/hostapd/ |
| D | ChangeLog | 806 MLME (driver_{hostap,nl80211,test}.c)
|