Searched defs:device_map (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/grappler/ |
| D | tf_optimizer_wrapper.cc | 42 std::unordered_map<std::string, tensorflow::DeviceProperties>* device_map) { in DetectDevices()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | costmodel_test.cc | 84 std::unordered_map<string, const Graph*> device_map; in TEST() local
|
| /external/tensorflow/tensorflow/core/data/service/ |
| D | auto_shard_rewriter.cc | 128 std::unordered_map<std::string, tensorflow::DeviceProperties> device_map; in ApplyAutoShardRewrite() local
|
| /external/tensorflow/tensorflow/core/data/ |
| D | rewrite_utils.cc | 116 std::unordered_map<std::string, tensorflow::DeviceProperties> device_map; in ApplyRewrites() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding_util.cc | 218 const std::map<int64, int64>& device_map, int64* top_count) { in SelectDominantDevice() argument
247 std::map<int64, int64> device_map; in GetMostOccurringDevice() local
262 std::map<int64, int64> device_map; in GetDominantDevice() local
|
| D | hlo_sharding.cc | 285 std::map<int64, int64> device_map; in UsedDevices() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| D | lower_saved_model.cc | 127 auto &device_map = hoisted_values_by_block_device[hoist_into_block(value)]; in ReplaceHoistedValues() local
147 const auto &device_map = block_iter.second; in ReplaceHoistedValues() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_variable_runtime_reformatting.cc | 432 auto device_map = devices_attr.getValue(); in HandleReplicateOp() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | step_stats_collector.cc | 261 const std::unordered_map<string, const Graph*>& device_map) { in BuildCostModel()
|
| /external/mesa3d/src/broadcom/vulkan/ |
| D | v3dv_device.c | 1502 device_map(struct v3dv_device *device, struct v3dv_device_memory *mem) in device_map() function
|