diff --git a/harness/pre-optimized-graphes/hey_snips_v4_model17/expected b/harness/pre-optimized-graphes/hey_snips_v4_model17/expected index 8ada746a78..7b5e5e184d 100644 --- a/harness/pre-optimized-graphes/hey_snips_v4_model17/expected +++ b/harness/pre-optimized-graphes/hey_snips_v4_model17/expected @@ -32,10 +32,10 @@ graph network(input_node) -> (i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolutio i"wavenet_2/dilation_layer_0-dilation_rate_1-gate-conv1d/convolution/Conv2D" = squeeze(transpose(i"wavenet_2/dilation_layer_0-dilation_rate_1-gate-conv1d/convolution/Conv2D_conv", axes = [0, 2, 1]), axes = [0]); i"wavenet_2/dilation_layer_0-dilation_rate_1-gate-conv1d/Sigmoid" = sigmoid(i"wavenet_2/dilation_layer_0-dilation_rate_1-gate-conv1d/convolution/Conv2D"); i"wavenet_2/mul" = mul(i"wavenet_2/dilation_layer_0-dilation_rate_1-filter-conv1d/Tanh", i"wavenet_2/dilation_layer_0-dilation_rate_1-gate-conv1d/Sigmoid"); - i"wavenet_2/dilation_layer_0-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_0-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters", shape = [64, 32]); - i"wavenet_2/dilation_layer_0-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul", i"wavenet_2/dilation_layer_0-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); - i"wavenet_2/dilation_layer_0-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_0-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters", shape = [64, 16]); - i"wavenet_2/dilation_layer_0-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul", i"wavenet_2/dilation_layer_0-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_0-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_0-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [32, 64]); + i"wavenet_2/dilation_layer_0-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul", i"wavenet_2/dilation_layer_0-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); + i"wavenet_2/dilation_layer_0-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_0-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [16, 64]); + i"wavenet_2/dilation_layer_0-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul", i"wavenet_2/dilation_layer_0-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/add" = add(i"wavenet_2/dilation_layer_0-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.einsum", i"wavenet_2/pre_conv-conv1d/convolution/Conv2D"); i"wavenet_2/dilation_layer_1-dilation_rate_2-filter-conv1d/convolution/Conv2D.delay" = tract_pulse_delay(i"wavenet_2/add", axis = 0, delay = 0, overlap = 4); i"wavenet_2/dilation_layer_1-dilation_rate_2-filter-conv1d/convolution/Conv2D_input" = transpose(unsqueeze(i"wavenet_2/dilation_layer_1-dilation_rate_2-filter-conv1d/convolution/Conv2D.delay", axes = [0]), axes = [0, 2, 1]); @@ -49,11 +49,11 @@ graph network(input_node) -> (i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolutio i"wavenet_2/dilation_layer_1-dilation_rate_2-gate-conv1d/convolution/Conv2D" = squeeze(transpose(i"wavenet_2/dilation_layer_1-dilation_rate_2-gate-conv1d/convolution/Conv2D_conv", axes = [0, 2, 1]), axes = [0]); i"wavenet_2/dilation_layer_1-dilation_rate_2-gate-conv1d/Sigmoid" = sigmoid(i"wavenet_2/dilation_layer_1-dilation_rate_2-gate-conv1d/convolution/Conv2D"); i"wavenet_2/mul_1" = mul(i"wavenet_2/dilation_layer_1-dilation_rate_2-filter-conv1d/Tanh", i"wavenet_2/dilation_layer_1-dilation_rate_2-gate-conv1d/Sigmoid"); - i"wavenet_2/dilation_layer_1-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_1-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters", shape = [64, 32]); - i"wavenet_2/dilation_layer_1-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_1", i"wavenet_2/dilation_layer_1-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_1-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_1-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [32, 64]); + i"wavenet_2/dilation_layer_1-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_1", i"wavenet_2/dilation_layer_1-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/AddN.0" = add(i"wavenet_2/dilation_layer_0-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.einsum", i"wavenet_2/dilation_layer_1-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.einsum"); - i"wavenet_2/dilation_layer_1-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_1-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters", shape = [64, 16]); - i"wavenet_2/dilation_layer_1-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_1", i"wavenet_2/dilation_layer_1-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_1-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_1-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [16, 64]); + i"wavenet_2/dilation_layer_1-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_1", i"wavenet_2/dilation_layer_1-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/add_1" = add(i"wavenet_2/dilation_layer_1-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.einsum", i"wavenet_2/add"); i"wavenet_2/dilation_layer_2-dilation_rate_4-filter-conv1d/convolution/Conv2D.delay" = tract_pulse_delay(i"wavenet_2/add_1", axis = 0, delay = 0, overlap = 8); i"wavenet_2/dilation_layer_2-dilation_rate_4-filter-conv1d/convolution/Conv2D_input" = transpose(unsqueeze(i"wavenet_2/dilation_layer_2-dilation_rate_4-filter-conv1d/convolution/Conv2D.delay", axes = [0]), axes = [0, 2, 1]); @@ -67,11 +67,11 @@ graph network(input_node) -> (i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolutio i"wavenet_2/dilation_layer_2-dilation_rate_4-gate-conv1d/convolution/Conv2D" = squeeze(transpose(i"wavenet_2/dilation_layer_2-dilation_rate_4-gate-conv1d/convolution/Conv2D_conv", axes = [0, 2, 1]), axes = [0]); i"wavenet_2/dilation_layer_2-dilation_rate_4-gate-conv1d/Sigmoid" = sigmoid(i"wavenet_2/dilation_layer_2-dilation_rate_4-gate-conv1d/convolution/Conv2D"); i"wavenet_2/mul_2" = mul(i"wavenet_2/dilation_layer_2-dilation_rate_4-filter-conv1d/Tanh", i"wavenet_2/dilation_layer_2-dilation_rate_4-gate-conv1d/Sigmoid"); - i"wavenet_2/dilation_layer_2-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_2-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters", shape = [64, 32]); - i"wavenet_2/dilation_layer_2-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_2", i"wavenet_2/dilation_layer_2-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_2-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_2-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [32, 64]); + i"wavenet_2/dilation_layer_2-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_2", i"wavenet_2/dilation_layer_2-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/AddN.1" = add(i"wavenet_2/AddN.0", i"wavenet_2/dilation_layer_2-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.einsum"); - i"wavenet_2/dilation_layer_2-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_2-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters", shape = [64, 16]); - i"wavenet_2/dilation_layer_2-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_2", i"wavenet_2/dilation_layer_2-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_2-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_2-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [16, 64]); + i"wavenet_2/dilation_layer_2-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_2", i"wavenet_2/dilation_layer_2-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/add_2" = add(i"wavenet_2/dilation_layer_2-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.einsum", i"wavenet_2/add_1"); i"wavenet_2/dilation_layer_3-dilation_rate_8-filter-conv1d/convolution/Conv2D.delay" = tract_pulse_delay(i"wavenet_2/add_2", axis = 0, delay = 0, overlap = 16); i"wavenet_2/dilation_layer_3-dilation_rate_8-filter-conv1d/convolution/Conv2D_input" = transpose(unsqueeze(i"wavenet_2/dilation_layer_3-dilation_rate_8-filter-conv1d/convolution/Conv2D.delay", axes = [0]), axes = [0, 2, 1]); @@ -85,11 +85,11 @@ graph network(input_node) -> (i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolutio i"wavenet_2/dilation_layer_3-dilation_rate_8-gate-conv1d/convolution/Conv2D" = squeeze(transpose(i"wavenet_2/dilation_layer_3-dilation_rate_8-gate-conv1d/convolution/Conv2D_conv", axes = [0, 2, 1]), axes = [0]); i"wavenet_2/dilation_layer_3-dilation_rate_8-gate-conv1d/Sigmoid" = sigmoid(i"wavenet_2/dilation_layer_3-dilation_rate_8-gate-conv1d/convolution/Conv2D"); i"wavenet_2/mul_3" = mul(i"wavenet_2/dilation_layer_3-dilation_rate_8-filter-conv1d/Tanh", i"wavenet_2/dilation_layer_3-dilation_rate_8-gate-conv1d/Sigmoid"); - i"wavenet_2/dilation_layer_3-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_3-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters", shape = [64, 32]); - i"wavenet_2/dilation_layer_3-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_3", i"wavenet_2/dilation_layer_3-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_3-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_3-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [32, 64]); + i"wavenet_2/dilation_layer_3-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_3", i"wavenet_2/dilation_layer_3-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/AddN.2" = add(i"wavenet_2/AddN.1", i"wavenet_2/dilation_layer_3-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.einsum"); - i"wavenet_2/dilation_layer_3-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_3-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters", shape = [64, 16]); - i"wavenet_2/dilation_layer_3-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_3", i"wavenet_2/dilation_layer_3-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_3-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_3-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [16, 64]); + i"wavenet_2/dilation_layer_3-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_3", i"wavenet_2/dilation_layer_3-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/add_3" = add(i"wavenet_2/dilation_layer_3-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.einsum", i"wavenet_2/add_2"); i"wavenet_2/dilation_layer_4-dilation_rate_1-filter-conv1d/convolution/Conv2D.delay" = tract_pulse_delay(i"wavenet_2/add_3", axis = 0, delay = 0, overlap = 2); i"wavenet_2/dilation_layer_4-dilation_rate_1-filter-conv1d/convolution/Conv2D_input" = transpose(unsqueeze(i"wavenet_2/dilation_layer_4-dilation_rate_1-filter-conv1d/convolution/Conv2D.delay", axes = [0]), axes = [0, 2, 1]); @@ -103,11 +103,11 @@ graph network(input_node) -> (i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolutio i"wavenet_2/dilation_layer_4-dilation_rate_1-gate-conv1d/convolution/Conv2D" = squeeze(transpose(i"wavenet_2/dilation_layer_4-dilation_rate_1-gate-conv1d/convolution/Conv2D_conv", axes = [0, 2, 1]), axes = [0]); i"wavenet_2/dilation_layer_4-dilation_rate_1-gate-conv1d/Sigmoid" = sigmoid(i"wavenet_2/dilation_layer_4-dilation_rate_1-gate-conv1d/convolution/Conv2D"); i"wavenet_2/mul_4" = mul(i"wavenet_2/dilation_layer_4-dilation_rate_1-filter-conv1d/Tanh", i"wavenet_2/dilation_layer_4-dilation_rate_1-gate-conv1d/Sigmoid"); - i"wavenet_2/dilation_layer_4-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_4-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters", shape = [64, 32]); - i"wavenet_2/dilation_layer_4-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_4", i"wavenet_2/dilation_layer_4-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_4-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_4-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [32, 64]); + i"wavenet_2/dilation_layer_4-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_4", i"wavenet_2/dilation_layer_4-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/AddN.3" = add(i"wavenet_2/AddN.2", i"wavenet_2/dilation_layer_4-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.einsum"); - i"wavenet_2/dilation_layer_4-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_4-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters", shape = [64, 16]); - i"wavenet_2/dilation_layer_4-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_4", i"wavenet_2/dilation_layer_4-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_4-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_4-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [16, 64]); + i"wavenet_2/dilation_layer_4-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_4", i"wavenet_2/dilation_layer_4-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/add_4" = add(i"wavenet_2/dilation_layer_4-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.einsum", i"wavenet_2/add_3"); i"wavenet_2/dilation_layer_5-dilation_rate_2-filter-conv1d/convolution/Conv2D.delay" = tract_pulse_delay(i"wavenet_2/add_4", axis = 0, delay = 0, overlap = 4); i"wavenet_2/dilation_layer_5-dilation_rate_2-filter-conv1d/convolution/Conv2D_input" = transpose(unsqueeze(i"wavenet_2/dilation_layer_5-dilation_rate_2-filter-conv1d/convolution/Conv2D.delay", axes = [0]), axes = [0, 2, 1]); @@ -121,11 +121,11 @@ graph network(input_node) -> (i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolutio i"wavenet_2/dilation_layer_5-dilation_rate_2-gate-conv1d/convolution/Conv2D" = squeeze(transpose(i"wavenet_2/dilation_layer_5-dilation_rate_2-gate-conv1d/convolution/Conv2D_conv", axes = [0, 2, 1]), axes = [0]); i"wavenet_2/dilation_layer_5-dilation_rate_2-gate-conv1d/Sigmoid" = sigmoid(i"wavenet_2/dilation_layer_5-dilation_rate_2-gate-conv1d/convolution/Conv2D"); i"wavenet_2/mul_5" = mul(i"wavenet_2/dilation_layer_5-dilation_rate_2-filter-conv1d/Tanh", i"wavenet_2/dilation_layer_5-dilation_rate_2-gate-conv1d/Sigmoid"); - i"wavenet_2/dilation_layer_5-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_5-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters", shape = [64, 32]); - i"wavenet_2/dilation_layer_5-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_5", i"wavenet_2/dilation_layer_5-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_5-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_5-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [32, 64]); + i"wavenet_2/dilation_layer_5-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_5", i"wavenet_2/dilation_layer_5-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/AddN.4" = add(i"wavenet_2/AddN.3", i"wavenet_2/dilation_layer_5-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.einsum"); - i"wavenet_2/dilation_layer_5-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_5-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters", shape = [64, 16]); - i"wavenet_2/dilation_layer_5-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_5", i"wavenet_2/dilation_layer_5-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_5-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_5-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [16, 64]); + i"wavenet_2/dilation_layer_5-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_5", i"wavenet_2/dilation_layer_5-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/add_5" = add(i"wavenet_2/dilation_layer_5-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.einsum", i"wavenet_2/add_4"); i"wavenet_2/dilation_layer_6-dilation_rate_4-filter-conv1d/convolution/Conv2D.delay" = tract_pulse_delay(i"wavenet_2/add_5", axis = 0, delay = 0, overlap = 8); i"wavenet_2/dilation_layer_6-dilation_rate_4-filter-conv1d/convolution/Conv2D_input" = transpose(unsqueeze(i"wavenet_2/dilation_layer_6-dilation_rate_4-filter-conv1d/convolution/Conv2D.delay", axes = [0]), axes = [0, 2, 1]); @@ -139,11 +139,11 @@ graph network(input_node) -> (i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolutio i"wavenet_2/dilation_layer_6-dilation_rate_4-gate-conv1d/convolution/Conv2D" = squeeze(transpose(i"wavenet_2/dilation_layer_6-dilation_rate_4-gate-conv1d/convolution/Conv2D_conv", axes = [0, 2, 1]), axes = [0]); i"wavenet_2/dilation_layer_6-dilation_rate_4-gate-conv1d/Sigmoid" = sigmoid(i"wavenet_2/dilation_layer_6-dilation_rate_4-gate-conv1d/convolution/Conv2D"); i"wavenet_2/mul_6" = mul(i"wavenet_2/dilation_layer_6-dilation_rate_4-filter-conv1d/Tanh", i"wavenet_2/dilation_layer_6-dilation_rate_4-gate-conv1d/Sigmoid"); - i"wavenet_2/dilation_layer_6-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_6-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters", shape = [64, 32]); - i"wavenet_2/dilation_layer_6-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_6", i"wavenet_2/dilation_layer_6-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_6-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_6-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [32, 64]); + i"wavenet_2/dilation_layer_6-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_6", i"wavenet_2/dilation_layer_6-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/AddN.5" = add(i"wavenet_2/AddN.4", i"wavenet_2/dilation_layer_6-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.einsum"); - i"wavenet_2/dilation_layer_6-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_6-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters", shape = [64, 16]); - i"wavenet_2/dilation_layer_6-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_6", i"wavenet_2/dilation_layer_6-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_6-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_6-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [16, 64]); + i"wavenet_2/dilation_layer_6-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_6", i"wavenet_2/dilation_layer_6-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/add_6" = add(i"wavenet_2/dilation_layer_6-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.einsum", i"wavenet_2/add_5"); i"wavenet_2/dilation_layer_7-dilation_rate_8-filter-conv1d/convolution/Conv2D.delay" = tract_pulse_delay(i"wavenet_2/add_6", axis = 0, delay = 0, overlap = 16); i"wavenet_2/dilation_layer_7-dilation_rate_8-filter-conv1d/convolution/Conv2D_input" = transpose(unsqueeze(i"wavenet_2/dilation_layer_7-dilation_rate_8-filter-conv1d/convolution/Conv2D.delay", axes = [0]), axes = [0, 2, 1]); @@ -157,11 +157,11 @@ graph network(input_node) -> (i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolutio i"wavenet_2/dilation_layer_7-dilation_rate_8-gate-conv1d/convolution/Conv2D" = squeeze(transpose(i"wavenet_2/dilation_layer_7-dilation_rate_8-gate-conv1d/convolution/Conv2D_conv", axes = [0, 2, 1]), axes = [0]); i"wavenet_2/dilation_layer_7-dilation_rate_8-gate-conv1d/Sigmoid" = sigmoid(i"wavenet_2/dilation_layer_7-dilation_rate_8-gate-conv1d/convolution/Conv2D"); i"wavenet_2/mul_7" = mul(i"wavenet_2/dilation_layer_7-dilation_rate_8-filter-conv1d/Tanh", i"wavenet_2/dilation_layer_7-dilation_rate_8-gate-conv1d/Sigmoid"); - i"wavenet_2/dilation_layer_7-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_7-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters", shape = [64, 32]); - i"wavenet_2/dilation_layer_7-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_7", i"wavenet_2/dilation_layer_7-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_7-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_7-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [32, 64]); + i"wavenet_2/dilation_layer_7-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_7", i"wavenet_2/dilation_layer_7-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/AddN.6" = add(i"wavenet_2/AddN.5", i"wavenet_2/dilation_layer_7-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.einsum"); - i"wavenet_2/dilation_layer_7-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_7-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters", shape = [64, 16]); - i"wavenet_2/dilation_layer_7-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_7", i"wavenet_2/dilation_layer_7-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_7-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_7-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [16, 64]); + i"wavenet_2/dilation_layer_7-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_7", i"wavenet_2/dilation_layer_7-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/add_7" = add(i"wavenet_2/dilation_layer_7-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.einsum", i"wavenet_2/add_6"); i"wavenet_2/dilation_layer_8-dilation_rate_1-filter-conv1d/convolution/Conv2D.delay" = tract_pulse_delay(i"wavenet_2/add_7", axis = 0, delay = 0, overlap = 2); i"wavenet_2/dilation_layer_8-dilation_rate_1-filter-conv1d/convolution/Conv2D_input" = transpose(unsqueeze(i"wavenet_2/dilation_layer_8-dilation_rate_1-filter-conv1d/convolution/Conv2D.delay", axes = [0]), axes = [0, 2, 1]); @@ -175,11 +175,11 @@ graph network(input_node) -> (i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolutio i"wavenet_2/dilation_layer_8-dilation_rate_1-gate-conv1d/convolution/Conv2D" = squeeze(transpose(i"wavenet_2/dilation_layer_8-dilation_rate_1-gate-conv1d/convolution/Conv2D_conv", axes = [0, 2, 1]), axes = [0]); i"wavenet_2/dilation_layer_8-dilation_rate_1-gate-conv1d/Sigmoid" = sigmoid(i"wavenet_2/dilation_layer_8-dilation_rate_1-gate-conv1d/convolution/Conv2D"); i"wavenet_2/mul_8" = mul(i"wavenet_2/dilation_layer_8-dilation_rate_1-filter-conv1d/Tanh", i"wavenet_2/dilation_layer_8-dilation_rate_1-gate-conv1d/Sigmoid"); - i"wavenet_2/dilation_layer_8-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_8-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters", shape = [64, 32]); - i"wavenet_2/dilation_layer_8-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_8", i"wavenet_2/dilation_layer_8-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_8-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_8-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [32, 64]); + i"wavenet_2/dilation_layer_8-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_8", i"wavenet_2/dilation_layer_8-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/AddN.7" = add(i"wavenet_2/AddN.6", i"wavenet_2/dilation_layer_8-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.einsum"); - i"wavenet_2/dilation_layer_8-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_8-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters", shape = [64, 16]); - i"wavenet_2/dilation_layer_8-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_8", i"wavenet_2/dilation_layer_8-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_8-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_8-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [16, 64]); + i"wavenet_2/dilation_layer_8-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_8", i"wavenet_2/dilation_layer_8-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/add_8" = add(i"wavenet_2/dilation_layer_8-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.einsum", i"wavenet_2/add_7"); i"wavenet_2/dilation_layer_9-dilation_rate_2-filter-conv1d/convolution/Conv2D.delay" = tract_pulse_delay(i"wavenet_2/add_8", axis = 0, delay = 0, overlap = 4); i"wavenet_2/dilation_layer_9-dilation_rate_2-filter-conv1d/convolution/Conv2D_input" = transpose(unsqueeze(i"wavenet_2/dilation_layer_9-dilation_rate_2-filter-conv1d/convolution/Conv2D.delay", axes = [0]), axes = [0, 2, 1]); @@ -193,11 +193,11 @@ graph network(input_node) -> (i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolutio i"wavenet_2/dilation_layer_9-dilation_rate_2-gate-conv1d/convolution/Conv2D" = squeeze(transpose(i"wavenet_2/dilation_layer_9-dilation_rate_2-gate-conv1d/convolution/Conv2D_conv", axes = [0, 2, 1]), axes = [0]); i"wavenet_2/dilation_layer_9-dilation_rate_2-gate-conv1d/Sigmoid" = sigmoid(i"wavenet_2/dilation_layer_9-dilation_rate_2-gate-conv1d/convolution/Conv2D"); i"wavenet_2/mul_9" = mul(i"wavenet_2/dilation_layer_9-dilation_rate_2-filter-conv1d/Tanh", i"wavenet_2/dilation_layer_9-dilation_rate_2-gate-conv1d/Sigmoid"); - i"wavenet_2/dilation_layer_9-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_9-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters", shape = [64, 32]); - i"wavenet_2/dilation_layer_9-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_9", i"wavenet_2/dilation_layer_9-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_9-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_9-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [32, 64]); + i"wavenet_2/dilation_layer_9-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_9", i"wavenet_2/dilation_layer_9-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/AddN.8" = add(i"wavenet_2/AddN.7", i"wavenet_2/dilation_layer_9-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.einsum"); - i"wavenet_2/dilation_layer_9-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_9-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters", shape = [64, 16]); - i"wavenet_2/dilation_layer_9-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_9", i"wavenet_2/dilation_layer_9-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_9-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_9-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [16, 64]); + i"wavenet_2/dilation_layer_9-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_9", i"wavenet_2/dilation_layer_9-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/add_9" = add(i"wavenet_2/dilation_layer_9-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.einsum", i"wavenet_2/add_8"); i"wavenet_2/dilation_layer_10-dilation_rate_4-filter-conv1d/convolution/Conv2D.delay" = tract_pulse_delay(i"wavenet_2/add_9", axis = 0, delay = 0, overlap = 8); i"wavenet_2/dilation_layer_10-dilation_rate_4-filter-conv1d/convolution/Conv2D_input" = transpose(unsqueeze(i"wavenet_2/dilation_layer_10-dilation_rate_4-filter-conv1d/convolution/Conv2D.delay", axes = [0]), axes = [0, 2, 1]); @@ -211,11 +211,11 @@ graph network(input_node) -> (i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolutio i"wavenet_2/dilation_layer_10-dilation_rate_4-gate-conv1d/convolution/Conv2D" = squeeze(transpose(i"wavenet_2/dilation_layer_10-dilation_rate_4-gate-conv1d/convolution/Conv2D_conv", axes = [0, 2, 1]), axes = [0]); i"wavenet_2/dilation_layer_10-dilation_rate_4-gate-conv1d/Sigmoid" = sigmoid(i"wavenet_2/dilation_layer_10-dilation_rate_4-gate-conv1d/convolution/Conv2D"); i"wavenet_2/mul_10" = mul(i"wavenet_2/dilation_layer_10-dilation_rate_4-filter-conv1d/Tanh", i"wavenet_2/dilation_layer_10-dilation_rate_4-gate-conv1d/Sigmoid"); - i"wavenet_2/dilation_layer_10-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_10-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters", shape = [64, 32]); - i"wavenet_2/dilation_layer_10-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_10", i"wavenet_2/dilation_layer_10-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_10-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_10-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [32, 64]); + i"wavenet_2/dilation_layer_10-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_10", i"wavenet_2/dilation_layer_10-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/AddN.9" = add(i"wavenet_2/AddN.8", i"wavenet_2/dilation_layer_10-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.einsum"); - i"wavenet_2/dilation_layer_10-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_10-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters", shape = [64, 16]); - i"wavenet_2/dilation_layer_10-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_10", i"wavenet_2/dilation_layer_10-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_10-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_10-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [16, 64]); + i"wavenet_2/dilation_layer_10-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_10", i"wavenet_2/dilation_layer_10-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/add_10" = add(i"wavenet_2/dilation_layer_10-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.einsum", i"wavenet_2/add_9"); i"wavenet_2/dilation_layer_11-dilation_rate_8-filter-conv1d/convolution/Conv2D.delay" = tract_pulse_delay(i"wavenet_2/add_10", axis = 0, delay = 0, overlap = 16); i"wavenet_2/dilation_layer_11-dilation_rate_8-filter-conv1d/convolution/Conv2D_input" = transpose(unsqueeze(i"wavenet_2/dilation_layer_11-dilation_rate_8-filter-conv1d/convolution/Conv2D.delay", axes = [0]), axes = [0, 2, 1]); @@ -229,11 +229,11 @@ graph network(input_node) -> (i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolutio i"wavenet_2/dilation_layer_11-dilation_rate_8-gate-conv1d/convolution/Conv2D" = squeeze(transpose(i"wavenet_2/dilation_layer_11-dilation_rate_8-gate-conv1d/convolution/Conv2D_conv", axes = [0, 2, 1]), axes = [0]); i"wavenet_2/dilation_layer_11-dilation_rate_8-gate-conv1d/Sigmoid" = sigmoid(i"wavenet_2/dilation_layer_11-dilation_rate_8-gate-conv1d/convolution/Conv2D"); i"wavenet_2/mul_11" = mul(i"wavenet_2/dilation_layer_11-dilation_rate_8-filter-conv1d/Tanh", i"wavenet_2/dilation_layer_11-dilation_rate_8-gate-conv1d/Sigmoid"); - i"wavenet_2/dilation_layer_11-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_11-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters", shape = [64, 32]); - i"wavenet_2/dilation_layer_11-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_11", i"wavenet_2/dilation_layer_11-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_11-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_11-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [32, 64]); + i"wavenet_2/dilation_layer_11-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_11", i"wavenet_2/dilation_layer_11-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/AddN.10" = add(i"wavenet_2/AddN.9", i"wavenet_2/dilation_layer_11-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.einsum"); - i"wavenet_2/dilation_layer_11-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_11-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters", shape = [64, 16]); - i"wavenet_2/dilation_layer_11-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_11", i"wavenet_2/dilation_layer_11-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_11-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_11-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [16, 64]); + i"wavenet_2/dilation_layer_11-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_11", i"wavenet_2/dilation_layer_11-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/add_11" = add(i"wavenet_2/dilation_layer_11-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.einsum", i"wavenet_2/add_10"); i"wavenet_2/dilation_layer_12-dilation_rate_1-filter-conv1d/convolution/Conv2D.delay" = tract_pulse_delay(i"wavenet_2/add_11", axis = 0, delay = 0, overlap = 2); i"wavenet_2/dilation_layer_12-dilation_rate_1-filter-conv1d/convolution/Conv2D_input" = transpose(unsqueeze(i"wavenet_2/dilation_layer_12-dilation_rate_1-filter-conv1d/convolution/Conv2D.delay", axes = [0]), axes = [0, 2, 1]); @@ -247,11 +247,11 @@ graph network(input_node) -> (i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolutio i"wavenet_2/dilation_layer_12-dilation_rate_1-gate-conv1d/convolution/Conv2D" = squeeze(transpose(i"wavenet_2/dilation_layer_12-dilation_rate_1-gate-conv1d/convolution/Conv2D_conv", axes = [0, 2, 1]), axes = [0]); i"wavenet_2/dilation_layer_12-dilation_rate_1-gate-conv1d/Sigmoid" = sigmoid(i"wavenet_2/dilation_layer_12-dilation_rate_1-gate-conv1d/convolution/Conv2D"); i"wavenet_2/mul_12" = mul(i"wavenet_2/dilation_layer_12-dilation_rate_1-filter-conv1d/Tanh", i"wavenet_2/dilation_layer_12-dilation_rate_1-gate-conv1d/Sigmoid"); - i"wavenet_2/dilation_layer_12-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_12-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters", shape = [64, 32]); - i"wavenet_2/dilation_layer_12-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_12", i"wavenet_2/dilation_layer_12-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_12-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_12-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [32, 64]); + i"wavenet_2/dilation_layer_12-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_12", i"wavenet_2/dilation_layer_12-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/AddN.11" = add(i"wavenet_2/AddN.10", i"wavenet_2/dilation_layer_12-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.einsum"); - i"wavenet_2/dilation_layer_12-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_12-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters", shape = [64, 16]); - i"wavenet_2/dilation_layer_12-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_12", i"wavenet_2/dilation_layer_12-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_12-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_12-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [16, 64]); + i"wavenet_2/dilation_layer_12-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_12", i"wavenet_2/dilation_layer_12-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/add_12" = add(i"wavenet_2/dilation_layer_12-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.einsum", i"wavenet_2/add_11"); i"wavenet_2/dilation_layer_13-dilation_rate_2-filter-conv1d/convolution/Conv2D.delay" = tract_pulse_delay(i"wavenet_2/add_12", axis = 0, delay = 0, overlap = 4); i"wavenet_2/dilation_layer_13-dilation_rate_2-filter-conv1d/convolution/Conv2D_input" = transpose(unsqueeze(i"wavenet_2/dilation_layer_13-dilation_rate_2-filter-conv1d/convolution/Conv2D.delay", axes = [0]), axes = [0, 2, 1]); @@ -265,11 +265,11 @@ graph network(input_node) -> (i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolutio i"wavenet_2/dilation_layer_13-dilation_rate_2-gate-conv1d/convolution/Conv2D" = squeeze(transpose(i"wavenet_2/dilation_layer_13-dilation_rate_2-gate-conv1d/convolution/Conv2D_conv", axes = [0, 2, 1]), axes = [0]); i"wavenet_2/dilation_layer_13-dilation_rate_2-gate-conv1d/Sigmoid" = sigmoid(i"wavenet_2/dilation_layer_13-dilation_rate_2-gate-conv1d/convolution/Conv2D"); i"wavenet_2/mul_13" = mul(i"wavenet_2/dilation_layer_13-dilation_rate_2-filter-conv1d/Tanh", i"wavenet_2/dilation_layer_13-dilation_rate_2-gate-conv1d/Sigmoid"); - i"wavenet_2/dilation_layer_13-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_13-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters", shape = [64, 32]); - i"wavenet_2/dilation_layer_13-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_13", i"wavenet_2/dilation_layer_13-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_13-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_13-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [32, 64]); + i"wavenet_2/dilation_layer_13-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_13", i"wavenet_2/dilation_layer_13-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/AddN.12" = add(i"wavenet_2/AddN.11", i"wavenet_2/dilation_layer_13-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.einsum"); - i"wavenet_2/dilation_layer_13-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_13-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters", shape = [64, 16]); - i"wavenet_2/dilation_layer_13-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_13", i"wavenet_2/dilation_layer_13-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_13-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_13-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [16, 64]); + i"wavenet_2/dilation_layer_13-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_13", i"wavenet_2/dilation_layer_13-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/add_13" = add(i"wavenet_2/dilation_layer_13-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.einsum", i"wavenet_2/add_12"); i"wavenet_2/dilation_layer_14-dilation_rate_4-filter-conv1d/convolution/Conv2D.delay" = tract_pulse_delay(i"wavenet_2/add_13", axis = 0, delay = 0, overlap = 8); i"wavenet_2/dilation_layer_14-dilation_rate_4-filter-conv1d/convolution/Conv2D_input" = transpose(unsqueeze(i"wavenet_2/dilation_layer_14-dilation_rate_4-filter-conv1d/convolution/Conv2D.delay", axes = [0]), axes = [0, 2, 1]); @@ -283,11 +283,11 @@ graph network(input_node) -> (i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolutio i"wavenet_2/dilation_layer_14-dilation_rate_4-gate-conv1d/convolution/Conv2D" = squeeze(transpose(i"wavenet_2/dilation_layer_14-dilation_rate_4-gate-conv1d/convolution/Conv2D_conv", axes = [0, 2, 1]), axes = [0]); i"wavenet_2/dilation_layer_14-dilation_rate_4-gate-conv1d/Sigmoid" = sigmoid(i"wavenet_2/dilation_layer_14-dilation_rate_4-gate-conv1d/convolution/Conv2D"); i"wavenet_2/mul_14" = mul(i"wavenet_2/dilation_layer_14-dilation_rate_4-filter-conv1d/Tanh", i"wavenet_2/dilation_layer_14-dilation_rate_4-gate-conv1d/Sigmoid"); - i"wavenet_2/dilation_layer_14-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_14-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters", shape = [64, 32]); - i"wavenet_2/dilation_layer_14-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_14", i"wavenet_2/dilation_layer_14-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_14-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_14-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [32, 64]); + i"wavenet_2/dilation_layer_14-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_14", i"wavenet_2/dilation_layer_14-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/AddN.13" = add(i"wavenet_2/AddN.12", i"wavenet_2/dilation_layer_14-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.einsum"); - i"wavenet_2/dilation_layer_14-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_14-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters", shape = [64, 16]); - i"wavenet_2/dilation_layer_14-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_14", i"wavenet_2/dilation_layer_14-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_14-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_14-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [16, 64]); + i"wavenet_2/dilation_layer_14-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_14", i"wavenet_2/dilation_layer_14-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/add_14" = add(i"wavenet_2/dilation_layer_14-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.einsum", i"wavenet_2/add_13"); i"wavenet_2/dilation_layer_15-dilation_rate_8-filter-conv1d/convolution/Conv2D.delay" = tract_pulse_delay(i"wavenet_2/add_14", axis = 0, delay = 0, overlap = 16); i"wavenet_2/dilation_layer_15-dilation_rate_8-filter-conv1d/convolution/Conv2D_input" = transpose(unsqueeze(i"wavenet_2/dilation_layer_15-dilation_rate_8-filter-conv1d/convolution/Conv2D.delay", axes = [0]), axes = [0, 2, 1]); @@ -301,11 +301,11 @@ graph network(input_node) -> (i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolutio i"wavenet_2/dilation_layer_15-dilation_rate_8-gate-conv1d/convolution/Conv2D" = squeeze(transpose(i"wavenet_2/dilation_layer_15-dilation_rate_8-gate-conv1d/convolution/Conv2D_conv", axes = [0, 2, 1]), axes = [0]); i"wavenet_2/dilation_layer_15-dilation_rate_8-gate-conv1d/Sigmoid" = sigmoid(i"wavenet_2/dilation_layer_15-dilation_rate_8-gate-conv1d/convolution/Conv2D"); i"wavenet_2/mul_15" = mul(i"wavenet_2/dilation_layer_15-dilation_rate_8-filter-conv1d/Tanh", i"wavenet_2/dilation_layer_15-dilation_rate_8-gate-conv1d/Sigmoid"); - i"wavenet_2/dilation_layer_15-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_15-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters", shape = [64, 32]); - i"wavenet_2/dilation_layer_15-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_15", i"wavenet_2/dilation_layer_15-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_15-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_15-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [32, 64]); + i"wavenet_2/dilation_layer_15-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_15", i"wavenet_2/dilation_layer_15-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/AddN.14" = add(i"wavenet_2/AddN.13", i"wavenet_2/dilation_layer_15-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.einsum"); - i"wavenet_2/dilation_layer_15-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_15-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters", shape = [64, 16]); - i"wavenet_2/dilation_layer_15-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_15", i"wavenet_2/dilation_layer_15-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_15-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_15-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [16, 64]); + i"wavenet_2/dilation_layer_15-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_15", i"wavenet_2/dilation_layer_15-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/add_15" = add(i"wavenet_2/dilation_layer_15-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.einsum", i"wavenet_2/add_14"); i"wavenet_2/dilation_layer_16-dilation_rate_1-filter-conv1d/convolution/Conv2D.delay" = tract_pulse_delay(i"wavenet_2/add_15", axis = 0, delay = 0, overlap = 2); i"wavenet_2/dilation_layer_16-dilation_rate_1-filter-conv1d/convolution/Conv2D_input" = transpose(unsqueeze(i"wavenet_2/dilation_layer_16-dilation_rate_1-filter-conv1d/convolution/Conv2D.delay", axes = [0]), axes = [0, 2, 1]); @@ -319,11 +319,11 @@ graph network(input_node) -> (i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolutio i"wavenet_2/dilation_layer_16-dilation_rate_1-gate-conv1d/convolution/Conv2D" = squeeze(transpose(i"wavenet_2/dilation_layer_16-dilation_rate_1-gate-conv1d/convolution/Conv2D_conv", axes = [0, 2, 1]), axes = [0]); i"wavenet_2/dilation_layer_16-dilation_rate_1-gate-conv1d/Sigmoid" = sigmoid(i"wavenet_2/dilation_layer_16-dilation_rate_1-gate-conv1d/convolution/Conv2D"); i"wavenet_2/mul_16" = mul(i"wavenet_2/dilation_layer_16-dilation_rate_1-filter-conv1d/Tanh", i"wavenet_2/dilation_layer_16-dilation_rate_1-gate-conv1d/Sigmoid"); - i"wavenet_2/dilation_layer_16-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_16-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters", shape = [64, 32]); - i"wavenet_2/dilation_layer_16-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_16", i"wavenet_2/dilation_layer_16-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_16-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_16-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [32, 64]); + i"wavenet_2/dilation_layer_16-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_16", i"wavenet_2/dilation_layer_16-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/AddN.15" = add(i"wavenet_2/AddN.14", i"wavenet_2/dilation_layer_16-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.einsum"); - i"wavenet_2/dilation_layer_16-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_16-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters", shape = [64, 16]); - i"wavenet_2/dilation_layer_16-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_16", i"wavenet_2/dilation_layer_16-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_16-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_16-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [16, 64]); + i"wavenet_2/dilation_layer_16-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_16", i"wavenet_2/dilation_layer_16-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/add_16" = add(i"wavenet_2/dilation_layer_16-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.einsum", i"wavenet_2/add_15"); i"wavenet_2/dilation_layer_17-dilation_rate_2-filter-conv1d/convolution/Conv2D.delay" = tract_pulse_delay(i"wavenet_2/add_16", axis = 0, delay = 0, overlap = 4); i"wavenet_2/dilation_layer_17-dilation_rate_2-filter-conv1d/convolution/Conv2D_input" = transpose(unsqueeze(i"wavenet_2/dilation_layer_17-dilation_rate_2-filter-conv1d/convolution/Conv2D.delay", axes = [0]), axes = [0, 2, 1]); @@ -337,11 +337,11 @@ graph network(input_node) -> (i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolutio i"wavenet_2/dilation_layer_17-dilation_rate_2-gate-conv1d/convolution/Conv2D" = squeeze(transpose(i"wavenet_2/dilation_layer_17-dilation_rate_2-gate-conv1d/convolution/Conv2D_conv", axes = [0, 2, 1]), axes = [0]); i"wavenet_2/dilation_layer_17-dilation_rate_2-gate-conv1d/Sigmoid" = sigmoid(i"wavenet_2/dilation_layer_17-dilation_rate_2-gate-conv1d/convolution/Conv2D"); i"wavenet_2/mul_17" = mul(i"wavenet_2/dilation_layer_17-dilation_rate_2-filter-conv1d/Tanh", i"wavenet_2/dilation_layer_17-dilation_rate_2-gate-conv1d/Sigmoid"); - i"wavenet_2/dilation_layer_17-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_17-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters", shape = [64, 32]); - i"wavenet_2/dilation_layer_17-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_17", i"wavenet_2/dilation_layer_17-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_17-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_17-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [32, 64]); + i"wavenet_2/dilation_layer_17-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_17", i"wavenet_2/dilation_layer_17-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/AddN.16" = add(i"wavenet_2/AddN.15", i"wavenet_2/dilation_layer_17-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.einsum"); - i"wavenet_2/dilation_layer_17-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_17-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters", shape = [64, 16]); - i"wavenet_2/dilation_layer_17-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_17", i"wavenet_2/dilation_layer_17-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_17-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_17-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [16, 64]); + i"wavenet_2/dilation_layer_17-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_17", i"wavenet_2/dilation_layer_17-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/add_17" = add(i"wavenet_2/dilation_layer_17-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.einsum", i"wavenet_2/add_16"); i"wavenet_2/dilation_layer_18-dilation_rate_4-filter-conv1d/convolution/Conv2D.delay" = tract_pulse_delay(i"wavenet_2/add_17", axis = 0, delay = 0, overlap = 8); i"wavenet_2/dilation_layer_18-dilation_rate_4-filter-conv1d/convolution/Conv2D_input" = transpose(unsqueeze(i"wavenet_2/dilation_layer_18-dilation_rate_4-filter-conv1d/convolution/Conv2D.delay", axes = [0]), axes = [0, 2, 1]); @@ -355,11 +355,11 @@ graph network(input_node) -> (i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolutio i"wavenet_2/dilation_layer_18-dilation_rate_4-gate-conv1d/convolution/Conv2D" = squeeze(transpose(i"wavenet_2/dilation_layer_18-dilation_rate_4-gate-conv1d/convolution/Conv2D_conv", axes = [0, 2, 1]), axes = [0]); i"wavenet_2/dilation_layer_18-dilation_rate_4-gate-conv1d/Sigmoid" = sigmoid(i"wavenet_2/dilation_layer_18-dilation_rate_4-gate-conv1d/convolution/Conv2D"); i"wavenet_2/mul_18" = mul(i"wavenet_2/dilation_layer_18-dilation_rate_4-filter-conv1d/Tanh", i"wavenet_2/dilation_layer_18-dilation_rate_4-gate-conv1d/Sigmoid"); - i"wavenet_2/dilation_layer_18-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_18-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters", shape = [64, 32]); - i"wavenet_2/dilation_layer_18-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_18", i"wavenet_2/dilation_layer_18-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_18-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_18-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [32, 64]); + i"wavenet_2/dilation_layer_18-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_18", i"wavenet_2/dilation_layer_18-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/AddN.17" = add(i"wavenet_2/AddN.16", i"wavenet_2/dilation_layer_18-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.einsum"); - i"wavenet_2/dilation_layer_18-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_18-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters", shape = [64, 16]); - i"wavenet_2/dilation_layer_18-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_18", i"wavenet_2/dilation_layer_18-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_18-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_18-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [16, 64]); + i"wavenet_2/dilation_layer_18-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_18", i"wavenet_2/dilation_layer_18-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/add_18" = add(i"wavenet_2/dilation_layer_18-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.einsum", i"wavenet_2/add_17"); i"wavenet_2/dilation_layer_19-dilation_rate_8-filter-conv1d/convolution/Conv2D.delay" = tract_pulse_delay(i"wavenet_2/add_18", axis = 0, delay = 0, overlap = 16); i"wavenet_2/dilation_layer_19-dilation_rate_8-filter-conv1d/convolution/Conv2D_input" = transpose(unsqueeze(i"wavenet_2/dilation_layer_19-dilation_rate_8-filter-conv1d/convolution/Conv2D.delay", axes = [0]), axes = [0, 2, 1]); @@ -373,11 +373,11 @@ graph network(input_node) -> (i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolutio i"wavenet_2/dilation_layer_19-dilation_rate_8-gate-conv1d/convolution/Conv2D" = squeeze(transpose(i"wavenet_2/dilation_layer_19-dilation_rate_8-gate-conv1d/convolution/Conv2D_conv", axes = [0, 2, 1]), axes = [0]); i"wavenet_2/dilation_layer_19-dilation_rate_8-gate-conv1d/Sigmoid" = sigmoid(i"wavenet_2/dilation_layer_19-dilation_rate_8-gate-conv1d/convolution/Conv2D"); i"wavenet_2/mul_19" = mul(i"wavenet_2/dilation_layer_19-dilation_rate_8-filter-conv1d/Tanh", i"wavenet_2/dilation_layer_19-dilation_rate_8-gate-conv1d/Sigmoid"); - i"wavenet_2/dilation_layer_19-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_19-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters", shape = [64, 32]); - i"wavenet_2/dilation_layer_19-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_19", i"wavenet_2/dilation_layer_19-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_19-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_19-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [32, 64]); + i"wavenet_2/dilation_layer_19-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_19", i"wavenet_2/dilation_layer_19-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/AddN.18" = add(i"wavenet_2/AddN.17", i"wavenet_2/dilation_layer_19-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.einsum"); - i"wavenet_2/dilation_layer_19-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_19-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters", shape = [64, 16]); - i"wavenet_2/dilation_layer_19-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_19", i"wavenet_2/dilation_layer_19-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_19-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_19-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [16, 64]); + i"wavenet_2/dilation_layer_19-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_19", i"wavenet_2/dilation_layer_19-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/add_19" = add(i"wavenet_2/dilation_layer_19-dilation_rate_8-1x1_conv_transform-conv1d/convolution/Conv2D.einsum", i"wavenet_2/add_18"); i"wavenet_2/dilation_layer_20-dilation_rate_1-filter-conv1d/convolution/Conv2D.delay" = tract_pulse_delay(i"wavenet_2/add_19", axis = 0, delay = 0, overlap = 2); i"wavenet_2/dilation_layer_20-dilation_rate_1-filter-conv1d/convolution/Conv2D_input" = transpose(unsqueeze(i"wavenet_2/dilation_layer_20-dilation_rate_1-filter-conv1d/convolution/Conv2D.delay", axes = [0]), axes = [0, 2, 1]); @@ -391,11 +391,11 @@ graph network(input_node) -> (i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolutio i"wavenet_2/dilation_layer_20-dilation_rate_1-gate-conv1d/convolution/Conv2D" = squeeze(transpose(i"wavenet_2/dilation_layer_20-dilation_rate_1-gate-conv1d/convolution/Conv2D_conv", axes = [0, 2, 1]), axes = [0]); i"wavenet_2/dilation_layer_20-dilation_rate_1-gate-conv1d/Sigmoid" = sigmoid(i"wavenet_2/dilation_layer_20-dilation_rate_1-gate-conv1d/convolution/Conv2D"); i"wavenet_2/mul_20" = mul(i"wavenet_2/dilation_layer_20-dilation_rate_1-filter-conv1d/Tanh", i"wavenet_2/dilation_layer_20-dilation_rate_1-gate-conv1d/Sigmoid"); - i"wavenet_2/dilation_layer_20-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_20-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters", shape = [64, 32]); - i"wavenet_2/dilation_layer_20-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_20", i"wavenet_2/dilation_layer_20-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_20-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_20-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [32, 64]); + i"wavenet_2/dilation_layer_20-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_20", i"wavenet_2/dilation_layer_20-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/AddN.19" = add(i"wavenet_2/AddN.18", i"wavenet_2/dilation_layer_20-dilation_rate_1-1x1_conv_skip-conv1d/convolution/Conv2D.einsum"); - i"wavenet_2/dilation_layer_20-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_20-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters", shape = [64, 16]); - i"wavenet_2/dilation_layer_20-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_20", i"wavenet_2/dilation_layer_20-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_20-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_20-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [16, 64]); + i"wavenet_2/dilation_layer_20-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_20", i"wavenet_2/dilation_layer_20-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/add_20" = add(i"wavenet_2/dilation_layer_20-dilation_rate_1-1x1_conv_transform-conv1d/convolution/Conv2D.einsum", i"wavenet_2/add_19"); i"wavenet_2/dilation_layer_21-dilation_rate_2-filter-conv1d/convolution/Conv2D.delay" = tract_pulse_delay(i"wavenet_2/add_20", axis = 0, delay = 0, overlap = 4); i"wavenet_2/dilation_layer_21-dilation_rate_2-filter-conv1d/convolution/Conv2D_input" = transpose(unsqueeze(i"wavenet_2/dilation_layer_21-dilation_rate_2-filter-conv1d/convolution/Conv2D.delay", axes = [0]), axes = [0, 2, 1]); @@ -409,11 +409,11 @@ graph network(input_node) -> (i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolutio i"wavenet_2/dilation_layer_21-dilation_rate_2-gate-conv1d/convolution/Conv2D" = squeeze(transpose(i"wavenet_2/dilation_layer_21-dilation_rate_2-gate-conv1d/convolution/Conv2D_conv", axes = [0, 2, 1]), axes = [0]); i"wavenet_2/dilation_layer_21-dilation_rate_2-gate-conv1d/Sigmoid" = sigmoid(i"wavenet_2/dilation_layer_21-dilation_rate_2-gate-conv1d/convolution/Conv2D"); i"wavenet_2/mul_21" = mul(i"wavenet_2/dilation_layer_21-dilation_rate_2-filter-conv1d/Tanh", i"wavenet_2/dilation_layer_21-dilation_rate_2-gate-conv1d/Sigmoid"); - i"wavenet_2/dilation_layer_21-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_21-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters", shape = [64, 32]); - i"wavenet_2/dilation_layer_21-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_21", i"wavenet_2/dilation_layer_21-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_21-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_21-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [32, 64]); + i"wavenet_2/dilation_layer_21-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_21", i"wavenet_2/dilation_layer_21-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/AddN.20" = add(i"wavenet_2/AddN.19", i"wavenet_2/dilation_layer_21-dilation_rate_2-1x1_conv_skip-conv1d/convolution/Conv2D.einsum"); - i"wavenet_2/dilation_layer_21-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_21-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters", shape = [64, 16]); - i"wavenet_2/dilation_layer_21-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_21", i"wavenet_2/dilation_layer_21-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_21-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_21-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [16, 64]); + i"wavenet_2/dilation_layer_21-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_21", i"wavenet_2/dilation_layer_21-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/add_21" = add(i"wavenet_2/dilation_layer_21-dilation_rate_2-1x1_conv_transform-conv1d/convolution/Conv2D.einsum", i"wavenet_2/add_20"); i"wavenet_2/dilation_layer_22-dilation_rate_4-filter-conv1d/convolution/Conv2D.delay" = tract_pulse_delay(i"wavenet_2/add_21", axis = 0, delay = 0, overlap = 8); i"wavenet_2/dilation_layer_22-dilation_rate_4-filter-conv1d/convolution/Conv2D_input" = transpose(unsqueeze(i"wavenet_2/dilation_layer_22-dilation_rate_4-filter-conv1d/convolution/Conv2D.delay", axes = [0]), axes = [0, 2, 1]); @@ -427,11 +427,11 @@ graph network(input_node) -> (i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolutio i"wavenet_2/dilation_layer_22-dilation_rate_4-gate-conv1d/convolution/Conv2D" = squeeze(transpose(i"wavenet_2/dilation_layer_22-dilation_rate_4-gate-conv1d/convolution/Conv2D_conv", axes = [0, 2, 1]), axes = [0]); i"wavenet_2/dilation_layer_22-dilation_rate_4-gate-conv1d/Sigmoid" = sigmoid(i"wavenet_2/dilation_layer_22-dilation_rate_4-gate-conv1d/convolution/Conv2D"); i"wavenet_2/mul_22" = mul(i"wavenet_2/dilation_layer_22-dilation_rate_4-filter-conv1d/Tanh", i"wavenet_2/dilation_layer_22-dilation_rate_4-gate-conv1d/Sigmoid"); - i"wavenet_2/dilation_layer_22-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_22-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters", shape = [64, 32]); - i"wavenet_2/dilation_layer_22-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_22", i"wavenet_2/dilation_layer_22-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_22-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_22-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [32, 64]); + i"wavenet_2/dilation_layer_22-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_22", i"wavenet_2/dilation_layer_22-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/AddN.21" = add(i"wavenet_2/AddN.20", i"wavenet_2/dilation_layer_22-dilation_rate_4-1x1_conv_skip-conv1d/convolution/Conv2D.einsum"); - i"wavenet_2/dilation_layer_22-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_22-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters", shape = [64, 16]); - i"wavenet_2/dilation_layer_22-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_22", i"wavenet_2/dilation_layer_22-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_22-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_22-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [16, 64]); + i"wavenet_2/dilation_layer_22-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_22", i"wavenet_2/dilation_layer_22-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/add_22" = add(i"wavenet_2/dilation_layer_22-dilation_rate_4-1x1_conv_transform-conv1d/convolution/Conv2D.einsum", i"wavenet_2/add_21"); i"wavenet_2/dilation_layer_23-dilation_rate_8-filter-conv1d/convolution/Conv2D.delay" = tract_pulse_delay(i"wavenet_2/add_22", axis = 0, delay = 0, overlap = 16); i"wavenet_2/dilation_layer_23-dilation_rate_8-filter-conv1d/convolution/Conv2D_input" = transpose(unsqueeze(i"wavenet_2/dilation_layer_23-dilation_rate_8-filter-conv1d/convolution/Conv2D.delay", axes = [0]), axes = [0, 2, 1]); @@ -445,15 +445,15 @@ graph network(input_node) -> (i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolutio i"wavenet_2/dilation_layer_23-dilation_rate_8-gate-conv1d/convolution/Conv2D" = squeeze(transpose(i"wavenet_2/dilation_layer_23-dilation_rate_8-gate-conv1d/convolution/Conv2D_conv", axes = [0, 2, 1]), axes = [0]); i"wavenet_2/dilation_layer_23-dilation_rate_8-gate-conv1d/Sigmoid" = sigmoid(i"wavenet_2/dilation_layer_23-dilation_rate_8-gate-conv1d/convolution/Conv2D"); i"wavenet_2/mul_23" = mul(i"wavenet_2/dilation_layer_23-dilation_rate_8-filter-conv1d/Tanh", i"wavenet_2/dilation_layer_23-dilation_rate_8-gate-conv1d/Sigmoid"); - i"wavenet_2/dilation_layer_23-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/dilation_layer_23-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters", shape = [64, 32]); - i"wavenet_2/dilation_layer_23-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_23", i"wavenet_2/dilation_layer_23-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/dilation_layer_23-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/dilation_layer_23-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [32, 64]); + i"wavenet_2/dilation_layer_23-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/mul_23", i"wavenet_2/dilation_layer_23-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/AddN.22" = add(i"wavenet_2/AddN.21", i"wavenet_2/dilation_layer_23-dilation_rate_8-1x1_conv_skip-conv1d/convolution/Conv2D.einsum"); i"wavenet_2/Relu.low.cst" = [[0.0]]; i"wavenet_2/Relu.low" = max(i"wavenet_2/AddN.22", i"wavenet_2/Relu.low.cst"); - i"wavenet_2/post_proc_1-1x1_conv-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/post_proc_1-1x1_conv-conv1d/convolution/Conv2D.filters", shape = [32, 32]); - i"wavenet_2/post_proc_1-1x1_conv-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/Relu.low", i"wavenet_2/post_proc_1-1x1_conv-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/post_proc_1-1x1_conv-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/post_proc_1-1x1_conv-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [32, 32]); + i"wavenet_2/post_proc_1-1x1_conv-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/Relu.low", i"wavenet_2/post_proc_1-1x1_conv-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); i"wavenet_2/post_proc_1-1x1_conv-conv1d/Relu.low.cst" = [[0.0]]; i"wavenet_2/post_proc_1-1x1_conv-conv1d/Relu.low" = max(i"wavenet_2/post_proc_1-1x1_conv-conv1d/convolution/Conv2D.einsum", i"wavenet_2/post_proc_1-1x1_conv-conv1d/Relu.low.cst"); - i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolution/Conv2D.filters" = variable(label = "wavenet_2/post_proc_2-1x1_conv-conv1d/convolution/Conv2D.filters", shape = [32, 2]); - i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/post_proc_1-1x1_conv-conv1d/Relu.low", i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolution/Conv2D.filters", transposeA = false, transposeB = false); + i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolution/Conv2D.filters_as_co_ci" = variable(label = "wavenet_2/post_proc_2-1x1_conv-conv1d/convolution/Conv2D.filters_as_co_ci", shape = [2, 32]); + i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolution/Conv2D.einsum" = matmul(i"wavenet_2/post_proc_1-1x1_conv-conv1d/Relu.low", i"wavenet_2/post_proc_2-1x1_conv-conv1d/convolution/Conv2D.filters_as_co_ci", transposeA = false, transposeB = true); } diff --git a/harness/pre-optimized-graphes/mdl-en-2019-Q3-librispeech/expected b/harness/pre-optimized-graphes/mdl-en-2019-Q3-librispeech/expected index eced13236e..68410fdc41 100644 --- a/harness/pre-optimized-graphes/mdl-en-2019-Q3-librispeech/expected +++ b/harness/pre-optimized-graphes/mdl-en-2019-Q3-librispeech/expected @@ -133,8 +133,8 @@ graph network(input) -> (output) { i"lda.output_bias" = variable(label = "lda.output_bias", shape = [200]); i"lda.output_conv" = conv(i"lda.output_input", i"lda.output_weigths", i"lda.output_bias", dilation = [1], stride = [1], border = "constant", groups = 1, padding = [(0, 0)]); i"lda.output" = squeeze(transpose(i"lda.output_conv", axes = [0, 2, 1]), axes = [0]); - i"tdnn1.affine.output.filters" = variable(label = "tdnn1.affine.output.filters", shape = [256, 200]); - i"tdnn1.affine.output.einsum" = matmul(i"tdnn1.affine.output.filters", i"lda.output", transposeA = false, transposeB = true); + i"tdnn1.affine.output.filters_as_co_ci" = variable(label = "tdnn1.affine.output.filters_as_co_ci", shape = [256, 200]); + i"tdnn1.affine.output.einsum" = matmul(i"tdnn1.affine.output.filters_as_co_ci", i"lda.output", transposeA = false, transposeB = true); i"tdnn1.affine.output.bias.cst" = variable(label = "tdnn1.affine.output.bias.cst", shape = [256, 1]); i"tdnn1.affine.output.bias" = add(i"tdnn1.affine.output.einsum", i"tdnn1.affine.output.bias.cst"); i"tdnn1.relu.output.low.cst" = [[0.0]]; @@ -186,7 +186,7 @@ graph network(input) -> (output) { i"fastlstm1.c_final.extracted.fastlstm1.four_parts.W.concat-einsum-k.0..256.split-over-1.768..1024.fix_b.0" = variable(label = "fastlstm1.c_final.extracted.fastlstm1.four_parts.W.concat-einsum-k.0..256.split-over-1.768..1024.fix_b.0", shape = [1, 256, 256]); i"fastlstm1.c_final.extracted.fastlstm1.four_parts.W.concat-einsum-k.0..256.split-over-1.768..1024" = matmul(i"fastlstm1.c_final.extracted.fastlstm1.four_parts.W.concat-einsum-k.0..256.split-over-1.768..1024.fix_a.0", i"fastlstm1.c_final.extracted.fastlstm1.four_parts.W.concat-einsum-k.0..256.split-over-1.768..1024.fix_b.0", transposeA = true, transposeB = false); i"fastlstm1.c_final.extracted.fastlstm1.four_parts.W.concat-einsum-k.0..256.split-over-1.768..1024.fix_c.0" = transpose(i"fastlstm1.c_final.extracted.fastlstm1.four_parts.W.concat-einsum-k.0..256.split-over-1.768..1024", axes = [1, 0, 2]); - i"incoming-80/0" = variable(label = "incoming-80/0", shape = [1, 256]); + i"incoming-83/0" = variable(label = "incoming-83/0", shape = [1, 256]); i"incoming-32/0" = variable(label = "incoming-32/0", shape = [128, 1]); i"fastlstm1.four_parts.W.concat-einsum-k.256..384.split-1-over-1.0..256.slice" = variable(label = "fastlstm1.four_parts.W.concat-einsum-k.256..384.split-1-over-1.0..256.slice", shape = [128, 256]); i"fastlstm1.four_parts.W.concat-einsum-k.256..384.split-1-over-1.256..512.slice" = variable(label = "fastlstm1.four_parts.W.concat-einsum-k.256..384.split-1-over-1.256..512.slice", shape = [128, 256]); @@ -201,7 +201,7 @@ graph network(input) -> (output) { i"fastlstm1.peephole0.mul.fix-rank-0-1" = variable(label = "fastlstm1.peephole0.mul.fix-rank-0-1", shape = [1, 256]); i"fastlstm1.peephole1.mul.fix-rank-0-1" = variable(label = "fastlstm1.peephole1.mul.fix-rank-0-1", shape = [1, 256]); i"fastlstm1.peephole2.mul.fix-rank-0-1" = variable(label = "fastlstm1.peephole2.mul.fix-rank-0-1", shape = [1, 256]); - ( i"fastlstm1.c_final", i"fastlstm1.c_final_1" ) = tract_core_scan(body = "scan_body_0", scan = [("fastlstm1.c_final.extracted.fastlstm1.four_parts.W.concat-einsum-k.0..256.split-over-1.0..256", i"fastlstm1.c_final.extracted.fastlstm1.four_parts.W.concat-einsum-k.0..256.split-over-1.0..256", 0, 1), ("fastlstm1.c_final.extracted.fastlstm1.four_parts.W.concat-einsum-k.0..256.split-over-1.512..768", i"fastlstm1.c_final.extracted.fastlstm1.four_parts.W.concat-einsum-k.0..256.split-over-1.512..768.fix_c.0", 0, 1), ("fastlstm1.c_final.extracted.fastlstm1.four_parts.W.concat-einsum-k.0..256.split-over-1.256..512", i"fastlstm1.c_final.extracted.fastlstm1.four_parts.W.concat-einsum-k.0..256.split-over-1.256..512.fix_c.0", 0, 1), ("fastlstm1.c_final.extracted.fastlstm1.four_parts.W.concat-einsum-k.0..256.split-over-1.768..1024", i"fastlstm1.c_final.extracted.fastlstm1.four_parts.W.concat-einsum-k.0..256.split-over-1.768..1024.fix_c.0", 0, 1)], full = [("fastlstm1.four_parts.W.concat-einsum-k.256..384.split-1-over-1.0..256.slice", i"fastlstm1.four_parts.W.concat-einsum-k.256..384.split-1-over-1.0..256.slice"), ("fastlstm1.four_parts.W.concat-einsum-k.256..384.split-1-over-1.256..512.slice", i"fastlstm1.four_parts.W.concat-einsum-k.256..384.split-1-over-1.256..512.slice"), ("fastlstm1.four_parts.W.concat-einsum-k.256..384.split-1-over-1.512..768.slice", i"fastlstm1.four_parts.W.concat-einsum-k.256..384.split-1-over-1.512..768.slice"), ("fastlstm1.four_parts.W.concat-einsum-k.256..384.split-1-over-1.768..1024.slice", i"fastlstm1.four_parts.W.concat-einsum-k.256..384.split-1-over-1.768..1024.slice"), ("fastlstm1.four_parts.split-1-over-1.0..256.slice", i"fastlstm1.four_parts.split-1-over-1.0..256.slice"), ("fastlstm1.four_parts.split-1-over-1.256..512.slice", i"fastlstm1.four_parts.split-1-over-1.256..512.slice"), ("fastlstm1.four_parts.split-1-over-1.512..768.slice", i"fastlstm1.four_parts.split-1-over-1.512..768.slice"), ("fastlstm1.four_parts.split-1-over-1.768..1024.slice", i"fastlstm1.four_parts.split-1-over-1.768..1024.slice"), ("fastlstm1.h_new.W.split-1-over-1.0..128.slice", i"fastlstm1.h_new.W.split-1-over-1.0..128.slice"), ("fastlstm1.h_new.split-1-over-1.0..128.slice", i"fastlstm1.h_new.split-1-over-1.0..128.slice"), ("fastlstm1.peephole0.mul.fix-rank-0-1", i"fastlstm1.peephole0.mul.fix-rank-0-1"), ("fastlstm1.peephole1.mul.fix-rank-0-1", i"fastlstm1.peephole1.mul.fix-rank-0-1"), ("fastlstm1.peephole2.mul.fix-rank-0-1", i"fastlstm1.peephole2.mul.fix-rank-0-1")], state = [("fastlstm1.c", i"incoming-80/0", "fastlstm1.c_new"), ("fastlstm1.r", i"incoming-32/0", "fastlstm1.r_new")], output = [("fastlstm1.r_new", "full", 1, 1), ("fastlstm1.h_new.W.split-over-1.128..256.prop_axis.a.input_0", "full", 0, 1)], skip = 2, reset_every_turn = false); + ( i"fastlstm1.c_final", i"fastlstm1.c_final_1" ) = tract_core_scan(body = "scan_body_0", scan = [("fastlstm1.c_final.extracted.fastlstm1.four_parts.W.concat-einsum-k.0..256.split-over-1.0..256", i"fastlstm1.c_final.extracted.fastlstm1.four_parts.W.concat-einsum-k.0..256.split-over-1.0..256", 0, 1), ("fastlstm1.c_final.extracted.fastlstm1.four_parts.W.concat-einsum-k.0..256.split-over-1.512..768", i"fastlstm1.c_final.extracted.fastlstm1.four_parts.W.concat-einsum-k.0..256.split-over-1.512..768.fix_c.0", 0, 1), ("fastlstm1.c_final.extracted.fastlstm1.four_parts.W.concat-einsum-k.0..256.split-over-1.256..512", i"fastlstm1.c_final.extracted.fastlstm1.four_parts.W.concat-einsum-k.0..256.split-over-1.256..512.fix_c.0", 0, 1), ("fastlstm1.c_final.extracted.fastlstm1.four_parts.W.concat-einsum-k.0..256.split-over-1.768..1024", i"fastlstm1.c_final.extracted.fastlstm1.four_parts.W.concat-einsum-k.0..256.split-over-1.768..1024.fix_c.0", 0, 1)], full = [("fastlstm1.four_parts.W.concat-einsum-k.256..384.split-1-over-1.0..256.slice", i"fastlstm1.four_parts.W.concat-einsum-k.256..384.split-1-over-1.0..256.slice"), ("fastlstm1.four_parts.W.concat-einsum-k.256..384.split-1-over-1.256..512.slice", i"fastlstm1.four_parts.W.concat-einsum-k.256..384.split-1-over-1.256..512.slice"), ("fastlstm1.four_parts.W.concat-einsum-k.256..384.split-1-over-1.512..768.slice", i"fastlstm1.four_parts.W.concat-einsum-k.256..384.split-1-over-1.512..768.slice"), ("fastlstm1.four_parts.W.concat-einsum-k.256..384.split-1-over-1.768..1024.slice", i"fastlstm1.four_parts.W.concat-einsum-k.256..384.split-1-over-1.768..1024.slice"), ("fastlstm1.four_parts.split-1-over-1.0..256.slice", i"fastlstm1.four_parts.split-1-over-1.0..256.slice"), ("fastlstm1.four_parts.split-1-over-1.256..512.slice", i"fastlstm1.four_parts.split-1-over-1.256..512.slice"), ("fastlstm1.four_parts.split-1-over-1.512..768.slice", i"fastlstm1.four_parts.split-1-over-1.512..768.slice"), ("fastlstm1.four_parts.split-1-over-1.768..1024.slice", i"fastlstm1.four_parts.split-1-over-1.768..1024.slice"), ("fastlstm1.h_new.W.split-1-over-1.0..128.slice", i"fastlstm1.h_new.W.split-1-over-1.0..128.slice"), ("fastlstm1.h_new.split-1-over-1.0..128.slice", i"fastlstm1.h_new.split-1-over-1.0..128.slice"), ("fastlstm1.peephole0.mul.fix-rank-0-1", i"fastlstm1.peephole0.mul.fix-rank-0-1"), ("fastlstm1.peephole1.mul.fix-rank-0-1", i"fastlstm1.peephole1.mul.fix-rank-0-1"), ("fastlstm1.peephole2.mul.fix-rank-0-1", i"fastlstm1.peephole2.mul.fix-rank-0-1")], state = [("fastlstm1.c", i"incoming-83/0", "fastlstm1.c_new"), ("fastlstm1.r", i"incoming-32/0", "fastlstm1.r_new")], output = [("fastlstm1.r_new", "full", 1, 1), ("fastlstm1.h_new.W.split-over-1.128..256.prop_axis.a.input_0", "full", 0, 1)], skip = 2, reset_every_turn = false); i"fastlstm1.h_new.W.split-over-1.128..256.fix_a.0" = transpose(i"fastlstm1.c_final_1", axes = [1, 0, 2]); i"fastlstm1.h_new.W.split-over-1.128..256.fix_b.0" = variable(label = "fastlstm1.h_new.W.split-over-1.128..256.fix_b.0", shape = [1, 256, 128]); i"fastlstm1.h_new.W.split-over-1.128..256" = matmul(i"fastlstm1.h_new.W.split-over-1.128..256.fix_b.0", i"fastlstm1.h_new.W.split-over-1.128..256.fix_a.0", transposeA = true, transposeB = true); @@ -251,7 +251,7 @@ graph network(input) -> (output) { i"fastlstm2.c_final.extracted.fastlstm2.four_parts.W.concat-einsum-k.0..256.split-over-1.768..1024.fix_b.0" = variable(label = "fastlstm2.c_final.extracted.fastlstm2.four_parts.W.concat-einsum-k.0..256.split-over-1.768..1024.fix_b.0", shape = [1, 256, 256]); i"fastlstm2.c_final.extracted.fastlstm2.four_parts.W.concat-einsum-k.0..256.split-over-1.768..1024" = matmul(i"fastlstm2.c_final.extracted.fastlstm2.four_parts.W.concat-einsum-k.0..256.split-over-1.768..1024.fix_a.0", i"fastlstm2.c_final.extracted.fastlstm2.four_parts.W.concat-einsum-k.0..256.split-over-1.768..1024.fix_b.0", transposeA = true, transposeB = false); i"fastlstm2.c_final.extracted.fastlstm2.four_parts.W.concat-einsum-k.0..256.split-over-1.768..1024.fix_c.0" = transpose(i"fastlstm2.c_final.extracted.fastlstm2.four_parts.W.concat-einsum-k.0..256.split-over-1.768..1024", axes = [1, 0, 2]); - i"incoming-140/0" = variable(label = "incoming-140/0", shape = [1, 256]); + i"incoming-143/0" = variable(label = "incoming-143/0", shape = [1, 256]); i"incoming-53/0" = variable(label = "incoming-53/0", shape = [1, 128]); i"fastlstm2.four_parts.W.concat-einsum-k.256..384.split-1-over-1.0..256.slice" = variable(label = "fastlstm2.four_parts.W.concat-einsum-k.256..384.split-1-over-1.0..256.slice", shape = [128, 256]); i"fastlstm2.four_parts.W.concat-einsum-k.256..384.split-1-over-1.256..512.slice" = variable(label = "fastlstm2.four_parts.W.concat-einsum-k.256..384.split-1-over-1.256..512.slice", shape = [128, 256]); @@ -266,7 +266,7 @@ graph network(input) -> (output) { i"fastlstm2.peephole0.mul.fix-rank-0-1" = variable(label = "fastlstm2.peephole0.mul.fix-rank-0-1", shape = [1, 256]); i"fastlstm2.peephole1.mul.fix-rank-0-1" = variable(label = "fastlstm2.peephole1.mul.fix-rank-0-1", shape = [1, 256]); i"fastlstm2.peephole2.mul.fix-rank-0-1" = variable(label = "fastlstm2.peephole2.mul.fix-rank-0-1", shape = [1, 256]); - ( i"fastlstm2.c_final", i"fastlstm2.c_final_1" ) = tract_core_scan(body = "scan_body_1", scan = [("fastlstm2.c_final.extracted.fastlstm2.four_parts.W.concat-einsum-k.0..256.split-over-1.0..256", i"fastlstm2.c_final.extracted.fastlstm2.four_parts.W.concat-einsum-k.0..256.split-over-1.0..256", 0, 1), ("fastlstm2.c_final.extracted.fastlstm2.four_parts.W.concat-einsum-k.0..256.split-over-1.512..768", i"fastlstm2.c_final.extracted.fastlstm2.four_parts.W.concat-einsum-k.0..256.split-over-1.512..768.fix_c.0", 0, 1), ("fastlstm2.c_final.extracted.fastlstm2.four_parts.W.concat-einsum-k.0..256.split-over-1.256..512", i"fastlstm2.c_final.extracted.fastlstm2.four_parts.W.concat-einsum-k.0..256.split-over-1.256..512.fix_c.0", 0, 1), ("fastlstm2.c_final.extracted.fastlstm2.four_parts.W.concat-einsum-k.0..256.split-over-1.768..1024", i"fastlstm2.c_final.extracted.fastlstm2.four_parts.W.concat-einsum-k.0..256.split-over-1.768..1024.fix_c.0", 0, 1)], full = [("fastlstm2.four_parts.W.concat-einsum-k.256..384.split-1-over-1.0..256.slice", i"fastlstm2.four_parts.W.concat-einsum-k.256..384.split-1-over-1.0..256.slice"), ("fastlstm2.four_parts.W.concat-einsum-k.256..384.split-1-over-1.256..512.slice", i"fastlstm2.four_parts.W.concat-einsum-k.256..384.split-1-over-1.256..512.slice"), ("fastlstm2.four_parts.W.concat-einsum-k.256..384.split-1-over-1.512..768.slice", i"fastlstm2.four_parts.W.concat-einsum-k.256..384.split-1-over-1.512..768.slice"), ("fastlstm2.four_parts.W.concat-einsum-k.256..384.split-1-over-1.768..1024.slice", i"fastlstm2.four_parts.W.concat-einsum-k.256..384.split-1-over-1.768..1024.slice"), ("fastlstm2.four_parts.split-1-over-1.0..256.slice", i"fastlstm2.four_parts.split-1-over-1.0..256.slice"), ("fastlstm2.four_parts.split-1-over-1.256..512.slice", i"fastlstm2.four_parts.split-1-over-1.256..512.slice"), ("fastlstm2.four_parts.split-1-over-1.512..768.slice", i"fastlstm2.four_parts.split-1-over-1.512..768.slice"), ("fastlstm2.four_parts.split-1-over-1.768..1024.slice", i"fastlstm2.four_parts.split-1-over-1.768..1024.slice"), ("fastlstm2.h_new.W.split-1-over-1.0..128.slice", i"fastlstm2.h_new.W.split-1-over-1.0..128.slice"), ("fastlstm2.h_new.split-1-over-1.0..128.slice", i"fastlstm2.h_new.split-1-over-1.0..128.slice"), ("fastlstm2.peephole0.mul.fix-rank-0-1", i"fastlstm2.peephole0.mul.fix-rank-0-1"), ("fastlstm2.peephole1.mul.fix-rank-0-1", i"fastlstm2.peephole1.mul.fix-rank-0-1"), ("fastlstm2.peephole2.mul.fix-rank-0-1", i"fastlstm2.peephole2.mul.fix-rank-0-1")], state = [("fastlstm2.c", i"incoming-140/0", "fastlstm2.c_new"), ("fastlstm2.r", i"incoming-53/0", "fastlstm2.r_new")], output = [("fastlstm2.r_new", "full", 0, 1), ("fastlstm2.h_new.W.split-over-1.128..256.prop_axis.a.input_0", "full", 0, 1)], skip = 6, reset_every_turn = false); + ( i"fastlstm2.c_final", i"fastlstm2.c_final_1" ) = tract_core_scan(body = "scan_body_1", scan = [("fastlstm2.c_final.extracted.fastlstm2.four_parts.W.concat-einsum-k.0..256.split-over-1.0..256", i"fastlstm2.c_final.extracted.fastlstm2.four_parts.W.concat-einsum-k.0..256.split-over-1.0..256", 0, 1), ("fastlstm2.c_final.extracted.fastlstm2.four_parts.W.concat-einsum-k.0..256.split-over-1.512..768", i"fastlstm2.c_final.extracted.fastlstm2.four_parts.W.concat-einsum-k.0..256.split-over-1.512..768.fix_c.0", 0, 1), ("fastlstm2.c_final.extracted.fastlstm2.four_parts.W.concat-einsum-k.0..256.split-over-1.256..512", i"fastlstm2.c_final.extracted.fastlstm2.four_parts.W.concat-einsum-k.0..256.split-over-1.256..512.fix_c.0", 0, 1), ("fastlstm2.c_final.extracted.fastlstm2.four_parts.W.concat-einsum-k.0..256.split-over-1.768..1024", i"fastlstm2.c_final.extracted.fastlstm2.four_parts.W.concat-einsum-k.0..256.split-over-1.768..1024.fix_c.0", 0, 1)], full = [("fastlstm2.four_parts.W.concat-einsum-k.256..384.split-1-over-1.0..256.slice", i"fastlstm2.four_parts.W.concat-einsum-k.256..384.split-1-over-1.0..256.slice"), ("fastlstm2.four_parts.W.concat-einsum-k.256..384.split-1-over-1.256..512.slice", i"fastlstm2.four_parts.W.concat-einsum-k.256..384.split-1-over-1.256..512.slice"), ("fastlstm2.four_parts.W.concat-einsum-k.256..384.split-1-over-1.512..768.slice", i"fastlstm2.four_parts.W.concat-einsum-k.256..384.split-1-over-1.512..768.slice"), ("fastlstm2.four_parts.W.concat-einsum-k.256..384.split-1-over-1.768..1024.slice", i"fastlstm2.four_parts.W.concat-einsum-k.256..384.split-1-over-1.768..1024.slice"), ("fastlstm2.four_parts.split-1-over-1.0..256.slice", i"fastlstm2.four_parts.split-1-over-1.0..256.slice"), ("fastlstm2.four_parts.split-1-over-1.256..512.slice", i"fastlstm2.four_parts.split-1-over-1.256..512.slice"), ("fastlstm2.four_parts.split-1-over-1.512..768.slice", i"fastlstm2.four_parts.split-1-over-1.512..768.slice"), ("fastlstm2.four_parts.split-1-over-1.768..1024.slice", i"fastlstm2.four_parts.split-1-over-1.768..1024.slice"), ("fastlstm2.h_new.W.split-1-over-1.0..128.slice", i"fastlstm2.h_new.W.split-1-over-1.0..128.slice"), ("fastlstm2.h_new.split-1-over-1.0..128.slice", i"fastlstm2.h_new.split-1-over-1.0..128.slice"), ("fastlstm2.peephole0.mul.fix-rank-0-1", i"fastlstm2.peephole0.mul.fix-rank-0-1"), ("fastlstm2.peephole1.mul.fix-rank-0-1", i"fastlstm2.peephole1.mul.fix-rank-0-1"), ("fastlstm2.peephole2.mul.fix-rank-0-1", i"fastlstm2.peephole2.mul.fix-rank-0-1")], state = [("fastlstm2.c", i"incoming-143/0", "fastlstm2.c_new"), ("fastlstm2.r", i"incoming-53/0", "fastlstm2.r_new")], output = [("fastlstm2.r_new", "full", 0, 1), ("fastlstm2.h_new.W.split-over-1.128..256.prop_axis.a.input_0", "full", 0, 1)], skip = 6, reset_every_turn = false); i"output.affine.output.W.concat-einsum-slice-k.0.0..128" = variable(label = "output.affine.output.W.concat-einsum-slice-k.0.0..128", shape = [1690, 128]); i"output.affine.output.W.concat-einsum-k.0..128" = matmul(i"fastlstm2.c_final", i"output.affine.output.W.concat-einsum-slice-k.0.0..128", transposeA = false, transposeB = true); i"fastlstm2.h_new.W.split-over-1.128..256.fix_a.0" = transpose(i"fastlstm2.c_final_1", axes = [1, 0, 2]);