- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 72 for TensorFlowDialect (1.11 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.cc
} TensorFlowDialect::ConstantFoldHook TensorFlowDialect::constant_fold_hook_; TensorFlowDialect::TensorFlowDialect(MLIRContext *context) : Dialect(/*name=*/"tf", context, TypeID::get<TensorFlowDialect>()) { context->getOrLoadDialect<::mlir::tf_type::TFTypeDialect>(); addOperations< #define GET_OP_LIST #include "tensorflow/compiler/mlir/tensorflow/ir/tf_all_ops.cc.inc" >();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/passes.td
that disperse values. (ex: convolution, dot_general) }]; let dependentDialects = [ "mlir::func::FuncDialect", "mlir::stablehlo::StablehloDialect", "TF::TensorFlowDialect", ]; } def ReplaceStablehloOpsInMainFunctionWithXlaCallModuleOpsPass : Pass<"stablehlo-replace-stablehlo-ops-in-main-function-with-xla-call-module-ops", "mlir::ModuleOp"> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/testing/passes.td
let description = [{ Runs the pre calibration passes for post-training quantization with default configuration. }]; let dependentDialects = [ "mlir::stablehlo::StablehloDialect", "mlir::TF::TensorFlowDialect", "mlir::func::FuncDialect", "mlir::tf_executor::TensorFlowExecutorDialect", "mlir::mhlo::MhloDialect", "mlir::vhlo::VhloDialect", ]; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 23:21:42 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/call_graph_util_test.cc
} func.func @func(%arg0: tensor<i32>) -> tensor<i32> { func.return %arg0 : tensor<i32> } )mlir"; mlir::MLIRContext context; context.loadDialect<mlir::func::FuncDialect, mlir::TF::TensorFlowDialect>(); mlir::OwningOpRef<mlir::ModuleOp> module = mlir::parseSourceString<mlir::ModuleOp>(code, &context); ASSERT_TRUE(module); auto entry_funcs = GetEntryFunctions(*module); EXPECT_EQ(entry_funcs.size(), 2);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 20 04:39:18 UTC 2023 - 11.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/passes.td
let description = [{ Convert TF Quant ops to MHLO quant ops. }]; let constructor = "mlir::quant::stablehlo::CreateConvertTFQuantOpsToMHLOPass()"; let dependentDialects = ["TF::TensorFlowDialect", "chlo::ChloDialect", "mhlo::MhloDialect", "tf_type::TFTypeDialect", "quant::QuantizationDialect"]; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 23 01:41:18 UTC 2024 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/utils/dialect_detection_utils_test.cc
using mlir::OperationState; using mlir::UnknownLoc; using mlir::chlo::ChloDialect; using mlir::TF::TensorFlowDialect; using tensorflow::tf2xla::internal::IsInBridgeAcceptableDialects; class SharedUtilsTest : public ::testing::Test {}; TEST_F(SharedUtilsTest, IsInFunctionalDialectPasses) { MLIRContext context; context.loadDialect<TensorFlowDialect>(); OpBuilder opBuilder(&context);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 08 20:33:37 UTC 2023 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/metrics/error_collector_inst_test.cc
class MockSuccessPass : public PassWrapper<MockSuccessPass, OperationPass<ModuleOp>> { void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect>(); } public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(MockSuccessPass) explicit MockSuccessPass() = default; private: void runOnOperation() override {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 25 01:48:36 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util_test.cc
func.return %0 : tensor<i32> } )mlir"; mlir::MLIRContext context; context.loadDialect<mlir::func::FuncDialect, mlir::TF::TensorFlowDialect>(); mlir::OwningOpRef<mlir::ModuleOp> module = mlir::parseSourceString<mlir::ModuleOp>(code, &context); ASSERT_TRUE(module); EXPECT_TRUE(IsSupportedByReplicatedBridge(*module)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 19:51:50 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/python/tfr_wrapper.cc
PYBIND11_MODULE(tfr_wrapper, m) { m.def("verify", [](std::string input) { mlir::DialectRegistry registry; registry.insert<mlir::arith::ArithDialect, mlir::scf::SCFDialect, mlir::TF::TensorFlowDialect, mlir::func::FuncDialect, mlir::shape::ShapeDialect, mlir::TFR::TFRDialect>(); mlir::MLIRContext ctx(registry); ctx.loadAllAvailableDialects();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 13 06:54:12 UTC 2024 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/odml_converter_main.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 19:05:30 UTC 2024 - 2.2K bytes - Viewed (0)