• 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."""
16import numpy as np
17import tensorflow.compat.v1 as tf
18
19from tensorflow.lite.testing.zip_test_utils import create_tensor_data
20from tensorflow.lite.testing.zip_test_utils import make_zip_of_tests
21from tensorflow.lite.testing.zip_test_utils import MAP_TF_TO_NUMPY_TYPE
22from tensorflow.lite.testing.zip_test_utils import register_make_test_function
23
24
25# This function tests various TensorFLow functions that generates Const op,
26# including `tf.ones`, `tf.zeros` and random functions.
27@register_make_test_function()
28def make_constant_tests(options):
29  """Make a set of tests to do constant ops."""
30
31  test_parameters = [{
32      "dtype": [tf.float32, tf.int32],
33      "input_shape": [[], [1], [2], [1, 1, 1, 1], [2, 2, 2, 2]],
34      "constant_is_also_output": [True, False],
35      # Models should not be rejected regardless whether it has unread inputs.
36      "has_unread_input": [True, False],
37  }]
38
39  def build_graph(parameters):
40    """Build a constant graph given `parameters`."""
41    dummy_input = tf.compat.v1.placeholder(
42        dtype=parameters["dtype"],
43        name="input1",
44        shape=parameters["input_shape"])
45    constant = tf.constant(
46        create_tensor_data(parameters["dtype"], parameters["input_shape"]))
47    outputs = [tf.maximum(dummy_input, constant)]
48    if parameters["constant_is_also_output"]:
49      outputs.append(constant)
50    inputs = [dummy_input]
51    if parameters["has_unread_input"]:
52      unread_input = tf.compat.v1.placeholder(
53          dtype=parameters["dtype"],
54          name="unread_input",
55          shape=parameters["input_shape"])
56      inputs.append(unread_input)
57
58    return inputs, outputs
59
60  def build_inputs(parameters, sess, inputs, outputs):
61    dummy_input = np.zeros(
62        parameters["input_shape"],
63        dtype=MAP_TF_TO_NUMPY_TYPE[parameters["dtype"]])
64    return [dummy_input], sess.run(outputs, feed_dict={inputs[0]: dummy_input})
65
66  make_zip_of_tests(options, test_parameters, build_graph, build_inputs)
67