• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Copyright 2019 The TensorFlow Authors. All Rights Reserved.
2#
3# Licensed under the Apache License, Version 2.0 (the "License");
4# you may not use this file except in compliance with the License.
5# You may obtain a copy of the License at
6#
7#     http://www.apache.org/licenses/LICENSE-2.0
8#
9# Unless required by applicable law or agreed to in writing, software
10# distributed under the License is distributed on an "AS IS" BASIS,
11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12# See the License for the specific language governing permissions and
13# limitations under the License.
14# ==============================================================================
15"""Test configs for constant ops."""
16from __future__ import absolute_import
17from __future__ import division
18from __future__ import print_function
19
20import numpy as np
21import tensorflow.compat.v1 as tf
22from tensorflow.lite.testing.zip_test_utils import create_tensor_data
23from tensorflow.lite.testing.zip_test_utils import make_zip_of_tests
24from tensorflow.lite.testing.zip_test_utils import register_make_test_function
25from tensorflow.lite.testing.zip_test_utils import TF_TYPE_INFO
26
27
28# This function tests various TensorFLow functions that generates Const op,
29# including `tf.ones`, `tf.zeros` and random functions.
30@register_make_test_function()
31def make_constant_tests(options):
32  """Make a set of tests to do constant ops."""
33
34  test_parameters = [{
35      "dtype": [tf.float32, tf.int32],
36      "input_shape": [[], [1], [2], [1, 1, 1, 1], [2, 2, 2, 2]],
37      "constant_is_also_output": [True, False],
38      # This is a regression test for a bug where Toco rejects models with
39      # unread inputs.
40      "has_unread_input": [True, False],
41  }]
42
43  def build_graph(parameters):
44    """Build a constant graph given `parameters`."""
45    dummy_input = tf.compat.v1.placeholder(
46        dtype=parameters["dtype"],
47        name="input1",
48        shape=parameters["input_shape"])
49    constant = tf.constant(
50        create_tensor_data(parameters["dtype"], parameters["input_shape"]))
51    outputs = [tf.maximum(dummy_input, constant)]
52    if parameters["constant_is_also_output"]:
53      outputs.append(constant)
54    inputs = [dummy_input]
55    if parameters["has_unread_input"]:
56      unread_input = tf.compat.v1.placeholder(
57          dtype=parameters["dtype"],
58          name="unread_input",
59          shape=parameters["input_shape"])
60      inputs.append(unread_input)
61
62    return inputs, outputs
63
64  def build_inputs(parameters, sess, inputs, outputs):
65    dummy_input = np.zeros(
66        parameters["input_shape"], dtype=TF_TYPE_INFO[parameters["dtype"]][0])
67    return [dummy_input], sess.run(outputs, feed_dict={inputs[0]: dummy_input})
68
69  make_zip_of_tests(options, test_parameters, build_graph, build_inputs)
70