Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for _FusedConv2D (0.17 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/fused_kernel_matcher.mlir

    // CHECK-LABEL: conv2DBiasAdd_noActivation
    func.func @conv2DBiasAdd_noActivation(%arg0: tensor<128xf32>, %arg1: tensor<1x1x3x128xf32>, %arg2: tensor<8x32x32x3xf32>) -> (tensor<*xf32>) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 13.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc

    //   <Contraction> + BiasAdd + <Activation> -> <FusedContraction>
    //
    // Note that fusion with activation is preferred, but a contraction and BiasAdd
    // can also be replaced by a _FusedConv2D if there is no other activation
    // function.
    // i.e., this class also supports the following fusion:
    //   <Contraction> + BiasAdd -> <FusedContraction>
    //
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/tf_to_corert_pipeline.mlir

    // CHECK-NEXT: [[o6:%.*]] = tfrt_fallback_async.executeop key({{[0-9]+}}) cost({{.*}}) device("/job:localhost/replica:0/task:0/device:CPU:0") "tf._FusedConv2D"([[arg1]], [[o3]], [[o2]])
    // CHECK-NEXT: [[o7:%.*]] = tfrt_fallback_async.executeop key({{[0-9]+}}) cost({{.*}}) device("/job:localhost/replica:0/task:0/device:CPU:0") "tf.AvgPool"([[o6]])
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 00:18:59 UTC 2024
    - 7.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-prefer-tf2xla.mlir

      %input_scale = "tf.Const"() {value = dense<1.0> : tensor<f32>} : () -> tensor<f32>
      %side_input_scale = "tf.Const"() {value = dense<2.0> : tensor<f32>} : () -> tensor<f32>
      %conv2d = "tf._FusedConv2D"(%input, %filter, %bias, %act, %input_scale, %side_input_scale) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 06 15:32:52 UTC 2024
    - 15.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/mark_for_compilation_pass.cc

               {"FusedBatchNorm", "FusedBatchNormV2", "FusedBatchNormV3",
                "_FusedBatchNormEx", "FusedBatchNormGrad", "FusedBatchNormGradV2",
                "FusedBatchNormGradV3"}},
              {"Conv", {"_FusedConv2D"}},
              {"SORT", {"TopKV2"}},  // XLA version much faster then TF version.
              {"MISC",
               // clang-format off
         {"ApproxTopK", "BroadcastTo", "ExpandDims", "Fill", "NoOp",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

    are specified by `args`.
    * If there is an op A specified, the output of op X is the input to op A, and op
    A produces the _FusedConv2D output. Otherwise, op X produces the _FusedConv2D
    output.
    
    *NOTE*: Do not invoke this operator directly in Python. Grappler is expected to
    create these operators.
      }];
    
      let arguments = (ins
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top