- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 10 for mlir_passthrough_op (0.34 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/mlir_passthrough_op.pbtxt
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ops/mlir_passthrough_op.cc
==============================================================================*/ #include "tensorflow/core/framework/common_shape_fns.h" #include "tensorflow/core/framework/op.h" namespace tensorflow { REGISTER_OP("MlirPassthroughOp") .Attr("mlir_module: string") .Attr("Tinputs : list(type) >= 0") .Input("inputs: Tinputs") .Attr("Toutputs : list(type) >= 0") .Output("outputs: Toutputs")
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Sep 14 23:15:53 UTC 2019 - 1.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/BUILD
"@local_xla//xla/translate/mhlo_to_hlo:type_to_shape", "@stablehlo//:stablehlo_ops", ], alwayslink = 1, ) cc_library( name = "mlir_passthrough_op", srcs = ["ops/mlir_passthrough_op.cc"], visibility = [ "//visibility:public", ], deps = [ "//tensorflow/core:framework", ], alwayslink = 1, ) cc_library(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 47.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/materialize_passthrough_op.mlir
// RUN: tf-opt -allow-unregistered-dialect -tf-materialize-passthrough-op %s | FileCheck %s // Check that the MlirPassthroughOp is eliminated and replaced by its attached // MLIR module. // CHECK-LABEL: func @main func.func @main(%arg0 : tensor<10xf32>, %arg1 : tensor<10xf32>) -> tensor<10x10xf32> { // CHECK-SAME: (%[[ARG0:.*]]: tensor<10xf32>, %[[ARG1:.*]]: tensor<10xf32>)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 30 10:34:48 UTC 2022 - 1.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/type_attr.mlir
// CHECK: name: "plain" // CHECK: Placeholder // CHECK: key: "type" // CHECK: type: DT_INT8 func.func @main(%arg0 : tensor<16xf32>) { tf_executor.graph { %1:2 = tf_executor.island wraps "tf.MlirPassthroughOp"(%arg0) {extra_type_attr = [tensor<5xi32>, tensor<16xf32>], Tinputs = [tensor<16xf32>], Toutputs = [tensor<16xf32>], mlir_module = ""} : (tensor<16xf32>) -> tensor<16xf32> tf_executor.fetch } func.return }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 1.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/materialize_mlir_passthrough_op.cc
: public impl::MaterializePassthroughOpBase<MaterializePassthroughOpPass> { public: void runOnOperation() override; }; void MaterializePassthroughOpPass::runOnOperation() { getOperation().walk([](TF::MlirPassthroughOp op) { std::string module_string(op.getMlirModule()); // Parse the module. auto nested_module = parseSourceString<ModuleOp>(module_string, op.getContext()); if (!nested_module) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
mlir_module = '''python func @main(%arg0 : tensor<10xf32>, %arg1 : tensor<10xf32>) -> tensor<10x10xf32> { %add = "magic.op"(%arg0, %arg1) : (tensor<10xf32>, tensor<10xf32>) -> tensor<10x10xf32> return %ret : tensor<10x10xf32> } ''' @tf.function def foo(x, y): return mlir_passthrough_op([x, y], mlir_module, Toutputs=[tf.float32])
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/passes.h
// Transforms region bases control flow operations in the TensorFlow dialect to // their functional counterparts. std::unique_ptr<OperationPass<ModuleOp>> CreateTFRegionControlFlowToFunctional(); // Materialize the MlirPassthroughOp by replacing it with the MLIR module // attached as an attribute. std::unique_ptr<OperationPass<func::FuncOp>> CreateMaterializePassthroughOpPass();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 31.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
return %2 : tensor<?xi32> ``` }]; } def MaterializePassthroughOp : Pass<"tf-materialize-passthrough-op", "mlir::func::FuncOp"> { let summary = "Materialize the MlirPassthroughOp by replacing it with the " "MLIR module attached as an attribute"; let constructor = "TF::CreateMaterializePassthroughOpPass()"; let description = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/g3doc/_includes/tf_passes.md
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 02 02:26:39 UTC 2023 - 96.4K bytes - Viewed (0)