| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| H A D | lower_saved_model.cc | 121 auto &device_map = hoisted_values_by_block_device[hoist_into_block(value)]; in ReplaceHoistedValues() local
141 const auto &device_map = block_iter.second; in ReplaceHoistedValues() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/grappler/ |
| H A D | tf_optimizer_wrapper.cc | 42 std::unordered_map<std::string, tensorflow::DeviceProperties>* device_map) { in DetectDevices()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | costmodel_test.cc | 84 std::unordered_map<string, const Graph*> device_map; in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/service/ |
| H A D | auto_shard_rewriter.cc | 115 std::unordered_map<std::string, tensorflow::DeviceProperties> device_map; in ApplyAutoShardRewrite() local
|
| /aosp_15_r20/frameworks/native/vulkan/vkjson/ |
| H A D | vkjson_instance.cc | 378 std::map<VkPhysicalDevice, uint32_t> device_map; in VkJsonGetInstance() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| H A D | trt_engine_op.cc | 1062 std::unordered_map<string, tensorflow::DeviceProperties> device_map; in BuildEngine() local
1337 std::unordered_map<string, tensorflow::DeviceProperties> device_map; in AllocateCalibrationResources() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_sharding_util.cc | 255 const std::map<int64_t, int64_t>& device_map, int64_t* top_count) { in SelectDominantDevice() argument
283 std::map<int64_t, int64_t> device_map; in GetMostOccurringDevice() local
298 std::map<int64_t, int64_t> device_map; in GetDominantDevice() local
|
| H A D | hlo_sharding.cc | 374 std::map<int64_t, int64_t> device_map; in UsedDevices() local
|
| /aosp_15_r20/frameworks/native/libs/input/rust/ |
| H A D | keyboard_classifier.rs | 44 device_map: HashMap<DeviceId, KeyboardInfo>, field
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | rewrite_utils.cc | 118 std::unordered_map<std::string, tensorflow::DeviceProperties> device_map; in ApplyRewrites() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | tpu_variable_runtime_reformatting.cc | 376 auto device_map = devices_attr.getValue(); in HandleReplicateOp() local
|
| /aosp_15_r20/packages/modules/Bluetooth/system/gd/common/ |
| D | metric_id_manager_unittest.cc | 45 std::unordered_map<bluetooth::hci::Address, int> device_map; in generateAddresses() local
|
| /aosp_15_r20/packages/modules/Bluetooth/system/common/ |
| D | metric_id_allocator_unittest.cc | 48 std::unordered_map<RawAddress, int> device_map; in generateAddresses() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | step_stats_collector.cc | 261 const std::unordered_map<string, const Graph*>& device_map) { in BuildCostModel()
|
| /aosp_15_r20/tools/asuite/adevice/src/ |
| H A D | fingerprint.rs | 474 let device_map = HashMap::from([ in diff_simple_trees() localVariable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_runtime.cc | 402 absl::flat_hash_map<xla::GlobalDeviceId, int> device_map; in RunCollectiveOp() local
|
| /aosp_15_r20/external/mesa3d/src/broadcom/vulkan/ |
| H A D | v3dv_device.c | 1986 device_map(struct v3dv_device *device, struct v3dv_device_memory *mem) in device_map() function
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/ |
| H A D | rpc_test.py | 5309 def _test_device_maps_gpu(self, x_from, y_from, z_to, device_map, dst=None, fn=None): argument
5836 def _test_custom_stream(self, fn, device_map): argument
|
| /aosp_15_r20/external/bcc/libbpf-tools/arm64/ |
| H A D | vmlinux_608.h | 79073 struct cxl_device_reg_map device_map; member
|
| H A D | vmlinux.h | 79073 struct cxl_device_reg_map device_map; member
|