- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 63 for _noinline (0.17 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/tests/mark_functions_noinline.mlir
// RUN: tf-quant-opt %s -mark-functions-noinline='noinline-functions=noinline0' \ // RUN: -allow-unregistered-dialect -mlir-disable-threading \ // RUN: -split-input-file -verify-diagnostics | FileCheck %s // Tests that the function is marked tf._noinline = true. // CHECK-LABEL: @noinline0 // CHECK-SAME: attributes {{{.*tf._noinline = true.*}}} func.func @noinline0() -> (tensor<0xf32>) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 12 00:02:46 UTC 2023 - 841 bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/mark_functions_noinline.cc
// Required when using LLVM_DEBUG macro. #define DEBUG_TYPE "mark-functions-noinline" namespace mlir { namespace quant { namespace { // Name of the boolean attribute indicating whether the function can be // inlined or not. constexpr StringRef kTfNoinlineAttr = "tf._noinline"; // This pass marks functions with the attribute `tf._noinline = true` so that
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 18 02:52:57 UTC 2023 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/strip_noinline.mlir
// RUN: tf-opt %s -tf-strip-noinline-attribute | FileCheck %s // CHECK-LABEL: func @strip_simple( // CHECK-NOT: tf._noinline func.func @strip_simple() -> tensor<2xi32> attributes {tf._noinline = true} { // CHECK-NEXT: %[[CST:.*]] = "tf.Const" %cst = "tf.Const"() { value = dense<2> : tensor<2xi32> } : () -> tensor<2xi32> func.return %cst : tensor<2xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 24 05:47:26 UTC 2022 - 364 bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/strip_noinline_attribute.cc
// This tranformation pass strips any "_noinline" attributes from the module. struct StripNoinlineAttributePass : public impl::StripNoinlineAttributePassBase<StripNoinlineAttributePass> { public: // void runOnOperation() override; void runOnOperation() override { // Strip the "tf._noinline" attribute from top-level functions. for (auto func_op : getOperation().getOps<func::FuncOp>())
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/lite/stablehlo/tests/legalize-skip-partitioned-calls.mlir
func.func private @some_func(%arg0: tensor<1x2x2x3xf32>) -> tensor<1x2x2x3xf32> attributes {tf._noinline = true} { return %arg0 : tensor<1x2x2x3xf32> } // CHECK-SKIP: func.func private @some_other_func func.func private @some_other_func(%arg0: tensor<1x2x2x3xf32>) -> tensor<1x2x2x3xf32> attributes {tf._noinline = true} { return %arg0 : tensor<1x2x2x3xf32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 08 20:05:12 UTC 2024 - 1.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/xla_cluster_formation.mlir
// Check that we only outline the outermost partitioned call with // `_xla_compile_device_type` in nested calls to a device cluster. The callee // of any partitioned call outside of a device cluster is marked with // `tf._noinline = true` to prevent it getting inlined to perseve runtime // performance. // CHECK-LABEL: func.func @nested_calls func.func @nested_calls(%arg0: tensor<i32>) -> tensor<i32> attributes {tf.entry_function = {}} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 19:09:44 UTC 2023 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/saved_model/testdata/xla_launch.mlir
func.return %1 : tensor<i32> } func.func private @xla_func_0(%arg0: tensor<1x3xf32>, %arg1: tensor<1x3xf32>) -> tensor<1x3xf32> attributes {tf._XlaMustCompile = true, tf._noinline = true, tf._original_func_name = "should_not_be_used"} { %1 = "tf.AddV2"(%arg0, %arg1) : (tensor<1x3xf32>, tensor<1x3xf32>) -> tensor<1x3xf32> %2 = "tf.Const"() {value = dense<0> : tensor<i32>} : () -> tensor<i32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 14 15:35:49 UTC 2023 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/saved_model/testdata/xla_launch_xla_reduce_window.mlir
%1 = "tf.ReadVariableOp"(%0) {device = "/device:CPU:0"} : (tensor<!tf_type.resource<tensor<f32>>>) -> tensor<f32> %2 = "tf.XlaLaunch"(%arg0, %1) {_noinline = true, _xla_compile_device_type = "GPU", device = "/device:GPU:0", function = @xla_func_0, operandSegmentSizes = array<i32: 0, 2, 0>} : (tensor<7xf32>, tensor<f32>) -> tensor<10xf32> func.return %2 : tensor<10xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 14 15:35:49 UTC 2023 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/inlining.mlir
%cst = "tf.Const"() { value = dense<2> : tensor<2xi32> } : () -> tensor<2xi32> func.return %cst : tensor<2xi32> } func.func private @simple_callee_with_noinline() -> tensor<2xi32> attributes {tf._noinline = true} { %cst = "tf.Const"() { value = dense<2> : tensor<2xi32> } : () -> tensor<2xi32> func.return %cst : tensor<2xi32> } // Test that simple TF operations can be inlined. // CHECK-LABEL: func @inline_simple(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/components/tf_to_stablehlo.mlir
return %arg0 : tensor<1x2x2x3xf32> } } // CHECK: module // CHECK: tf.StatefulPartitionedCall // CHECK: func.func private @some_func // CHECK-NOT: func.call // ----- // This test makes sure functions without tf._noinline=true is inlined. module {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 08 20:05:12 UTC 2024 - 13.6K bytes - Viewed (0)