Home
last modified time | relevance | path

Searched refs:kComputeSynchronized (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dlocal_device_state.h70 kComputeSynchronized, enumerator
Dpjrt_stream_executor_client.cc282 case LocalDeviceState::kComputeSynchronized: in StallStreamOnError()
348 LocalDeviceState::kComputeSynchronized && in RecordUsage()
390 LocalDeviceState::kComputeSynchronized) { in AllocateDestinationBuffer()
408 LocalDeviceState::kComputeSynchronized) { in AllocateDestinationBuffer()
437 LocalDeviceState::kComputeSynchronized) { in AllocateDestinationBuffer()
1190 LocalDeviceState::kComputeSynchronized) { in Release()
1703 LocalDeviceState::kComputeSynchronized) { in MakeTupleHelper()
Dgpu_device.cc198 executor, xla_client, LocalDeviceState::kComputeSynchronized, in BuildLocalDeviceStates()