Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dlocal_device_state.h92 AllocationModel allocation_model, bool asynchronous,
102 AllocationModel allocation_model() const { return allocation_model_; } in allocation_model() function
Dlocal_device_state.cc32 AllocationModel allocation_model, in LocalDeviceState() argument
34 : allocation_model_(allocation_model), in LocalDeviceState()
Dpjrt_stream_executor_client.cc272 switch (local_device->allocation_model()) { in StallStreamOnError()
338 (stream_local_device->allocation_model() == in RecordUsage()
341 (stream_local_device->allocation_model() == in RecordUsage()
379 if (local_device->allocation_model() == in AllocateDestinationBuffer()
397 if (local_device->allocation_model() == in AllocateDestinationBuffer()
426 if (local_device->allocation_model() == in AllocateDestinationBuffer()
1083 if (local_device_state->allocation_model() == in Release()
1384 CHECK_EQ(dst_local_device->allocation_model(), in CopyToDevice()
1385 transfer_local_device->allocation_model()); in CopyToDevice()
1508 if (local_device->allocation_model() == in MakeTupleHelper()
[all …]