Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for BiasAddV1 (0.14 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/canonicalize.td

      [(IsRank4Tensor $input), (IsRank2Tensor $crops)], [(CopyAttrs $src, $dest)]>;
    
    //===----------------------------------------------------------------------===//
    // BiasAddV1 op patterns.
    //===----------------------------------------------------------------------===//
    
    def BiasAddV1ToBiasAdd : Pat<
      (TF_BiasAddV1Op:$src $arg0, $arg1),
      (TF_BiasAddOp:$dest $arg0, $arg1,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Dec 06 18:42:28 UTC 2023
    - 17K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc

        BiasAddOp bias_add = GetBiasAdd(contraction.getResult());
        if (!bias_add) {
          return rewriter.notifyMatchFailure(
              contraction, "does not feed into a tf.BiasAdd/tf.BiasAddV1 op");
        }
    
        if (!AreFuseCompatible(contraction, bias_add, rewriter)) {
          return rewriter.notifyMatchFailure(
              contraction, "cannot fuse with the subsequent BiasAdd op");
        }
    
    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/jit/mark_for_compilation_pass.cc

    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/tests/canonicalize.mlir

      // CHECK: "tf.BiasAdd"(%arg0, %arg1) <{data_format = "NHWC"}> {device = "/job:localhost/replica:0/task:0/device:GPU:0"} : (tensor<*xf32>, tensor<128xf32>) -> tensor<*xf32>
      %0 = "tf.BiasAddV1"(%arg0, %arg1) {device = "/job:localhost/replica:0/task:0/device:GPU:0"} : (tensor<*xf32>, tensor<128xf32>) -> tensor<*xf32>
      func.return %0: tensor<*xf32>
    }
    
    // CHECK-LABEL: func @testLeakyRelu
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 132.1K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

      );
    
      TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<0>;
    
      let hasVerifier = 1;
    }
    
    def TF_BiasAddV1Op : TF_Op<"BiasAddV1", [Pure]> {
      let summary = "Adds `bias` to `value`.";
    
      let description = [{
    This is a deprecated version of BiasAdd and will be soon removed.
    
    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