| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/ |
| H A D | xrt_state.h | 64 int device_ordinal_; variable
284 const int device_ordinal_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | transfer_ops.h | 81 int device_ordinal_; variable
|
| H A D | tpu_embedding_enqueue_ops.cc | 192 int device_ordinal_ = -1; member in tensorflow::__anonede597610111::EnqueueTPUEmbeddingArbitraryTensorBatchOp
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | buffer_allocations.h | 90 int device_ordinal_; variable
|
| H A D | gpu_conv_algorithm_picker.cc | 82 const int device_ordinal_; member in xla::gpu::__anon042fc3bc0111::ScratchAllocator
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | tracked_device_buffer.h | 223 int device_ordinal_; variable
|
| H A D | pjrt_stream_executor_client.h | 136 const int device_ordinal_; // -1 means not local. variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | executable_run_options.h | 160 int device_ordinal_ = -1; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | xla_launch_util.h | 194 int device_ordinal_; variable
|
| H A D | xla_device.h | 227 const int device_ordinal_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | shaped_buffer.h | 132 int device_ordinal_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | device_memory_allocator.h | 147 int device_ordinal_; // Negative one for inactive object. variable
|
| H A D | stream_executor_pimpl.h | 730 int device_ordinal_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/ |
| H A D | gpu_executor.h | 408 int device_ordinal_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_driver.cc | 76 const int device_ordinal_; member in stream_executor::gpu::GpuContext
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/profiler/backends/gpu/ |
| H A D | cupti_tracer.cc | 970 absl::optional<uint32> device_ordinal_; member in tensorflow::profiler::__anon2fb572530111::ScopedCudaContext
|