Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 55 for getUses (0.14 sec)

  1. tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc

          // with corresponding results of new parallel_execute op
          for (auto& use : llvm::make_early_inc_range(
                   old_parallel_execute->getResult(output_index).getUses())) {
            use.set(new_parallel_execute->getResult(output_index));
          }
          continue;
        }
    
        int tpu_cluster_output_index = output_index - num_results_pre_cluster;
        const auto& output_sharding =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 22 21:28:13 UTC 2024
    - 34K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_sharding_identification_pass.cc

        llvm::SmallVector<Value, 4> next_values_to_visit;
        for (Value value_to_visit : values_to_visit) {
          if (!visited_values.insert(value_to_visit).second) continue;
    
          for (auto& use : value_to_visit.getUses()) {
            Operation* owner = use.getOwner();
            if (auto sharding = llvm::dyn_cast<mlir::TF::XlaShardingOp>(owner))
              return sharding.get_XlaSharding();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 02:01:13 UTC 2024
    - 28.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_communication.cc

            // Only `mlir::func::CallOp` is supported as this requires knowing how
            // to rewrite arguments and results to a function.
            if (!isa<mlir::func::CallOp>(use.getUser())) continue;
            auto caller_parent_func =
                use.getUser()->getParentOfType<func::FuncOp>();
            if (!caller_parent_func) continue;
    
            FuncToRewrite func_to_rewrite{/*original=*/caller_parent_func,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 40.5K bytes
    - Viewed (0)
  4. tests/transaction_test.go

    		t.Fatalf("Should find saved record")
    	}
    }
    
    func TestNestedTransactionWithBlock(t *testing.T) {
    	var (
    		user  = *GetUser("transaction-nested", Config{})
    		user1 = *GetUser("transaction-nested-1", Config{})
    		user2 = *GetUser("transaction-nested-2", Config{})
    	)
    
    	if err := DB.Transaction(func(tx *gorm.DB) error {
    		tx.Create(&user)
    
    Registered: Wed Jun 12 16:27:09 UTC 2024
    - Last Modified: Wed May 08 04:07:58 UTC 2024
    - 10.9K bytes
    - Viewed (0)
  5. tests/associations_has_many_test.go

    	}
    
    	AssertAssociationCount(t, user2, "Team", 0, "after clear")
    }
    
    func TestHasManyAssociationForSlice(t *testing.T) {
    	users := []User{
    		*GetUser("slice-hasmany-1", Config{Pets: 2}),
    		*GetUser("slice-hasmany-2", Config{Pets: 0}),
    		*GetUser("slice-hasmany-3", Config{Pets: 4}),
    	}
    
    	DB.Create(&users)
    
    	// Count
    	AssertAssociationCount(t, users, "Pets", 6, "")
    
    	// Find
    	var pets []Pet
    Registered: Wed Jun 12 16:27:09 UTC 2024
    - Last Modified: Wed Jun 12 10:49:45 UTC 2024
    - 16K bytes
    - Viewed (0)
  6. tests/preload_test.go

    	DB.Preload("Pets.Toy").Find(&user4, "id = ?", user.ID)
    	CheckUser(t, *user4, user)
    }
    
    func TestNestedPreloadForSlice(t *testing.T) {
    	users := []User{
    		*GetUser("slice_nested_preload_1", Config{Pets: 2}),
    		*GetUser("slice_nested_preload_2", Config{Pets: 0}),
    		*GetUser("slice_nested_preload_3", Config{Pets: 3}),
    	}
    
    	for _, user := range users {
    		for idx, pet := range user.Pets {
    Registered: Wed Jun 12 16:27:09 UTC 2024
    - Last Modified: Wed Jun 12 10:00:47 UTC 2024
    - 15.9K bytes
    - Viewed (0)
  7. tests/joins_test.go

    	}
    
    	CheckUser(t, user2, user)
    }
    
    func TestJoinsForSlice(t *testing.T) {
    	users := []User{
    		*GetUser("slice-joins-1", Config{Company: true, Manager: true, Account: true}),
    		*GetUser("slice-joins-2", Config{Company: true, Manager: true, Account: true}),
    		*GetUser("slice-joins-3", Config{Company: true, Manager: true, Account: true}),
    	}
    
    	DB.Create(&users)
    
    	var userIDs []uint
    Registered: Wed Jun 12 16:27:09 UTC 2024
    - Last Modified: Wed Jun 12 10:52:33 UTC 2024
    - 15K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/layout_optimization.cc

      // Bypass Transpose nodes for all results.
      for (OpResult result : op->getResults()) {
        result.setType(
            cast<TransposeOp>(*result.getUsers().begin()).getY().getType());
        for (Operation* transpose : result.getUsers()) {
          transpose->getResult(0).replaceAllUsesWith(result);
        }
      }
    
      // Maybe add a Transpose node for all operands (or reuse existing transposes).
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 19.3K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_types.cc

    // Check if an op result value is consumed by qint -> int TF Cast OP.
    bool IsQintValueQintToIntCast(Value v) {
      if (!IsIllegalType(v.getType())) {
        return true;
      }
      if (v.getUsers().empty()) {
        return false;
      }
      return llvm::all_of(v.getUsers(), [&](OpOperand operand) {
        return IsQintToIntCast(operand.getOwner());
      });
    }
    
    // Check if an op operand value is defined by int -> qint TF Cast OP.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 12.6K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h

          min -= kNearZeroTolerance;
          max += kNearZeroTolerance;
        }
      }
    };
    
    template <typename VerifierT>
    bool UsedBy(Operation* op) {
      for (Operation* user : op->getUsers()) {
        if (llvm::isa_and_nonnull<VerifierT>(user)) return true;
      }
      return false;
    }
    
    template <typename VerifierT>
    void CreateVerifier(Operation* quantizing_op, Operation* quantized_op,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 20:30:06 UTC 2024
    - 41.7K bytes
    - Viewed (0)
Back to top