- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 16 for getFAttr (0.17 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tpu_model_to_cpu.cc
private: LogicalResult matchAndRewrite(TF::TPUPartitionedCallOp call_op, PatternRewriter& rewriter) const override { auto f_attr = mlir::dyn_cast<FlatSymbolRefAttr>(call_op.getFAttr()); auto module_op = call_op->getParentOfType<ModuleOp>(); SymbolTable symbol_table(module_op); auto f_name = f_attr.getValue(); func::FuncOp float_func =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.5K 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/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/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) -
tensorflow/compiler/mlir/quantization/tensorflow/calibrator/custom_aggregator_op.cc
float max_percentile; OP_REQUIRES_OK(context, context->GetAttr("calibration_method", &calibration_method_value)); OP_REQUIRES_OK(context, context->GetAttr("num_bins", &num_bins)); OP_REQUIRES_OK(context, context->GetAttr("min_percentile", &min_percentile)); OP_REQUIRES_OK(context, context->GetAttr("max_percentile", &max_percentile));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 01:09:50 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/lite/tfl_to_std.cc
auto dcast = b.create<DequantizeOp>(dq.getLoc(), dq.getResult().getType(), dq.getArg()); dq.getResult().replaceAllUsesWith(dcast); if (auto extra_attr = op->getAttr(mlir::quant::kVolatileOpAttrName)) { dcast->setAttr(mlir::quant::kVolatileOpAttrName, extra_attr); } dq.erase(); } else if (auto q = llvm::dyn_cast<quantfork::QuantizeCastOp>(op)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 02:50:01 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_custom_aggregation_op_to_quant_stats.cc
LogicalResult matchAndRewrite(TF::CustomAggregatorOp op, PatternRewriter &rewriter) const override { FloatAttr min = mlir::dyn_cast_or_null<FloatAttr>(op->getAttr("min")); FloatAttr max = mlir::dyn_cast_or_null<FloatAttr>(op->getAttr("max")); // When there are no min and max attributes, remove op. if (min == nullptr || max == nullptr) { op.getOutput().replaceAllUsesWith(op.getInput());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/calibrator/calibration_statistics_saver_op.cc
OP_REQUIRES_OK(context, context->GetAttr("output_file_path", &output_file_path)); OP_REQUIRES_OK(context, context->env()->NewWritableFile(output_file_path, &output_file_)); OP_REQUIRES_OK(context, context->GetAttr("ids", &ids_)); OP_REQUIRES_OK(context, context->GetAttr("calibration_methods",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 13 01:31:23 UTC 2024 - 8K bytes - Viewed (0) -
src/cmd/vendor/golang.org/x/sys/unix/symaddr_zos_s390x.s
MOVD $·Pipe2(SB), R8 MOVD R8, ret+0(FP) RET TEXT ·get_FlockAddr(SB), NOSPLIT|NOFRAME, $0-8 MOVD $·Flock(SB), R8 MOVD R8, ret+0(FP) RET TEXT ·get_GetxattrAddr(SB), NOSPLIT|NOFRAME, $0-8 MOVD $·Getxattr(SB), R8 MOVD R8, ret+0(FP) RET TEXT ·get_NanosleepAddr(SB), NOSPLIT|NOFRAME, $0-8 MOVD $·Nanosleep(SB), R8 MOVD R8, ret+0(FP) RET TEXT ·get_SetxattrAddr(SB), NOSPLIT|NOFRAME, $0-8
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Wed May 08 16:12:58 UTC 2024 - 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)