- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 230 for GetOperation (0.16 sec)
-
tensorflow/compiler/mlir/lite/transforms/legalize_tensorlist.cc
using OpRewritePattern::OpRewritePattern; LogicalResult matchAndRewrite(TF::AddNOp op, PatternRewriter& rewriter) const override { if (!HasVariantInputOrOutput(op.getOperation())) { return failure(); } auto converted = rewriter.create<TFL::CustomOp>( op->getLoc(), op->getResultTypes(), op->getOperands(), "VariantAddN",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 23:04:40 UTC 2024 - 10.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device_ops.td
StrAttr:$device ); let results = (outs Variadic<AnyType>:$results ); let regions = (region SizedRegion<1>:$body); let extraClassDeclaration = [{ Block &GetBody() { return getOperation()->getRegion(0).front(); } bool WrapsSingleOp(); }]; let builders = [ OpBuilder<(ins "StringAttr":$device, "TypeRange":$result_types), [{ $_state.addAttribute("device", device);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 23:53:20 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc
"target device."); } SmallVector<Location, 3> locations{contraction.getLoc(), bias_add.getLoc()}; SmallVector<Attribute, 2> fused_ops{StringAttr::get( context, bias_add.getOperation()->getName().stripDialect())}; // BiasAdd may or may not feed into an activation function. auto activation = GetActivation(bias_add);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_rewrite.cc
cluster_func_op.replaceAllUsesWith(xla_launch_op.getResults()); cluster_func_op.erase(); } void XlaRewritePass::runOnOperation() { ModuleOp module = getOperation(); SymbolTable symtab(module); OpBuilder builder(&getContext()); module.walk([&](tf_device::ClusterFuncOp cluster_func_op) { RewriteCall(cluster_func_op, symtab, builder); }); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/smuggle_disallowed_ops.cc
ConversionTarget target(getContext()); target.addIllegalOp<TF::ResizeBilinearOp, TF::ResizeNearestNeighborOp>(); target.addLegalDialect<mlir::stablehlo::StablehloDialect>(); if (failed(applyPartialConversion(getOperation(), target, std::move(patterns)))) { signalPassFailure(); } } }; std::unique_ptr<Pass> CreateSmuggleDisallowedOpsPass() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 19 01:14:04 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/testing/test_tf_to_stablehlo_pass.cc
public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(TestTFToStablehloPass) private: void runOnOperation() override; }; void TestTFToStablehloPass::runOnOperation() { ModuleOp module_op = getOperation(); MLIRContext* ctx = &getContext(); mlir::PassManager pm(ctx); AddTFToStablehloPasses(pm); if (!RunPassesOnModuleOp( /*mlir_dump_file_name=*/"test_tf_to_stablehlo_pass", pm, module_op)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 17:19:36 UTC 2024 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantize_composite_functions.cc
// TODO: b/321729008 - move this implementation to quantization_patterns.cc. if (merge_fusion_with_dequantize_) { pm.addPass(createMergeFusionWithDequantizePass()); } ModuleOp module_op = getOperation(); if (const absl::Status pm_run_status = RunPassesOnModuleOp(mlir_dump_file_name_, pm, module_op); !pm_run_status.ok()) { signalPassFailure(); } } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 02:59:01 UTC 2024 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/default_quant_params.cc
quant::AccumulatorScaleFunc func); quant::QuantParams default_quant_params_; }; } // namespace void DefaultQuantParamsPass::runOnOperation() { func::FuncOp func = getOperation(); OpBuilder builder(func); std::vector<Value> activation_values; std::vector<Value> bias_values; // First of all, collect all the values (block arguments and op results) which
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_validate_inputs.cc
"marker."; return failure(); } } return success(); } void XlaValidateInputsPass::runOnOperation() { ModuleOp module = getOperation(); SymbolTable symtab(module); llvm::SmallVector<func::FuncOp> entry_funcs = GetEntryFunctions(module); if (entry_funcs.empty()) { LOG(WARNING) << "missing entry functions"; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 19:29:14 UTC 2024 - 3.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/unfold_splat_constant_pass.cc
// pass and folders are not applied by default. class UnfoldSplatConstantPass : public impl::UnfoldSplatConstantPassBase<UnfoldSplatConstantPass> { public: void runOnOperation() override { auto module = getOperation(); mlir::OpBuilder op_builder(&module.getBodyRegion()); // Cannot use the pattern rewriter because the driver applies folders by // default. module.walk([&](mhlo::ConstantOp const_op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.8K bytes - Viewed (0)