TOCO - размеры должны совпадать (оператор Concat)

Question

Я преобразую модель семантической сегментации в 8-битный Quantised TFLite с использованием TOCO, однако получаю следующую ошибку.

tensorflow/contrib/lite/toco/graph_transformations/propagate_fixed_sizes.cc:116] Check failed: dim_x == dim_y (3 vs. 13)Dimensions must match
    Abort trap: 6

Я использую оператор Concat TF несколько раз, каждый раз на оси 3, так что размеры concat([1,20,40,13], [1,20,40,3]) = [1,20,40,16], что, очевидно, где происходит ошибка, однако вызываемая функция

bazel build tensorflow/contrib/lite/toco:toco && \
./bazel-bin/tensorflow/contrib/lite/toco/toco \
    --input_file=/path/to/MyModel/log/quantised/optimised_graph.pb \
    --output_file=/path/to/MyModel/log/quantised/quantised_graph.tflite \
    --input_format="TENSORFLOW_GRAPHDEF" \
    --output_format="TFLITE" \
    --inference_type="QUANTIZED_UINT8" \
    --inference_input_type="QUANTIZED_UINT8" \
    --allow_custom_ops=true \
    --drop_control_dependency=false \
    --allow_nudging_weights_to_use_fast_gemm_kernel=true \
    --quantize_weights=false \
    --input_arrays="MyModel/downsampling_block_1_conv/Conv2D","MyModel/downsampling_block_1_pool/MaxPool" \
    --output_arrays="MyModel/Softmax" \
    --input_shapes="1,360,480,3":"1,360,480,3"

Состояния, чтобы разрешить пользовательские операции, которые работают для определенных операций, которые я использую, однако здесь это не работает. От Googling эта ошибка называется еще не реализованной ошибкой типа, однако, наверняка, разрешение пользовательских операций должно исправить это?

Наконец, запуск этой функции создает выходной файл, однако это пустой 0-байтовый файл.