- Sort Score
- Result 10 results
- Languages All
Results 141 - 150 of 382 for get_operations (0.17 sec)
-
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/tfr/passes/rewrite_quantized_io.cc
"storage types."; } void runOnOperation() override; }; void RewriteQuantizedIOPass::runOnOperation() { ModuleOp module = getOperation(); OpBuilder builder(module); module.walk([&](func::FuncOp func) { Block& block = func.front(); Operation* terminator = block.getTerminator(); // Replace input_arg(tensor<quant_type>) -> tfr.cast
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 4.5K bytes - Viewed (0) -
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/transforms/strip_saved_module_metadata.cc
for (auto namedAttr : stripAttrs) { func.removeResultAttr(i, namedAttr.getName()); } } } void StripSavedModuleMetadataPass::runOnOperation() { auto module = getOperation(); StripModule(module); module.walk(StripFunction); } } // namespace std::unique_ptr<OperationPass<ModuleOp>> CreateStripSavedModuleMetadataPass() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 08:06:04 UTC 2023 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/strip_tf_attributes.cc
for (auto namedAttr : stripAttrs) { func.removeResultAttr(i, namedAttr.getName()); } } } void StripTfAttributesPass::runOnOperation() { ModuleOp module = getOperation(); // strip module itself auto stripAttrs = llvm::to_vector<4>(llvm::make_filter_range( module->getAttrs(), [](NamedAttribute namedAttr) { return ShouldStripAttr(namedAttr); }));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 08:06:04 UTC 2023 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/order_by_dialect.cc
return predecessor && predecessor->getName().getDialectNamespace() == op->getName().getDialectNamespace(); } void OrderByDialectPass::runOnOperation() { ModuleOp module = getOperation(); for (func::FuncOp func : module.getOps<func::FuncOp>()) { std::vector<std::pair<Operation*, Operation*>> side_effect_data; const detail::SideEffectAnalysisInfo* info = nullptr; auto extra_dependencies =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 08 17:01:11 UTC 2022 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/tac_pass.h
template <typename T> class TacFunctionPass : public TacPass<func::FuncOp> { public: using TacPass<func::FuncOp>::TacPass; ~TacFunctionPass() override = default; mlir::func::FuncOp getFunction() { return getOperation(); } virtual void runOnFunction() = 0; void runOnOperation() final { if (!getFunction().isExternal()) runOnFunction(); } protected: // Returns the derived pass name.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 03:08:33 UTC 2023 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_session_initializer_to_function.cc
ConvertSessionInitializerToFunctionPass> { void runOnOperation() override; }; void ConvertSessionInitializerToFunctionPass::runOnOperation() { ModuleOp module = getOperation(); auto session_initializer = tf_saved_model::GetSessionInitializerOp(module); if (!session_initializer) return; OpBuilder builder(session_initializer); const char *name = "session_initializer";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 01 05:03:09 UTC 2023 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/topological_sort.cc
if (extraDependencies) { return extraDependencies(op, /*incoming=*/false); } else { return empty_op_set; } }; int i = 0; for (Operation& op : block.getOperations()) { int incoming_ctrl_edges = 0; int incoming_data_edges = 0; op.walk([&](Operation* child) { ancestor[child] = &op; for (Operation* predecessor : ctrlPredecessors(child)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 08 17:01:11 UTC 2022 - 5K bytes - Viewed (0)