Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 12 for cur_idx (0.39 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc

      for (int32_t cur_idx = 0; cur_idx < call_op.getArgs().size(); cur_idx++) {
        // Check if the only the weight index has QuantizeCastOp.
        auto cur_op = dyn_cast_or_null<quantfork::QuantizeCastOp>(
            call_op.getArgs()[cur_idx].getDefiningOp());
        if (!cur_op && spec->quantizable_operands.contains(cur_idx)) {
          return false;
        } else if (cur_op) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 54.5K bytes
    - Viewed (0)
  2. callbacks.go

    		if c.before != "" { // if defined before callback
    			if c.before == "*" && len(sorted) > 0 {
    				if curIdx := getRIndex(sorted, c.name); curIdx == -1 {
    					sorted = append([]string{c.name}, sorted...)
    				}
    			} else if sortedIdx := getRIndex(sorted, c.before); sortedIdx != -1 {
    				if curIdx := getRIndex(sorted, c.name); curIdx == -1 {
    					// if before callback already sorted, append current callback just after it
    Registered: Wed Jun 12 16:27:09 UTC 2024
    - Last Modified: Tue Mar 26 03:33:36 UTC 2024
    - 8.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/ir/tf_arith_ops_folder.cc

      DenseIntElementsAttr dims_attr;
      if (!matchPattern(dims, m_Constant(&dims_attr))) return success();
      for (const auto &dim_pair : llvm::enumerate(dims_attr)) {
        int64_t cur_dim = dim_pair.value().getSExtValue();
        if (cur_dim < -rank || cur_dim >= rank)
          return emitError(loc)
                 << dim_pair.index() << "-th dimension should be in the range of [-"
                 << rank << ", " << rank << ")";
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/common/lift_as_function_call.cc

          rhs_batch_dim_size++;
        } else if (absl::StrContains(out_eq, c)) {
          int out_idx = out_eq.find(c);
          if (out_idx < rhs_out_idx_end) {
            return false;
          }
          if (out_idx < rhs_out_idx_start) rhs_out_idx_start = out_idx;
          if (out_idx > rhs_out_idx_end) rhs_out_idx_end = out_idx;
        }
      }
    
      if (lhs_batch_dim_size != rhs_batch_dim_size && lhs_batch_dim_size != 0 &&
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 21.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/variable_info_util.cc

      result->reserve(variable_indices.size());
      for (int var_idx : variable_indices) {
        Var* variable = nullptr;
        if (inputs[var_idx]->NumElements() == 0) {
          return errors::InvalidArgument("Empty resource tensor passed  at index ",
                                         var_idx,
                                         " to GetVariableInfosFromInputs.");
        }
        ResourceHandle handle = inputs[var_idx]->flat<ResourceHandle>()(0);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 7K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_config.cc

            const std::vector<std::string> indices =
                absl::StrSplit(node_specification, '-');
            for (const std::string& cur_index : indices) {
              custom_op_map[node_name].quantizable_input_indices.push_back(
                  std::stoi(cur_index));
            }
            break;
          }
          case CustomOpUpdateOptions::kWeightOnly:
            custom_op_map[node_name].is_weight_only =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 05 07:39:40 UTC 2024
    - 5.9K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tfr/python/tfr_gen.py

      def in_scf_scope(self):
        return self.scf_scope > 0
    
      def lookup(self, name):
        curr_idx = len(self.symbols) - 1
        while curr_idx >= 0 and (name not in self.symbols[curr_idx]['symbols']):
          curr_idx -= 1
        if curr_idx < 0:
          return None
        return self.symbols[curr_idx]['symbols'][name]
    
    
    class TFRGen(transformer.CodeGenerator):
      """Visit the AST and generate MLIR TFR functions."""
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 27 15:27:03 UTC 2022
    - 55.8K bytes
    - Viewed (0)
  8. schema/schema_test.go

    	if err != nil {
    		t.Fatalf("failed to parse embedded struct with primary key, got error %v", err)
    	}
    
    	fields := []schema.Field{
    		{Name: "ID", DBName: "cor_id", BindNames: []string{"CorpBase", "Model", "ID"}, DataType: schema.Uint, PrimaryKey: true, Size: 64, HasDefaultValue: true, AutoIncrement: true, TagSettings: map[string]string{"PRIMARYKEY": "PRIMARYKEY"}},
    Registered: Wed Jun 12 16:27:09 UTC 2024
    - Last Modified: Fri Dec 15 08:31:23 UTC 2023
    - 12.9K bytes
    - Viewed (0)
  9. src/cmd/link/internal/ld/pcln.go

    			off += 12
    		}
    		off = sb.SetUint32(ctxt.Arch, off, uint32(numPCData(ldr, s, fi)))
    
    		// Store the offset to compilation unit's file table.
    		cuIdx := ^uint32(0)
    		if cu := ldr.SymUnit(s); cu != nil {
    			cuIdx = cuOffsets[cu.PclnIndex]
    		}
    		off = sb.SetUint32(ctxt.Arch, off, cuIdx)
    
    		// startLine int32
    		off = sb.SetUint32(ctxt.Arch, off, uint32(startLine))
    
    		// funcID uint8
    		var funcID abi.FuncID
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Wed Feb 21 22:16:54 UTC 2024
    - 29.6K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/quantization/tensorflow/passes/replace_cast_hacks_with_tf_xla_ops.cc

          target_eq[v] = '_';
        }
      }
    
      if (target_eq.size() > out_eq.size()) return false;
    
      for (int i = 0; i < target_eq.size(); i++) {
        int out_idx = out_eq.size() - target_eq.size() + i;
        if (target_eq[i] != '_' && out_eq[out_idx] != target_eq[i]) {
          return false;
        }
    
        if (target_eq[i] != '_') out_dims.push_back(i);
      }
    
      return true;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 47.1K bytes
    - Viewed (0)
Back to top