- Sort Score
- Result 10 results
- Languages All
Results 81 - 90 of 537 for funcMap (0.24 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/tfxla_device_specific_transforms.cc
get_alg_op->erase(); return success(); } void TFXLADeviceSpecificTransforms::runOnOperation() { if (!device_type_.hasValue()) return; auto func_op = getOperation(); auto walk_result = func_op->walk([&](TF::StatelessRandomGetAlgOp op) { if (failed(ConvertGetAlgOp(op))) { op->emitOpError( "Could not convert and remove Device specific information");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 05:56:39 UTC 2023 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/input_lowering_metrics_pass.cc
public: void runOnOperation() override; }; void InputMetricsLoweringPass::runOnOperation() { bool has_dynamic_op = false; Operation* func_op = getOperation(); func_op->walk([&](Operation* op) { auto abstractOp = op->getRegisteredInfo(); if (!abstractOp) return WalkResult::advance(); if (mlir::mhlo::IsDynamicPadderOp(abstractOp->getTypeID())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 08:55:35 UTC 2024 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/verify_no_outside_compilation_markers_pass.cc
return op.getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr) != nullptr; } void VerifyNoOutsideCompilationMarkersPass::runOnOperation() { Operation* func_op = getOperation(); auto walk_result = func_op->walk([&](Operation* op) { if (IsDeviceClusterOp(*op) && HasChildLaunchDeviceOp(*op)) { std::string launch_error = absl::StrCat("Node `", op->getName().getStringRef().str(), "` ",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Aug 22 19:52:08 UTC 2023 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.cc
} return init_func_ops; } func::FuncOp GetInitializerFunction(ModuleOp module_op, const StringRef initializer_type) { SmallVector<func::FuncOp, 2> init_func_ops = GetInitializerFunctions(module_op); auto init_func_itr = absl::c_find_if( init_func_ops, [initializer_type](const func::FuncOp init_func_op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/verify_quant_legalization.cc
} bool IsMhloUniformQuantizedOp(Operation& op) { return llvm::isa<mhlo::UniformQuantizeOp, mhlo::UniformDequantizeOp>(op); } void VerifyQuantLegalization::runOnOperation() { Operation* func_op = getOperation(); auto walk_result = func_op->walk([&](Operation* op) { // Verify all uq and qint types are lowered. if (llvm::any_of(op->getOperandTypes(), IsQuantType) || llvm::any_of(op->getResultTypes(), IsQuantType) ||
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/prepare_quantize.cc
auto func_op_quant_scale_spec = GetStableHloQuantConstraints; for (auto func_op : module_op.getOps<func::FuncOp>()) { // The function might contain more stats ops than required, and it will // introduce requantize if the calibration stats have conflicts. This tries // to remove all the redundant stats ops. RemoveRedundantStatsOps(func_op, func_op_quant_spec, func_op_quant_scale_spec);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 03 05:11:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils.cc
Value input = func_op.getArgument(0); Value output_init_state = func_op.getArgument(1); Value hidden_init_state = func_op.getArgument(2); Value weight_kernel = func_op.getArgument(3); Value recurrent_kernel = func_op.getArgument(4); Value bias = func_op.getArgument(5); // The func op should have 5 outputs. if (func_op.getNumResults() != 5) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 36.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/common/outline_operations.cc
std::string function_name = absl::StrCat("func_", subgraph.subgraph_id_); func::FuncOp new_func = func::FuncOp::create(builder.getUnknownLoc(), function_name, function_type); new_func.setVisibility(func::FuncOp::Visibility::Private); new_func.addEntryBlock(); // To form the body of the new function we need to clone each
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/stablehlo/passes/replace_stablehlo_ops_in_main_function_with_xla_call_module_ops.cc
// Quantizer via the tf_quant.composite_function UnitAttr attached to // func ops. We remove this attribute as this interferes with VHLO conversion. // Remove this temporary hack. for (auto func_op : module_op.getOps<func::FuncOp>()) { func_op->removeAttr(kFusedFunctionAttr); } } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_import.cc
absl::flat_hash_set<FunctionName> existing_func_names; module_op->walk([&](func::FuncOp func_op) { FunctionName func_name = func_op.getSymName().str(); existing_func_names.insert(func_name); // We may retrieve the original function's name from the attribute. // Functions without this attribute are ignored. auto original_func_name = func_op->getAttrOfType<StringAttr>("tf._original_func_name"); if (original_func_name) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 12:49:45 UTC 2024 - 6.4K bytes - Viewed (0)