- Sort Score
- Result 10 results
- Languages All
Results 181 - 190 of 293 for runOnOperation (0.16 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/decompose_optionals.cc
#define GEN_PASS_DEF_DECOMPOSEOPTIONALSPASS #include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc" struct DecomposeOptionalsPass : public impl::DecomposeOptionalsPassBase<DecomposeOptionalsPass> { void runOnOperation() override; }; class HandleOptionalFrom : public OpRewritePattern<TF::OptionalFromValueOp> { public: using OpRewritePattern::OpRewritePattern; LogicalResult matchAndRewrite(TF::OptionalFromValueOp op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 9.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/smuggle_disallowed_ops.cc
return "Smuggle disallowed ops via stablehlo.custom_calls"; } void getDependentDialects(DialectRegistry& registry) const final { registry.insert<mlir::stablehlo::StablehloDialect>(); } void runOnOperation() override { RewritePatternSet patterns(&getContext()); patterns.add<SmuggleOpPattern<TF::ResizeBilinearOp>>(&getContext()); patterns.add<SmuggleOpPattern<TF::ResizeNearestNeighborOp>>(&getContext());
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/tensorflow/transforms/check_control_dependencies.cc
class TFExecutorCheckControlDependencies : public impl::ExecutorCheckControlDependenciesPassBase< TFExecutorCheckControlDependencies> { public: void runOnOperation() override; }; using ResourceIdVec = llvm::SmallVector<std::pair<TF::ResourceId, bool>>; // MapVector provides both fast lookup and deterministic iteration order. using IslandToIslandMapVec =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 10.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_resource_partitioning.cc
struct TPUResourceReadsWritesPartitioningPass : public impl::TPUResourceReadsWritesPartitioningPassBase< TPUResourceReadsWritesPartitioningPass> { void runOnOperation() override; }; bool AllResourceTypesHaveSubtypes(TypeRange resources) { for (Type resource : resources) if (!llvm::hasSingleElement(resource.cast<TensorType>()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/parallel_execute_to_islands.cc
: public impl::ParallelExecuteToIslandsPassBase< ParallelExecuteToIslandsPass> { explicit ParallelExecuteToIslandsPass(bool legacy_graph_export) { legacy_graph_export_ = legacy_graph_export; } void runOnOperation() override; }; // Convert parallel_execute op to a set of islands where each region of // parallel_execute op becomes a separate island. This ensures that the regions
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 19 19:47:16 UTC 2023 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/insert_weight_param.cc
public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(InsertWeightParamPass) using impl::InsertWeightParamPassBase< InsertWeightParamPass>::InsertWeightParamPassBase; private: void runOnOperation() override; }; // Inserts quantization parameters for weights for hybrid quantization of // `stablehlo.convolution` and `stablehlo.dot_general`. class InsertWeightParamPattern
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 05:56:10 UTC 2024 - 10.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/default_quant_params.cc
this->default_min_ = default_min; this->default_max_ = default_max; this->is_signed_ = is_signed; } MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(DefaultQuantParamsPass) void runOnOperation() override; private: // Whether the value is used as a bias input of another op. Here we assume // bias is used immediately by the user. This assumption is always correct // after constant folding.
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/remove_unused_arguments.cc
// Removes unused arguments from functions and their callers. struct RemoveUnusedArgumentsPass : public impl::RemoveUnusedArgumentsPassBase<RemoveUnusedArgumentsPass> { void runOnOperation() override; }; // Return a bitvector that marks all return values that always come from // the same value. llvm::BitVector GetInvariantReturns(Region* region, int number_of_results) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/functional_control_flow_to_regions.cc
explicit FunctionalControlFlowToRegions(bool allow_passthrough_args) : FunctionalControlFlowToRegionsPassBase( FunctionalControlFlowToRegionsPassOptions{allow_passthrough_args}) { } void runOnOperation() override; }; // Creates a call to function `func` in region `caller_region`. Use `args` as // the call arguments, and terminate the region with a yield. The arguments are
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/tac_pass.h
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. StringRef getName() const override { return llvm::getTypeName<T>(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 03:08:33 UTC 2023 - 3.1K bytes - Viewed (0)