/aosp_15_r20/external/mesa3d/src/gfxstream/guest/vulkan_enc/ |
H A D | VkEncoder.cpp | 855 VkDevice local_device; in vkGetDeviceProcAddr() local 1021 VkDevice local_device; in vkDestroyDevice() local 1619 VkDevice local_device; in vkGetDeviceQueue() local 1811 VkDevice local_device; in vkDeviceWaitIdle() local 1858 VkDevice local_device; in vkAllocateMemory() local 1963 VkDevice local_device; in vkFreeMemory() local 2062 VkDevice local_device; in vkFlushMappedMemoryRanges() local 2167 VkDevice local_device; in vkInvalidateMappedMemoryRanges() local 2269 VkDevice local_device; in vkGetDeviceMemoryCommitment() local 2328 VkDevice local_device; in vkBindBufferMemory() local [all …]
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
H A D | pjrt_stream_executor_client.cc | 274 void StallStreamOnError(LocalDeviceState* local_device, se::Stream* stream) { in StallStreamOnError() 375 LocalDeviceState* local_device, se::Stream* copy_stream, in AllocateDestinationBuffer() 490 LocalDeviceState* local_device, in AddDestinationBufferSynchronization() 595 auto* local_device = device_->local_device_state(); in logical_on_device_shape() local 1319 LocalDeviceState* local_device = device_->local_device_state(); in ToLiteral() local 1675 PjRtStreamExecutorClient* client, LocalDeviceState* local_device, in MakeTupleHelper() 1747 PjRtDevice* device, LocalDeviceState* local_device, in OutputBufferHelper()
|
H A D | gpu_device.cc | 366 std::unique_ptr<LocalDeviceState> local_device; in BuildDistributedDevices() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
H A D | remote_device.cc | 67 Device* local_device; in AsRemoteDevices() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/ |
H A D | master.proto | 275 repeated DeviceAttributes local_device = 1; field
|
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
H A D | eager_service_impl_test.cc | 949 Device* local_device; in TEST_F() local 1003 Device* local_device; in TEST_F() local
|
/aosp_15_r20/external/pytorch/torch/distributed/_shard/sharded_tensor/ |
H A D | api.py | 1287 *size, local_device, tensor_properties: TensorProperties argument
|
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/ |
H A D | rpc_test.py | 5984 def _test_rref_synchronization(self, local_device, remote_device): argument 6030 local_device, argument 6095 def _test_rref_forward_synchronization(self, local_device, remote_device): argument 6163 def _test_owner_rref_forward_synchronization(self, local_device, remote_device): argument
|
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
H A D | op_kernel_test.cc | 227 const string& local_device = "/job:localhost/replica:0/task:0/device:CPU:0"; in TEST_F() local
|
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/ |
D | master.pb.h | 5044 inline const ::tensorflow::DeviceAttributes& ListDevicesResponse::local_device(int index) const { in local_device() function 5057 ListDevicesResponse::local_device() const { in local_device() function
|
/aosp_15_r20/external/pytorch/test/distributed/ |
H A D | test_c10d_nccl.py | 3547 def local_device(self): member in NCCLTraceTestBase
|
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
H A D | placer_test.cc | 1602 const string local_device = "/job:b/replica:0/task:0/device:FakeCPU:0"; in TEST_F() local
|
/aosp_15_r20/external/vulkan-validation-layers/tests/ |
H A D | vklayertests_others.cpp | 1449 VkDevice local_device; in TEST_F() local
|