Searched refs:unique_ids (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | clustering_ops.py | 457 unique_ids, unique_idx = array_ops.unique(cluster_idx)
458 num_unique_cluster_idx = array_ops.size(unique_ids)
461 old_counts = array_ops.gather(total_counts, unique_ids)
464 old_cluster_centers = array_ops.gather(cluster_centers, unique_ids)
494 update_counts = state_ops.scatter_add(total_counts, unique_ids,
497 cluster_centers, unique_ids, cluster_center_updates)
|
| /external/deqp-deps/SPIRV-Tools/source/fuzz/ |
| D | fuzzer_util.cpp | 452 std::unordered_map<uint32_t, opt::Instruction*> unique_ids; in IsValidAndWellFormed() local
455 &unique_ids](opt::Instruction* inst) { in IsValidAndWellFormed()
456 if (unique_ids.count(inst->unique_id()) != 0) { in IsValidAndWellFormed()
462 unique_ids.insert({inst->unique_id(), inst}); in IsValidAndWellFormed()
|
| /external/swiftshader/third_party/SPIRV-Tools/source/fuzz/ |
| D | fuzzer_util.cpp | 452 std::unordered_map<uint32_t, opt::Instruction*> unique_ids; in IsValidAndWellFormed() local
455 &unique_ids](opt::Instruction* inst) { in IsValidAndWellFormed()
456 if (unique_ids.count(inst->unique_id()) != 0) { in IsValidAndWellFormed()
462 unique_ids.insert({inst->unique_id(), inst}); in IsValidAndWellFormed()
|
| /external/angle/third_party/vulkan-deps/spirv-tools/src/source/fuzz/ |
| D | fuzzer_util.cpp | 503 std::unordered_map<uint32_t, opt::Instruction*> unique_ids; in IsValidAndWellFormed() local
506 &unique_ids](opt::Instruction* inst) { in IsValidAndWellFormed()
507 if (unique_ids.count(inst->unique_id()) != 0) { in IsValidAndWellFormed()
513 unique_ids.insert({inst->unique_id(), inst}); in IsValidAndWellFormed()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo.proto | 537 // of the module (a module may go through multiple unique_ids). If a module
|