- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for OpSideEffectCollector (0.22 sec)
-
tensorflow/compiler/mlir/tensorflow/analysis/side_effect_analysis.h
// Maps group IDs to branch IDs. using ParallelIdsMap = std::map<std::string, std::string>; using OpToParallelIdsMap = absl::flat_hash_map<Operation*, ParallelIdsMap>; namespace detail { class OpSideEffectCollector; using StackResourceToOps = std::vector< absl::flat_hash_map<ResourceId, absl::flat_hash_set<Operation*>>>; // Side effect analysis info for a single function. //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/side_effect_analysis.cc
} // Class for propagating op-based side effects bottom-up and collecting them // per op, by resource ID. class OpSideEffectCollector { public: // Recursively collects op-based side effects for all ops in module and // populates `op_side_effect_map_`. explicit OpSideEffectCollector(ModuleOp module) { symbol_table_collection_.getSymbolTable(module); for (auto func : module.getOps<func::FuncOp>()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 41.2K bytes - Viewed (0)