Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dlocal_device_state.h92 AllocationModel allocation_model,
103 AllocationModel allocation_model() const { return allocation_model_; } in allocation_model() function
H A Dpjrt_stream_executor_client.cc275 switch (local_device->allocation_model()) { in StallStreamOnError()
344 (stream_local_device->allocation_model() == in RecordUsage()
347 (stream_local_device->allocation_model() == in RecordUsage()
389 if (local_device->allocation_model() == in AllocateDestinationBuffer()
407 if (local_device->allocation_model() == in AllocateDestinationBuffer()
436 if (local_device->allocation_model() == in AllocateDestinationBuffer()
1189 if (local_device_state->allocation_model() == in Release()
1511 CHECK_EQ(dst_local_device->allocation_model(), in CopyToDevice()
1512 transfer_local_device->allocation_model()); in CopyToDevice()
1702 if (local_device->allocation_model() == in MakeTupleHelper()
[all …]
H A Dlocal_device_state.cc31 AllocationModel allocation_model, in LocalDeviceState() argument
35 : allocation_model_(allocation_model), in LocalDeviceState()