Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 12 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/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)
  6. 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)
  7. 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)
  8. tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc

        } else if (auto island =
                       llvm::dyn_cast<mlir::tf_executor::IslandOp>(inst)) {
          Operation& inner_op = island.GetBody().front();
          auto op_name = GetTensorFlowOpName(inner_op.getName().getStringRef());
          if (llvm::isa<FuncOp>(inner_op) && op_name.ok()) {
            // If it is TF Control dialect specific op, look up custom operation
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc

        } else if (auto island =
                       llvm::dyn_cast<mlir::tf_executor::IslandOp>(inst)) {
          Operation& inner_op = island.GetBody().front();
          auto op_name = GetTensorFlowOpName(inner_op.getName().getStringRef());
          if (llvm::isa<FuncOp>(inner_op) && op_name.ok()) {
            // If it is TF Control dialect specific op, look up custom operation
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 23:04:51 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

            }
            auto attr_value =
                mlir::DenseI32ArrayAttr::get(inner_op->getContext(), values);
            inner_op->setAttr(attr_name, attr_value);
          };
    
      if (inner_op->hasTrait<mlir::OpTrait::AttrSizedOperandSegments>() ||
          inner_op->hasTrait<mlir::OpTrait::AttrSizedResultSegments>()) {
        // The op has multiple variadic operands or results.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
Back to top