Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 34 for GetBlock (0.13 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_program_key.cc

    };
    
    // Checks if `op` is nested in `block`.
    bool OpInBlock(Operation* op, Block* block) {
      Block* op_block = op->getBlock();
      while (op_block) {
        if (op_block == block) return true;
        if (auto* parent_op = op_block->getParentOp()) {
          op_block = parent_op->getBlock();
        } else {
          break;
        }
      }
      return false;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/cluster_ops_by_policy.cc

        assert(op && "member operation must be not null");
    
        for (Operation *user : op->getUsers()) {
          // Skip users in other blocks.
          if (user->getBlock() != op->getBlock()) continue;
    
          // Skip users is in the `dst_root` or `src_root` clusters, if we'll merge
          // roots they'll become a single cluster and will not violate the
          // dominance property after that.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 27.9K bytes
    - Viewed (0)
  3. platforms/core-execution/persistent-cache/src/main/java/org/gradle/cache/internal/btree/FreeListBlockStore.java

            store.close();
        }
    
        @Override
        public void clear() {
            store.clear();
        }
    
        @Override
        public void remove(BlockPayload block) {
            Block container = block.getBlock();
            store.remove(block);
            freeListBlock.add(container.getPos(), container.getSize());
        }
    
        @Override
        public <T extends BlockPayload> T readFirst(Class<T> payloadType) {
    Registered: Wed Jun 12 18:38:38 UTC 2024
    - Last Modified: Fri Apr 26 16:02:32 UTC 2024
    - 8.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_island_coarsening.cc

    LogicalResult SortTopologically(Block::iterator begin, Block::iterator end) {
      Block* block = begin->getBlock();
      // Either sort from `begin` to end of block or both `begin` and
      // `end` should belong to the same block.
      assert(end == block->end() ||
             end->getBlock() == block && "ops must be in the same block");
    
      // Track the ops that still need to be scheduled in a set.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 27.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/layout_optimization.cc

        if (args_permutation.empty() || res_permutation.empty()) return;
    
        mlir::Operation* op = layout_sensitive_interface.getOperation();
        Location loc = op->getLoc();
        OpBuilder builder = OpBuilder::atBlockEnd(op->getBlock());
    
        auto perm_attr = [&](Permutation permutation) -> DenseIntElementsAttr {
          auto perm_ty = RankedTensorType::get({4}, builder.getIntegerType(64));
          return DenseIntElementsAttr::get(perm_ty, permutation);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 19.3K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/transforms/reduce_while_operands.cc

          if (mlir::dyn_cast_or_null<BlockArgument>(operand)) continue;
          auto operand_op = operand.getDefiningOp();
          if (IsConstant(operand_op)) continue;
          if (operand_op->getBlock() != op->getBlock()) {
            return WalkResult::interrupt();
          }
        }
        return WalkResult::advance();
      });
      return !walk_result.wasInterrupted();
    }
    
    // It reduces the following pattern:
    //
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.4K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/launch_to_device_attribute.cc

        return failure();
    
      // Move all inner ops of the launch to the block containing the launch.
      auto body = launch.GetBody().without_terminator();
      Operation* launch_op = launch.getOperation();
      launch_op->getBlock()->getOperations().splice(
          launch_op->getIterator(), launch.GetBody().getOperations(), body.begin(),
          body.end());
    
      launch.erase();
    
      return success();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.5K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/lower_variable_ops_to_ml_program.cc

        if (!initial_value) {
          initial_value = builder.getArrayAttr({});
        }
      }
    
      if (!type) return nullptr;
    
      auto globalOp = builder.create<ml_program::GlobalOp>(
          builder.getBlock()->getParentOp()->getLoc(), name, type, false,
          initial_value, nullptr);
      symbol_table.insert(globalOp);
    
      return globalOp;
    }
    
    }  // namespace
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 06 23:53:00 UTC 2024
    - 7.6K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tf2xla/internal/passes/extract_head_tail_outside_compilation.cc

    }
    
    // Checks if `op` is nested in `block`.
    bool OpInBlock(Operation* op, Block* block) {
      Block* op_block = op->getBlock();
      while (op_block) {
        if (op_block == block) return true;
        if (auto* parent_op = op_block->getParentOp()) {
          op_block = parent_op->getBlock();
        } else {
          break;
        }
      }
      return false;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 20.6K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_outline_tpu_island.cc

                    SymbolTable &to) {
      if (to.lookup<func::FuncOp>(symbol_ref.getValue())) return;
      func::FuncOp callee = from.lookup<func::FuncOp>(symbol_ref.getValue());
      callee.getOperation()->getBlock()->getOperations().remove(
          callee.getOperation());
      to.insert(callee);
    }
    
    void TPUBridgeExecutorIslandOutlining::runOnOperation() {
      MLIRContext *ctx = &getContext();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.7K bytes
    - Viewed (0)
Back to top