• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• utensor_cgen-master
  • .circleci
    • config.yml
  • Makefile
  • _config.yml
  • requirements_rtd.txt
  • readthedocs.yml
  • LICENSE
  • example_config.toml
  • Pipfile
  • README.rst
  • setup.py
  • tutorials
    • end2end_training
      • tutorial_end2end_keras.ipynb
      • images
      • input_image.h
      • .gitignore
    • add_custom_operators
      • images
        • utensor-cli-components.drawio.svg
        • reduceModel.svg
      • custom_operators
        • my_custom_ops.py
        • __init__.py
      • tf_models
        • reduceModel.tflite
      • README.md
      • .gitignore
      • generate_rmean_model.ipynb
    • component_registration
      • backend_registration.ipynb
      • images
        • utensor-cli-components.drawio.svg
      • my_plugins
        • my_transformer.py
        • dummy_backend
          • __init__.py
        • __init__.py
        • txt_frontend.py
      • models
        • consts_model.txt
        • simple_graph.pb
      • transformer_registration.ipynb
      • README.md
      • .gitignore
      • frontend_registration.ipynb
    • README.md
  • Pipfile.lock
  • simple_graph.pb
  • utensor_cgen
    • matcher
      • _matcher_impl.py
      • __init__.py
      • _morphism.py
    • backend
      • transformer.py
      • graph_lower
        • __init__.py
        • alloc_plan.py
        • generic_graph_lower.py
      • utensor
        • snippets
          • composer.py
          • templates
            • containers
              • rearch
                • simple.cpp
                • time_slot.cpp
                • model_api.cpp
              • legacy
                • weight_header.hpp
                • get_ctx.cpp
            • snippets
              • rearch
                • simple.hpp
                • eval_op.cpp
                • construct_op.cpp
                • tensor_free.cpp
                • weight.hpp
                • op_missing.cpp
                • model_api.hpp
                • declare_rom_tensor.cpp
                • declare_op.cpp
                • declare_ram_tensor.cpp
              • legacy
                • add_op.cpp
                • fused_conv2d_maxpool_op.cpp
                • pack_op.cpp
                • softmax_op.cpp
                • create_tensor_new.cpp
                • qmax_pool_op.cpp
                • create_tensor_idx.cpp
                • quant_range_for_multiplication_op.cpp
                • qmatmul_op.cpp
                • matmul_op.cpp
                • dequantize_op.cpp
                • requant_op.cpp
                • qrelu_op.cpp
                • cmsis_uint8q7origin_op.cpp
                • argmax_op.cpp
                • qmul_op.cpp
                • requant_range_op.cpp
                • cmsis_nn_fc_op.cpp
                • shape_op.cpp
                • min_op.cpp
                • qadd_op.cpp
                • max_pool_op.cpp
                • weight_snippet.hpp
                • create_tensor_binary.cpp
                • strided_slice_op.cpp
                • get_ctx.hpp
                • quantV2_op.cpp
                • qconv2d_op.cpp
                • qreshape_op.cpp
                • comments.cpp
                • reshape_op.cpp
                • quantized_fused_conv2d_maxpool_op.cpp
                • conv2d_op.cpp
                • gather_op.cpp
                • relu_op.cpp
                • max_op.cpp
          • __init__.py
          • rearch
            • __init__.py
            • _snippets.py
          • template_env.py
          • legacy
            • __init__.py
            • _snippets.py
          • _base.py
          • _types.py
        • code_generator
          • __init__.py
          • rearch
            • _code_generator.py
            • _operators
              • _impls.py
              • __init__.py
              • _base.py
            • __init__.py
          • legacy
            • _code_generator.py
            • _operators.py
            • __init__.py
        • __init__.py
        • _backend_impl.py
        • _graph_lower
          • _op_lower.py
          • __init__.py
      • api.py
      • __init__.py
      • base.py
    • api
      • convert.py
      • export.py
      • __init__.py
      • utils.py
      • backend.py
    • cli
      • convert.py
      • __init__.py
      • main.py
      • utils.py
      • backend.py
    • transformer
      • quantize.py
      • optimizer.py
      • pipeline.py
      • ns_transformer.py
      • graph_viz.py
      • linear_reoder.py
      • conv_pool.py
      • __init__.py
      • base.py
    • ir
      • graph_builder.py
      • misc
        • graph_viz.py
        • __init__.py
      • instr.py
      • __init__.py
      • utils.py
      • converter.py
      • base.py
    • _extensions.py
    • logger.py
    • __main__.py
    • legalizer
      • tflite.py
      • api.py
      • onnx.py
      • __init__.py
      • tensorflow.py
      • base.py
    • __init__.py
    • utils.py
    • frontend
      • tflite_flatbuffer
        • ReverseV2Options.py
        • FakeQuantOptions.py
        • SVDFOptions.py
        • BatchToSpaceNDOptions.py
        • MatrixSetDiagOptions.py
        • DequantizeOptions.py
        • QuantizeOptions.py
        • Pool2DOptions.py
        • SliceOptions.py
        • CustomQuantization.py
        • FloorModOptions.py
        • SkipGramOptions.py
        • SpaceToBatchNDOptions.py
        • QuantizationDetails.py
        • ExpandDimsOptions.py
        • EmbeddingLookupSparseOptions.py
        • LocalResponseNormalizationOptions.py
        • MatrixDiagOptions.py
        • SquareOptions.py
        • PackOptions.py
        • GreaterEqualOptions.py
        • TileOptions.py
        • MirrorPadMode.py
        • Padding.py
        • ConcatenationOptions.py
        • Conv2DOptions.py
        • DivOptions.py
        • SparseToDenseOptions.py
        • Buffer.py
        • TensorType.py
        • BidirectionalSequenceRNNOptions.py
        • LSHProjectionOptions.py
        • GatherOptions.py
        • LogicalAndOptions.py
        • TopKV2Options.py
        • FloorDivOptions.py
        • L2NormOptions.py
        • CosOptions.py
        • MaximumMinimumOptions.py
        • UniqueOptions.py
        • UnidirectionalSequenceLSTMOptions.py
        • SequenceRNNOptions.py
        • LogicalOrOptions.py
        • FullyConnectedOptions.py
        • FillOptions.py
        • FullyConnectedOptionsWeightsFormat.py
        • CastOptions.py
        • ArgMinOptions.py
        • UnpackOptions.py
        • NegOptions.py
        • SubGraph.py
        • SplitVOptions.py
        • QuantizationParameters.py
        • NotEqualOptions.py
        • SelectOptions.py
        • BuiltinOperator.py
        • Operator.py
        • EqualOptions.py
        • CallOptions.py
        • BidirectionalSequenceLSTMOptions.py
        • ResizeNearestNeighborOptions.py
        • LSHProjectionType.py
        • LSTMKernelType.py
        • GatherNdOptions.py
        • AddOptions.py
        • DepthwiseConv2DOptions.py
        • SoftmaxOptions.py
        • PowOptions.py
        • ReshapeOptions.py
        • CustomOptionsFormat.py
        • __init__.py
        • PadOptions.py
        • ConcatEmbeddingsOptions.py
        • ArgMaxOptions.py
        • ActivationFunctionType.py
        • LessOptions.py
        • SqueezeOptions.py
        • AddNOptions.py
        • BuiltinOptions.py
        • ReducerOptions.py
        • Tensor.py
        • RankOptions.py
        • ResizeBilinearOptions.py
        • StridedSliceOptions.py
        • ReverseSequenceOptions.py
        • LogSoftmaxOptions.py
        • RangeOptions.py
        • SpaceToDepthOptions.py
        • PadV2Options.py
        • GreaterOptions.py
        • MulOptions.py
        • OperatorCode.py
        • ExpOptions.py
        • SubOptions.py
        • OneHotOptions.py
        • LeakyReluOptions.py
        • MirrorPadOptions.py
        • Model.py
        • LogicalNotOptions.py
        • LessEqualOptions.py
        • WhereOptions.py
        • AbsOptions.py
        • TransposeConvOptions.py
        • RNNOptions.py
        • CombinerType.py
        • LSTMOptions.py
        • TransposeOptions.py
        • SplitOptions.py
        • SquaredDifferenceOptions.py
        • ZerosLikeOptions.py
        • ShapeOptions.py
      • utensor.py
      • tflite.py
      • onnx.py
      • __init__.py
      • tensorflow.py
      • base.py
  • tests
    • test_utils
      • test_config.py
      • __init__.py
      • conftest.py
      • test_kwparser.py
    • matcher
      • imgs
    • test_backend
      • test_utensor.py
      • __init__.py
      • conftest.py
    • deep_cnn
      • .gitignore
    • test_frontend
      • test_tensorflow.py
      • __init__.py
      • test_selector.py
      • test_tflite.py
      • test_onnx.py
      • model_files
      • conftest.py
    • temp
    • test_transformer
      • test_id_remove
        • test_transformer.py
        • __init__.py
        • conftest.py
      • test_convpool
        • test_vgg.py
        • __init__.py
        • conftest.py
      • test_pipeline
        • test_pipeline_transformer.py
        • __init__.py
        • conftest.py
      • __init__.py
      • test_linear_reorder
        • test_transformer.py
        • __init__.py
        • conftest.py
      • test_refcnt
        • test_refcnt_optimizer.py
        • __init__.py
        • conftest.py
      • test_dropout
        • test_dropout_transormer.py
        • __init__.py
        • conftest.py
      • test_inline
        • test_inline_optimizer.py
        • __init__.py
        • conftest.py
    • test_cli.py
    • simple_graph.pb
    • deep_mlp
      • main.cpp
      • end_to_end.ipynb
      • readme_imgs
      • README.md
      • .gitignore
    • test_graph_constructor
      • test_ctx.py
      • __init__.py
      • test_ops
        • test_op_maxpool.py
        • test_op_add.py
        • test_op_relu.py
        • test_op_argmax.py
        • test_op_const.py
        • __init__.py
        • test_op_min.py
        • test_op_conv2d.py
        • test_op_max.py
        • test_op_matmul.py
        • conftest.py
    • test_matcher
      • test_permutation
        • test_match_patrn1.py
        • __init__.py
        • conftest.py
      • test_replace
        • __init__.py
        • test_replace.py
        • conftest.py
      • __init__.py
    • test_ir
      • test_AttrValueConverter
        • test_converter.py
        • __init__.py
        • conftest.py
      • test_TensorProtoConverter
        • test_converter.py
        • __init__.py
        • conftest.py
      • test_AttrListValueConverter
        • test_converter.py
        • __init__.py
        • conftest.py
      • test_uTensorGraph
        • test_graph.py
        • __init__.py
        • conftest.py
      • test_NameAttrListConverter
        • test_converter.py
        • __init__.py
        • conftest.py
      • test_DataTypeConverter
        • test_converter.py
        • __init__.py
      • __init__.py
  • requirements.txt
  • Dockerfile
  • .gitignore
  • MANIFEST.in
  • doc
    • Makefile
    • source
      • api.rst
      • _images
        • utensor-cli-components.drawio.svg
      • images.rst
      • utensor_cgen
        • ir
          • base.rst
          • converter.rst
        • matcher.rst
        • utils.rst
      • index.rst
      • conf.py
    • make.bat
    • README.rst
  • Dockerfile-ci

import numpy as np
import pytest
from tensorflow import make_tensor_proto
from tensorflow.core.framework import types_pb2

from utensor_cgen.ir.converter import TensorProtoConverter


@pytest.fixture(scope='session')
def generic_array():
    np_array = np.random.randn(3, 3).astype(np.float32)
    return TensorProtoConverter.__utensor_generic_type__(np_array=np_array)


@pytest.fixture(scope='session')
def tf_qint8_tensor():
    return make_tensor_proto(127*np.random.rand(3, 3),
                             types_pb2.DT_QINT8)

@pytest.fixture(scope='session')
def tf_quint8_tensor():
    return make_tensor_proto(255*np.random.rand(3, 3),
                             types_pb2.DT_QUINT8)