Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 23 for getIterator (0.18 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_merge_variables_with_execute.cc

      llvm::SmallDenseSet<int64_t> resource_ids;
      bool previous_unknown_resource_access = false;
      for (Operation& op : llvm::reverse(llvm::make_range(
               first_read->getIterator(), execute_parent->getIterator()))) {
        if (auto read_op = llvm::dyn_cast<TF::ReadVariableOp>(&op)) {
          VLOG(2) << "Processing read op " << debugString(op);
          auto info_it =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 17:52:11 UTC 2024
    - 27K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/tf_executor_to_functional.cc

    // order of ops in tf_executor.graph.
    LogicalResult LiftIslandOpInnerOpsFromGraph(tf_executor::GraphOp graph) {
      auto graph_position = graph.getOperation()->getIterator();
      Block* parent_block = graph.getOperation()->getBlock();
      for (Operation& op : graph.GetBody().without_terminator()) {
        auto island_op = llvm::dyn_cast<tf_executor::IslandOp>(op);
        if (!island_op)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Aug 11 20:52:36 UTC 2023
    - 3.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/materialize_mlir_passthrough_op.cc

        for (const auto &arg_mapping :
             llvm::zip(block.getArguments(), op->getOperands())) {
          std::get<0>(arg_mapping).replaceAllUsesWith(std::get<1>(arg_mapping));
        }
        op->getBlock()->getOperations().splice(op->getIterator(),
                                               block.getOperations(), block.begin(),
                                               std::prev(block.end()));
        Operation &return_op = block.front();
        for (auto ret_mapping :
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 03 12:35:38 UTC 2022
    - 4.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/launch_to_device_attribute.cc

      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();
    }
    
    void LaunchToDeviceAttributePass::runOnOperation() {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/localize_var_handles.cc

      } else if (auto write = llvm::dyn_cast<TF::AssignVariableOp>(op)) {
        resource = write.getResource();
      } else if (auto next = llvm::dyn_cast<TF::IteratorGetNextOp>(op)) {
        resource = next.getIterator();
      }
    
      if (llvm::dyn_cast_or_null<TF::VarHandleOp>(resource.getDefiningOp())) {
        return;  // We're already directly after a VarHandleOp.
      }
    
      const TF::ResourceDataflowState* state =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 06 23:53:00 UTC 2024
    - 5K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/tpu_dynamic_layout_pass.cc

        const TF::ResourceAliasAnalysis::Info& resource_alias_analysis) {
      TF::IteratorGetNextOp iterator_op = llvm::dyn_cast<TF::IteratorGetNextOp>(op);
      if (!iterator_op) return false;
    
      Value resource_iterator = iterator_op.getIterator();
    
      if (resource_alias_analysis.IsUnknownResource(resource_iterator))
        return false;
      llvm::SmallSetVector<Value, 8> aliases =
          resource_alias_analysis.GetResourceAliases(resource_iterator);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 12.7K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/executor_island_coarsening.cc

      Block* block = new_yield_op->getBlock();
    
      auto move_inner_ops = [block, new_yield_op](IslandOp island) {
        auto& island_body = island.GetBody().getOperations();
        block->getOperations().splice(new_yield_op->getIterator(), island_body,
                                      island_body.begin(),
                                      std::prev(island_body.end()));
      };
      for (IslandOp island : merged_island.islands) move_inner_ops(island);
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_island_coarsening.cc

        SmallPtrSet<Operation*, 16>& islands_set,
        SmallPtrSet<Operation*, 16>& wrapped_ops) {
      for (Operation& candidate_op : llvm::make_early_inc_range(
               llvm::make_range(op->getIterator(), op->getBlock()->end()))) {
        IslandOp candidate_island = dyn_cast<IslandOp>(candidate_op);
        if (!candidate_island || !candidate_island.WrapsSingleOp()) continue;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 27.6K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/ir/tf_executor.cc

        auto &island_body = island_op.GetBody().getOperations();
        Operation *operation = op.getOperation();
        operation->getBlock()->getOperations().splice(
            operation->getIterator(), island_body, island_body.begin(),
            std::prev(island_body.end()));
        rewriter.replaceOp(op, new_rets);
    
        return success();
      }
    };
    }  // anonymous namespace
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 42.7K bytes
    - Viewed (0)
  10. okhttp/src/test/java/okhttp3/HeadersKotlinTest.kt

    import assertk.assertions.isNull
    import java.time.Instant
    import java.util.Date
    import okhttp3.Headers.Companion.headersOf
    import org.junit.jupiter.api.Test
    
    class HeadersKotlinTest {
      @Test fun getOperator() {
        val headers = headersOf("a", "b", "c", "d")
        assertThat(headers["a"]).isEqualTo("b")
        assertThat(headers["c"]).isEqualTo("d")
        assertThat(headers["e"]).isNull()
      }
    
      @Test fun iteratorOperator() {
    Registered: Sun Jun 16 04:42:17 UTC 2024
    - Last Modified: Thu Dec 21 01:54:49 UTC 2023
    - 2K bytes
    - Viewed (0)
Back to top