Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 15 for inner_loop (0.2 sec)

  1. tensorflow/compiler/jit/deadness_analysis_test.cc

                                       root, "dependent_inner_iv0", "inner_loop",
                                       iv_inner.loop_cond, dependent_outer_iv0)
                                       .induction_var;
      Output dependent_inner_iv1 = CreateDependentLoopInvariantValue(
                                       root, "dependent_inner_iv1", "inner_loop",
                                       iv_inner.loop_cond, dependent_outer_iv1)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 06:59:07 UTC 2024
    - 51.6K bytes
    - Viewed (0)
  2. tensorflow/c/while_loop_test.cc

      node_name = "test_loop/body/add";
      EXPECT_TRUE(TF_GraphOperationByName(graph_, node_name) != nullptr);
      node_name = "test_loop/body/inner_loop/body/one";
      EXPECT_TRUE(TF_GraphOperationByName(graph_, node_name) != nullptr);
      node_name = "test_loop/body/inner_loop/cond/less_than";
      EXPECT_TRUE(TF_GraphOperationByName(graph_, node_name) != nullptr);
    
      // Run the graph
      Run({0, 0});
      ExpectOutputValue(0, 8);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 11 06:05:56 UTC 2024
    - 15.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/replicate_invariant_op_hoisting.cc

      for (Operation& inner_op :
           llvm::make_early_inc_range(replicate_op.GetBody())) {
        if (llvm::isa<tf_device::ReturnOp>(inner_op)) continue;
        // Skip hoisting if the inner op device attribute is a virtual device
        // defined by tf_device.replicate.
        if (DescendantUsesVirtualDevice(virtual_device_list, &inner_op)) continue;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc

               inner_op.getAttr(
                   llvm::StringRef(std::string(key_and_value_pair[1])))});
        }
      }
    
      // Set the attributes for ops with the attr_map attribute.
      for (Operation& inner_op : quantized_func.getBody().front().getOperations()) {
        if (!inner_op.hasAttr(kAttrMapAttribute)) continue;
    
        std::string attr_map_str =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 54.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/transforms/raise_custom_ops.cc

        Operation *inner_op = builder.clone(*op);
    
        new_block->addArguments(op->getOperandTypes(),
                                SmallVector<Location>(op->getNumOperands(), loc));
        for (const auto &idx_args : llvm::enumerate(new_block->getArguments())) {
          inner_op->setOperand(idx_args.index(), idx_args.value());
        }
        custom_op->setAttrs(inner_op->getAttrs());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 17 07:31:01 UTC 2023
    - 4.3K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/transforms/pin_ops_with_side_effects.cc

        // We have to recurse, since we might have wrapped a side-effectful operator
        // in a tfl::CustomTfOp.
        if (op.walk([&](Operation *inner_op) {
                return op_has_side_effects_(inner_op) ? WalkResult::interrupt()
                                                      : WalkResult::advance();
              }).wasInterrupted()) {
          ops_with_side_effects.push_back(&op);
        }
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc

                             op->getSuccessors(), new_regions);
          inner_op = rewriter.create(state);
        } else {
          bool disable_call_shape_inference = false;
          if (op->hasAttr("_disable_call_shape_inference")) {
            disable_call_shape_inference =
                op->getAttrOfType<BoolAttr>("_disable_call_shape_inference")
                    .getValue();
          }
          inner_op =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 21.2K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/graph_pruning.cc

    }
    
    // An island should be preserved if any of its inner ops should be preserved.
    bool GraphPruningPass::ShouldPreserveIsland(IslandOp island) {
      auto result = island.walk([this](Operation* inner_op) {
        if (ShouldPreserveOp(inner_op)) return WalkResult::interrupt();
        return WalkResult::advance();
      });
      return result.wasInterrupted();
    }
    
    // Prunes unreachable operations of a tf_executor.graph operation.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Oct 05 23:50:19 UTC 2022
    - 7.6K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc

      for (Operation* cluster_op : cluster_ops) {
        cluster_op->moveBefore(body, body->end());
        cluster_op->walk([&](Operation* inner_op) {
          inner_op->removeAttr(mlir::TF::kReplicationInfoAttr);
          inner_op->removeAttr(mlir::TF::kCompileDeviceTypeAttr);
    
          if (auto attr = inner_op->getAttrOfType<StringAttr>(kDeviceAttr)) {
            // Preserve device attribute if the op is placed on a replicated core
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 22:03:30 UTC 2024
    - 39.3K bytes
    - Viewed (0)
  10. src/cmd/compile/internal/ssa/loopbce.go

    //
    // Look for variables and blocks that satisfy the following
    //
    //	 loop:
    //	   ind = (Phi min nxt),
    //	   if ind < max
    //	     then goto enter_loop
    //	     else goto exit_loop
    //
    //	   enter_loop:
    //		do something
    //	      nxt = inc + ind
    //		goto loop
    //
    //	 exit_loop:
    func findIndVar(f *Func) []indVar {
    	var iv []indVar
    	sdom := f.Sdom()
    
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Tue Nov 07 17:37:47 UTC 2023
    - 11.8K bytes
    - Viewed (0)
Back to top