a.smali

.class public Lsl/a;
.super Ljava/lang/Object;
.source "NormalizeOp.java"

# interfaces
.implements Lrl/e;


# instance fields
.field public final a:[F

.field public final b:[F

.field public final c:I

.field public final d:Z


# direct methods
.method static constructor <clinit>()V
    .registers 1

    return-void
.end method

.method public constructor <init>(FF)V
    .registers 9

    invoke-direct {p0}, Ljava/lang/Object;-><init>()V

    const/high16 v0, 0x3f800000    # 1.0f

    const/4 v1, 0x0

    cmpl-float v2, p1, v1

    if-nez v2, :cond_16

    cmpl-float v2, p2, v1

    if-eqz v2, :cond_14

    invoke-static {p2}, Ljava/lang/Float;->isInfinite(F)Z

    move-result v2

    if-eqz v2, :cond_16

    :cond_14
    const/high16 p2, 0x3f800000    # 1.0f

    :cond_16
    const/4 v2, 0x0

    const/4 v3, 0x1

    cmpl-float v4, p2, v1

    if-eqz v4, :cond_1e

    const/4 v4, 0x1

    goto :goto_1f

    :cond_1e
    const/4 v4, 0x0

    :goto_1f
    const-string v5, "Stddev cannot be zero."

    invoke-static {v4, v5}, Lrl/d;->a(ZLjava/lang/Object;)V

    cmpl-float v1, p1, v1

    if-nez v1, :cond_2e

    cmpl-float v0, p2, v0

    if-nez v0, :cond_2e

    const/4 v0, 0x1

    goto :goto_2f

    :cond_2e
    const/4 v0, 0x0

    :goto_2f
    iput-boolean v0, p0, Lsl/a;->d:Z

    new-array v0, v3, [F

    aput p1, v0, v2

    iput-object v0, p0, Lsl/a;->a:[F

    new-array p1, v3, [F

    aput p2, p1, v2

    iput-object p1, p0, Lsl/a;->b:[F

    iput v3, p0, Lsl/a;->c:I

    return-void
.end method


# virtual methods
.method public a(Lwl/a;)Lwl/a;
    .registers 9
    .param p1    # Lwl/a;
        .annotation build Landroidx/annotation/NonNull;
        .end annotation
    .end param
    .annotation build Landroidx/annotation/NonNull;
    .end annotation

    iget-boolean v0, p0, Lsl/a;->d:Z

    if-eqz v0, :cond_5

    return-object p1

    :cond_5
    invoke-virtual {p1}, Lwl/a;->f()[I

    move-result-object v0

    iget v1, p0, Lsl/a;->c:I

    const/4 v2, 0x0

    const/4 v3, 0x1

    if-eq v1, v3, :cond_1b

    array-length v4, v0

    if-eqz v4, :cond_19

    array-length v4, v0

    sub-int/2addr v4, v3

    aget v4, v0, v4

    if-ne v4, v1, :cond_19

    goto :goto_1b

    :cond_19
    const/4 v1, 0x0

    goto :goto_1c

    :cond_1b
    :goto_1b
    const/4 v1, 0x1

    :goto_1c
    const-string v4, "Number of means (stddevs) is not same with number of channels (size of last axis)."

    invoke-static {v1, v4}, Lrl/d;->a(ZLjava/lang/Object;)V

    invoke-virtual {p1}, Lwl/a;->d()[F

    move-result-object v1

    const/4 v4, 0x0

    :goto_26
    array-length v5, v1

    if-ge v2, v5, :cond_3e

    aget v5, v1, v2

    iget-object v6, p0, Lsl/a;->a:[F

    aget v6, v6, v4

    sub-float/2addr v5, v6

    iget-object v6, p0, Lsl/a;->b:[F

    aget v6, v6, v4

    div-float/2addr v5, v6

    aput v5, v1, v2

    add-int/2addr v4, v3

    iget v5, p0, Lsl/a;->c:I

    rem-int/2addr v4, v5

    add-int/lit8 v2, v2, 0x1

    goto :goto_26

    :cond_3e
    invoke-virtual {p1}, Lwl/a;->h()Z

    move-result p1

    if-eqz p1, :cond_4b

    sget-object p1, Lorg/tensorflow/lite/DataType;->FLOAT32:Lorg/tensorflow/lite/DataType;

    invoke-static {p1}, Lwl/a;->a(Lorg/tensorflow/lite/DataType;)Lwl/a;

    move-result-object p1

    goto :goto_51

    :cond_4b
    sget-object p1, Lorg/tensorflow/lite/DataType;->FLOAT32:Lorg/tensorflow/lite/DataType;

    invoke-static {v0, p1}, Lwl/a;->a([ILorg/tensorflow/lite/DataType;)Lwl/a;

    move-result-object p1

    :goto_51
    invoke-virtual {p1, v1, v0}, Lwl/a;->a([F[I)V

    return-object p1
.end method

.method public bridge synthetic apply(Ljava/lang/Object;)Ljava/lang/Object;
    .registers 2
    .param p1    # Ljava/lang/Object;
        .annotation build Landroidx/annotation/NonNull;
        .end annotation
    .end param
    .annotation build Landroidx/annotation/NonNull;
    .end annotation

    check-cast p1, Lwl/a;

    invoke-virtual {p0, p1}, Lsl/a;->a(Lwl/a;)Lwl/a;

    move-result-object p1

    return-object p1
.end method