Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for OpSideEffectCollector (0.14 sec)

  1. 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)
  2. 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)
Back to top