Home
last modified time | relevance | path

Searched defs:local_device (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/mesa3d/src/gfxstream/guest/vulkan_enc/
H A DVkEncoder.cpp855 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 Dpjrt_stream_executor_client.cc274 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 Dgpu_device.cc366 std::unique_ptr<LocalDeviceState> local_device; in BuildDistributedDevices() local
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dremote_device.cc67 Device* local_device; in AsRemoteDevices() local
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dmaster.proto275 repeated DeviceAttributes local_device = 1; field
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/
H A Deager_service_impl_test.cc949 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 Dapi.py1287 *size, local_device, tensor_properties: TensorProperties argument
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/
H A Drpc_test.py5984 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 Dop_kernel_test.cc227 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/
Dmaster.pb.h5044 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 Dtest_c10d_nccl.py3547 def local_device(self): member in NCCLTraceTestBase
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dplacer_test.cc1602 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 Dvklayertests_others.cpp1449 VkDevice local_device; in TEST_F() local