- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 106 for created (0.57 sec)
-
tensorflow/compiler/mlir/lite/transforms/dense_to_sparse.cc
auto s_qconst = builder.create<SparseQConstOp>(op->getLoc(), cst.getQtypeAttr(), cst.getValue(), s_param, new_value); value.replaceAllUsesWith(s_qconst.getResult()); cst.erase(); } if (result.needs_densify) { auto value = op->getOperand(operand); auto densify = builder.create<DensifyOp>(op->getLoc(), value.getType(), value);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/dot_general.cc
RankedTensorType::get(static_cast<int32_t>(1), builder.getI32Type()); auto out_segids_cst = builder.create<TFL::ConstOp>( builder.getI32TensorAttr(flattened_out_segids)); auto contracting_segids_cst = builder.create<TFL::ConstOp>( builder.getI32TensorAttr(flattened_contracting_segids)); auto num_segids_tensor = builder.create<TFL::ConstOp>(DenseIntElementsAttr::get(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.2K bytes - Viewed (0) -
tensorflow/compiler/jit/flags.cc
"Disable deadness related safety checks when clustering (this is " "unsound)."), Flag("tf_xla_disable_resource_variable_safety_checks_for_debugging", &mark_for_compilation_flags ->tf_xla_disable_resource_variable_safety_checks_for_debugging, "Disable resource variables related safety checks when clustering " "(this is unsound)."),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 24.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/quantize.cc
// be categorized as cusom ops despite each of them may require different // behaviors. In that case, these ops can be marked in the custom map and // treated separately in this pass. auto custom_op = llvm::dyn_cast_or_null<CustomOp>(op); if (!custom_op) return false; // Custom op which is marked in the custom op map is quantizable.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
nodes_[inst] = node; return absl::OkStatus(); } bool IsEntryFunctionArg(BlockArgument arg) { return arg.getParentRegion()->getParentOfType<FuncOp>().getName() == "main"; } // Creates argument nodes from Block argument. If a name is supplied, that // name will be used instead of generating a unique name. Status Exporter::AddArgumentNode(BlockArgument arg, unsigned index,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
nodes_[inst] = node; return absl::OkStatus(); } bool IsEntryFunctionArg(BlockArgument arg) { return arg.getParentRegion()->getParentOfType<FuncOp>().getName() == "main"; } // Creates argument nodes from Block argument. If a name is supplied, that // name will be used instead of generating a unique name. Status Exporter::AddArgumentNode(BlockArgument arg, unsigned index,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize.cc
/*quant_dim=*/-1, num_bits, narrow_range, is_signed); builder.setInsertionPoint(block, insertion_point); auto q_op = builder.create<quantfork::QuantizeCastOp>( loc, params.getValue(), arg); auto dq_op = builder.create<quantfork::DequantizeCastOp>( loc, input_type, q_op.getResult()); arg.replaceAllUsesWith(dq_op.getResult()); q_op.setOperand(arg); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/resource_alias_analysis.cc
// Keep track of IDs of resources that are allocated by ops with // `UniqueResourceAllocation` trait, this can be utilized for while-loop // parallelization (every iteration creates a new unique resource). if (op->hasTrait<OpTrait::TF::UniqueResourceAllocation>()) { unique_resource_allocation_ids_.insert(resource_handle.id); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 28.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/collection_ops_util.cc
if (elem_type && elem_type.hasStaticShape()) return elem_type; } } return std::nullopt; } // Creates a ReadVariableOp on a local variable. Value ReadLocalVariable(Value local_var, OpBuilder builder, Location loc) { return builder .create<TF::ReadVariableOp>( loc, ArrayRef<Type>{mlir::cast<TF::ResourceType>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc
const llvm::StringMap<Attribute>& derived_attrs) const { // Create the new op Location loc = call_op.getLoc(); rewriter.setInsertionPointAfter(call_op); std::string tf_op_name = GetTFOpName(call_op.getCallee()); OperationState new_state(loc, tf_op_name, inputs, output_types, attr_list); Operation* new_op = rewriter.create(new_state); if (materialize_derived_attrs_) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.8K bytes - Viewed (0)