- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 111 for Setxattr (0.11 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/ops/tf_op_quant_spec.cc
auto spec = std::make_unique<OpQuantSpec>(); if (auto call_op = dyn_cast<TF::PartitionedCallOp>(op)) { StringRef function_name = mlir::cast<FlatSymbolRefAttr>(call_op.getFAttr()).getValue(); if (!function_name.starts_with("composite_")) { return spec; } if (function_name.contains("depthwise_conv2d")) { spec->coeff_op_quant_dim[1] = 3;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/common/cost.h
namespace mlir { namespace TFL { namespace tac { // Cost attribute string on the TFL dialect. constexpr char kCost[] = "tac.cost"; inline void UpdateCost(Operation* op, float cost, OpBuilder* builder) { op->setAttr(kCost, builder->getF32FloatAttr(cost)); } // Get the cost annotated with kCost. inline bool GetCostOnOp(Operation* op, float* cost) { auto cost_type = op->getAttrOfType<FloatAttr>(kCost); if (cost_type == nullptr) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jul 21 01:22:53 UTC 2021 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_variables.cc
void runOnOperation() override { auto module = getOperation(); // If TFLite variable legalization is not allowed, then we skip this pass. if (auto legalize_tfl_variables_attr = module->getAttr(kLegalizeTflVariables)) { if (!mlir::cast<BoolAttr>(legalize_tfl_variables_attr).getValue()) return; } RewritePatternSet patterns(&getContext()); populateWithGenerated(patterns);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/initialize_variables_in_session_init.cc
auto func = builder.create<func::FuncOp>(module->getLoc(), kSessionInitFuncName, func_type); func->setAttr(kTfSavedModelExportedNamesAttr, builder.getStrArrayAttr({kSessionInitFuncName})); func->setAttr(kTfSavedModelInitializerTypeAttr, builder.getStringAttr(kTfSavedModelInitializerRestoreType)); func.setVisibility(mlir::func::FuncOp::Visibility::Public);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/manipulate_model_attr.cc
const std::string new_inputs_attr_str = llvm::join(std::move(inputs_attrs), /*Separator=*/","); named_attr.setValue(StringAttr::get(ctx, new_inputs_attr_str)); } func_op->setAttr(kTfEntryFunctionAttr, DictionaryAttr::get(ctx, entry_func_attrs)); } } // namespace quant
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 26 01:13:26 UTC 2023 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_dynamic_layout_pass.cc
builder.setInsertionPoint(execute_launch); auto copy_with_layout = BuildCopyWithLayout(execute_launch, compile_launch, get_layout, input, &builder); copy_with_layout->setAttr(kDeviceAttr, execute_launch.getDeviceAttr()); execute.setOperand(execute_arg_index, copy_with_layout); } // Performs transformation for replicated inputs. Returns true if this is a
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/add_dump_tensor_op.cc
new_call_op->setAttr(kEntryFuncAttrName, rewriter.getStringAttr(new_ref_func_name.getValue())); new_call_op->setAttrs(call_op->getAttrs()); new_call_op->removeAttr(rewriter.getStringAttr(kQuantTraitAttrName)); FlatSymbolRefAttr new_func_name_attr = FlatSymbolRefAttr::get(rewriter.getContext(), new_ref_func_name); new_call_op->setAttr(kEntryFuncAttrName, new_func_name_attr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 22:55:22 UTC 2024 - 13K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/propagate_quantize_type.cc
} LogicalResult matchAndRewrite(TF::PartitionedCallOp op, PatternRewriter& rewriter) const override { const auto f_attr = mlir::dyn_cast<FlatSymbolRefAttr>(op.getFAttr()); StringRef function_name = f_attr.getValue(); if (!function_name.starts_with(kDequantizeFunctionName)) return failure(); llvm::SmallVector<Operation*> users(op->getUsers().begin(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/canonicalize_compile_and_replicate_attributes.cc
// Convert `_tpu_replicate`. if (op->hasAttr(TF::kTpuReplicateAttr)) { op->setAttr(tensorflow::kReplicationInfoAttr, op->getAttr(TF::kTpuReplicateAttr)); op->removeAttr(tensorflow::kTpuReplicateAttr); op->setAttr(tensorflow::kCompileDeviceTypeAttr, builder.getStringAttr(tensorflow::kTpuDevice)); } // Convert `_XlaMustCompile`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_quantizable_spots_as_functions_drq.cc
<< min_num_elements_for_weights_ << " elements)."; call_op->removeAttr(kQuantTraitAttrName); } } StringRef function_name = mlir::cast<FlatSymbolRefAttr>(call_op.getFAttr()).getValue(); if ((quantization_method_ == tensorflow::quantization::QuantizationMethod:: METHOD_DYNAMIC_RANGE_INT8) && (function_name.contains("batch_matmul") ||
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.5K bytes - Viewed (0)