- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 229 for getOperation (0.19 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/tflite_legalize_hlo.cc
// TODO: b/304003568 - Add TF_TransposeOp folding logic to tflite. target.addIllegalOp<mhlo::DotGeneralOp, mhlo::DotOp, mhlo::TransposeOp>(); if (failed(applyPartialConversion(getOperation(), target, std::move(patterns)))) { getOperation().emitError("mhlo to TFLite legalization failed."); signalPassFailure(); } } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/quantize_variables.cc
var_handle_op.erase(); } } void QuantizeVariablesPass::runOnOperation() { ResourceIdMap resource_id_map; ResourceMap resource_map; // Collect all resource identities. getOperation().walk([&](TFL::VarHandleOp var_handle_op) { auto identity = std::make_pair(var_handle_op.getContainer().str(), var_handle_op.getSharedName().str()); resource_id_map.insert(
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/lite/transforms/optimize_functional_ops.cc
if (op.use_empty() && (op.getIsStateless() || (IsSideEffectFree(then_func) && IsSideEffectFree(else_func)))) { rewriter.eraseOp(op.getOperation()); return success(); } // Extract the constant cond value. DenseElementsAttr cond; if (!matchPattern(op.getCond(), m_Constant(&cond))) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/strip_noinline_attribute.cc
public: // void runOnOperation() override; void runOnOperation() override { // Strip the "tf._noinline" attribute from top-level functions. for (auto func_op : getOperation().getOps<func::FuncOp>()) func_op->removeAttr("tf._noinline"); } }; } // namespace std::unique_ptr<OperationPass<ModuleOp>> CreateStripNoinlineAttributePass() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 1.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables_test_pass.cc
struct FreezeVariableTestPass : public impl::FreezeVariablesTestPassBase<FreezeVariableTestPass> { void runOnOperation() override { TF::test_util::FakeSession session; if (tf_saved_model::FreezeVariables(getOperation(), &session).failed()) signalPassFailure(); } }; } // namespace std::unique_ptr<OperationPass<ModuleOp>> CreateFreezeVariableTestPass() { return std::make_unique<FreezeVariableTestPass>(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 04 09:19:38 UTC 2022 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/verify_suitable_for_graph_export_pass.cc
class VerifySuitableForExportPass : public impl::VerifySuitableForExportPassBase< VerifySuitableForExportPass> { public: void runOnOperation() override { if (failed(tensorflow::VerifyExportSuitable(getOperation()))) return signalPassFailure(); } }; } // namespace std::unique_ptr<OperationPass<ModuleOp>> CreateVerifySuitableForExportPass() { return std::make_unique<VerifySuitableForExportPass>(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_functional_to_executor.cc
to_fetch = to_fetch.drop_back(); } builder.create<tf_executor::FetchOp>(loc, to_fetch); // Build Island. island.getBody().push_back(new Block); island.getBody().front().getOperations().splice( island.getBody().front().begin(), body.getOperations(), copy_range.begin(), copy_range.end()); builder.setInsertionPointToEnd(&island.getBody().front()); builder.create<tf_executor::YieldOp>(loc, return_op.getOperands());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 11 20:52:36 UTC 2023 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/transforms/shlo_simplify.cc
class SHLOSimplifyPass : public impl::SHLOSimplifyPassBase<SHLOSimplifyPass> { public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(SHLOSimplifyPass) void runOnOperation() override { ModuleOp module = getOperation(); RewritePatternSet patterns(&getContext()); populateWithGenerated(patterns); PopulateFolderPatterns(patterns); if (failed(applyPatternsAndFoldGreedily(module, std::move(patterns)))) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 03:05:20 UTC 2024 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/remove_sharding_custom_call.cc
RemoveShardingCustomCallPass>::RemoveShardingCustomCallPassBase; private: void runOnOperation() override; }; void RemoveShardingCustomCallPass::runOnOperation() { func::FuncOp func_op = getOperation(); MLIRContext& ctx = getContext(); RewritePatternSet patterns(&ctx); populateWithGenerated(patterns); FrozenRewritePatternSet frozen_patterns(std::move(patterns));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 13 07:04:47 UTC 2024 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/print.cc
void PrintPass::runOnOperation() { llvm::sys::SmartScopedLock<true> instrumentationLock(mutex_); OpPrintingFlags flags = OpPrintingFlags().elideLargeElementsAttrs().enableDebugInfo(false); getOperation()->print(*os_, flags); } } // namespace std::unique_ptr<OperationPass<ModuleOp>> CreatePrintPass(raw_ostream* os) { return std::make_unique<PrintPass>(os); } } // namespace TF
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 12 23:15:17 UTC 2023 - 2.1K bytes - Viewed (0)