- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 111 for Setxattr (0.09 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize.cc
static bool AllowDynamicRangeQuantizedOperand( Operation* quantized_op, const CustomMap& custom_op_map) { auto call_op = cast<TF::PartitionedCallOp>(quantized_op); StringRef function_name = call_op.getFAttr().cast<FlatSymbolRefAttr>().getValue(); // The below can be generalized as there are more read-only ops added such // as slice. const bool is_gather = function_name.contains("gather");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 23.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/device_util.cc
devices.push_back(builder.getNamedAttr(name, metadata)); } } op->setAttr(kDevicesAttr, builder.getDictionaryAttr(devices)); } mlir::LogicalResult GetDevicesFromOp(mlir::Operation* op, mlir::TF::RuntimeDevices* devices) { auto devices_attr = op->getAttr(kDevicesAttr); if (!devices_attr) return mlir::success();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/attrs_and_constraints.h
} // Returns the function attribute for the given call op which is lifted for // quantization. inline FlatSymbolRefAttr GetFuncAttr(TF::PartitionedCallOp call_op) { return mlir::dyn_cast<FlatSymbolRefAttr>(call_op.getFAttr()); } inline FlatSymbolRefAttr GetFuncAttr(TF::XlaCallModuleOp call_op) { return call_op->getAttrOfType<FlatSymbolRefAttr>( TF::kStablehloEntryFunctionAttrName); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/add_functions_for_exported_names.cc
// Set the (array of) exported name(s) of a (public) function to just // contain the given string. void SetExportedName(func::FuncOp f, StringRef name) { OpBuilder b(f); f->removeAttr(kTfSavedModelExportedNamesAttr); f->setAttr(kTfSavedModelExportedNamesAttr, b.getStrArrayAttr({name})); } // Convert a savedmodel public function into a private function. // This means we need to remove any attributes that are only allowed
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 08:06:04 UTC 2023 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/analyze_variables.cc
getElementTypeOrSelf(operand.getType()))) { legalize_to_tfl = false; return WalkResult::interrupt(); } } return WalkResult::advance(); }); module->setAttr(kLegalizeTflVariables, BoolAttr::get(context, legalize_to_tfl)); } } // namespace std::unique_ptr<OperationPass<ModuleOp>> CreateAnalyzeVariablesPass() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/unfreeze_constants.cc
/*arg_types=*/{}, /*arg_locs=*/{}); init_func->setAttr(kTfSavedModelExportedNamesAttr, builder.getStrArrayAttr( {"tf_saved_model.session_initializer_restore_op"})); init_func->setAttr( kTfSavedModelInitializerTypeAttr, builder.getStringAttr(kTfSavedModelInitializerRestoreType));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 16 15:04:53 UTC 2023 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/quantization_context.cc
input_specs.push_back(TypeAttr::get(requantize.params)); } else { input_specs.push_back(TypeAttr::get(state.params)); } } op->setAttr("input_specs", ArrayAttr::get(context, input_specs)); llvm::SmallVector<Attribute, 4> output_specs; auto original_output_specs = op.getOutputSpecs().getValue();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 08 01:38:03 UTC 2024 - 13.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/colocate_tpu_copy_with_dynamic_shape.cc
state = solver.lookupState<Device>(result); if (state) break; } if (!state || !state->hasDevice()) { return WalkResult::advance(); } op->setAttr(kDevice, state->getDevice()); return WalkResult::advance(); }); } } // namespace std::unique_ptr<OperationPass<ModuleOp>> CreateColocateTPUCopyWithDynamicShapePass() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 23 00:30:27 UTC 2023 - 5.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/post_quantize.cc
PatternRewriter& rewriter) const override { auto input_op = op.getArg().getDefiningOp(); if (auto q = llvm::dyn_cast_or_null<quantfork::QuantizeCastOp>(input_op)) { if (!q->getAttr(kVolatileOpAttrName)) return failure(); if (remove_volatile_ops_type == kPreserveInputsAndOutputs) { // Don't remove leading and trailing QDQ for PTQ workflow, so the io // modifying lib can work correctly.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 07:39:40 UTC 2024 - 5.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/group_by_dialect.cc
auto func = builder.create<mlir::func::FuncOp>( ops[0]->getLoc(), dialect.str() + std::to_string(function_id), builder.getFunctionType(input_types, output_types)); func->setAttr("dialect", builder.getStringAttr(dialect)); auto block = func.addEntryBlock(); llvm::DenseSet<Operation*> all_operations; for (Operation* outer : ops) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 17 07:31:01 UTC 2023 - 8K bytes - Viewed (0)