Home
last modified time | relevance | path

Searched defs:device_map (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/grappler/
Dtf_optimizer_wrapper.cc42 std::unordered_map<std::string, tensorflow::DeviceProperties>* device_map) { in DetectDevices()
/external/tensorflow/tensorflow/core/graph/
Dcostmodel_test.cc84 std::unordered_map<string, const Graph*> device_map; in TEST() local
/external/tensorflow/tensorflow/core/kernels/data/
Drewrite_utils.cc125 std::unordered_map<string, tensorflow::DeviceProperties> device_map; in ApplyRewrites() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.cc198 const std::map<int64, int64>& device_map, int64* top_count) { in SelectDominantDevice() argument
227 std::map<int64, int64> device_map; in GetMostOccurringDevice() local
242 std::map<int64, int64> device_map; in GetDominantDevice() local
Dhlo_sharding.cc209 std::map<int64, int64> device_map; in UsedDevices() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_variable_runtime_reformatting.cc423 auto device_map = devices_attr.getValue(); in HandleReplicateOp() local
/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.cc261 const std::unordered_map<string, const Graph*>& device_map) { in BuildCostModel()
/external/mesa3d/src/broadcom/vulkan/
Dv3dv_device.c1502 device_map(struct v3dv_device *device, struct v3dv_device_memory *mem) in device_map() function