Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for fused_batch_norm_v3 (0.25 sec)

  1. tensorflow/compiler/mlir/quantization/stablehlo/python/integration_test/quantize_model_test_base.py

            )
            if bias_fn is not None:
              out = nn_ops.bias_add(out, self.bias)
            if has_batch_norm:
              # Fusing is supported for non-training case.
              out, _, _, _, _, _ = nn_ops.fused_batch_norm_v3(
                  out, scale, offset, mean, variance, is_training=False
              )
            if activation_fn is not None:
              out = activation_fn(out)
            return {'output': out}
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 06:31:57 UTC 2024
    - 18.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test_base.py

            )
            if has_bias:
              out = nn_ops.bias_add(out, self.bias)
            if has_batch_norm:
              # Fusing is supported for non-training case.
              out, _, _, _, _, _ = nn_ops.fused_batch_norm_v3(
                  out, scale, offset, mean, variance, is_training=False
              )
            if activation_fn is not None:
              out = activation_fn(out)
            return {'output': out}
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 21 08:51:46 UTC 2024
    - 51.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc

        // Match
        auto fused_batch_norm_op =
            dyn_cast_or_null<::mlir::TF::FusedBatchNormV3Op>(fused_batch_norm);
        root = fused_batch_norm_op;
        x = fused_batch_norm_op.getODSOperands(0);
        scale = fused_batch_norm_op.getODSOperands(1);
        offset = fused_batch_norm_op.getODSOperands(2);
        mean = fused_batch_norm_op.getODSOperands(3);
        variance = fused_batch_norm_op.getODSOperands(4);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 64.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py

              # activation in this test.
              if has_batch_norm:
                # Fusing is supported for non-training case.
                out, _, _, _, _, _ = nn_ops.fused_batch_norm_v3(
                    out, scale, offset, mean, variance, is_training=False
                )
              out = activation_fn(out)
            out_min = array_ops.constant([-0.18, -0.32], dtype=dtypes.float32)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 03:36:50 UTC 2024
    - 235.6K bytes
    - Viewed (0)
Back to top