- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for cur_idx (0.39 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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)