| /aosp_15_r20/external/tensorflow/tensorflow/lite/testing/op_tests/ |
| H A D | transpose_conv.py | 18 from tensorflow.lite.testing.zip_test_utils import create_tensor_data
147 filter_input = create_tensor_data(
150 filter_input = create_tensor_data(np.float32, filter_shape)
160 bias_input = create_tensor_data(
165 bias_input = create_tensor_data(np.float32,
170 mul_data = create_tensor_data(np.float32,
180 create_tensor_data(np.float32, input_shape),
181 create_tensor_data(np.float32, filter_shape)
186 create_tensor_data(
190 values = [create_tensor_data(np.float32, input_shape),]
|
| H A D | global_batch_norm.py | 17 from tensorflow.lite.testing.zip_test_utils import create_tensor_data
38 scale = create_tensor_data(parameters["dtype"], scale_shape)
39 offset = create_tensor_data(parameters["dtype"], scale_shape)
40 mean = create_tensor_data(parameters["dtype"], scale_shape)
41 variance = create_tensor_data(parameters["dtype"], scale_shape)
43 x = create_tensor_data(parameters["dtype"], parameters["input_shape"])
56 input_value = create_tensor_data(parameters["dtype"],
|
| H A D | fused_batch_norm.py | 19 from tensorflow.lite.testing.zip_test_utils import create_tensor_data
50 scale = create_tensor_data(parameters["dtype"], scale_shape)
51 offset = create_tensor_data(parameters["dtype"], scale_shape)
52 mean = create_tensor_data(parameters["dtype"], scale_shape)
53 variance = create_tensor_data(parameters["dtype"], scale_shape)
82 create_tensor_data(parameters["dtype"], input_shape),
83 create_tensor_data(parameters["dtype"], input_shape)
|
| H A D | one_hot.py | 17 from tensorflow.lite.testing.zip_test_utils import create_tensor_data
62 create_tensor_data(
67 create_tensor_data(tf.int32, shape=None, min_value=1, max_value=10),
72 create_tensor_data(
75 create_tensor_data(
|
| H A D | gather.py | 17 from tensorflow.lite.testing.zip_test_utils import create_tensor_data
76 params = create_tensor_data(parameters["params_dtype"],
86 indices = create_tensor_data(
107 params = create_tensor_data(
114 indices = create_tensor_data(
|
| H A D | conv_bias_activation.py | 18 from tensorflow.lite.testing.zip_test_utils import create_tensor_data
76 bias_input = create_tensor_data(np.float32, (filter_shape[-1],))
86 filter_input = create_tensor_data(
106 filter_input_2 = create_tensor_data(
127 create_tensor_data(
|
| H A D | identify_dilated_conv.py | 18 from tensorflow.lite.testing.zip_test_utils import create_tensor_data
57 filter_input = create_tensor_data(
81 create_tensor_data(np.float32, input_shape, min_value=-1, max_value=1)
84 values.append(create_tensor_data(np.float32, filter_shape))
|
| H A D | binary_op.py | 17 from tensorflow.lite.testing.zip_test_utils import create_tensor_data
222 input1 = create_tensor_data(
224 input2 = create_tensor_data(
227 input1 = create_tensor_data(parameters["dtype"], input_shape_1)
228 input2 = create_tensor_data(parameters["dtype"], input_shape_2)
|
| H A D | depthwiseconv.py | 18 from tensorflow.lite.testing.zip_test_utils import create_tensor_data
113 filter_input = create_tensor_data(np.float32, filter_shape)
139 create_tensor_data(np.float32, input_shape, min_value=-1, max_value=1)
143 create_tensor_data(
|
| H A D | conv.py | 18 from tensorflow.lite.testing.zip_test_utils import create_tensor_data
115 filter_input = create_tensor_data(
137 create_tensor_data(np.float32, input_shape, min_value=-1, max_value=1)
140 values.append(create_tensor_data(np.float32, filter_shape))
|
| H A D | conv_activation.py | 18 from tensorflow.lite.testing.zip_test_utils import create_tensor_data
90 filter_input = create_tensor_data(
113 create_tensor_data(
117 values.append(create_tensor_data(np.float32, filter_shape))
|
| H A D | lstm.py | 17 from tensorflow.lite.testing.zip_test_utils import create_tensor_data
70 kernel_values = create_tensor_data(parameters["dtype"],
73 bias_values = create_tensor_data(parameters["dtype"], [bias.shape[0]], 0,
82 tensor_data = create_tensor_data(parameters["dtype"],
|
| H A D | static_rnn_with_control_flow_v2.py | 17 from tensorflow.lite.testing.zip_test_utils import create_tensor_data
77 kernel_values = create_tensor_data(parameters["dtype"],
80 bias_values = create_tensor_data(parameters["dtype"], [bias.shape[0]], 0,
89 tensor_data = create_tensor_data(parameters["dtype"],
|
| H A D | fully_connected.py | 18 from tensorflow.lite.testing.zip_test_utils import create_tensor_data
147 input_tensor2 = create_tensor_data(
172 create_tensor_data(
177 create_tensor_data(
|
| H A D | batchmatmul.py | 17 from tensorflow.lite.testing.zip_test_utils import create_tensor_data
60 data = create_tensor_data(
86 input0_value = create_tensor_data(
95 input1_value = create_tensor_data(
|
| H A D | where_v2.py | 17 from tensorflow.lite.testing.zip_test_utils import create_tensor_data
123 input_condition = create_tensor_data(parameters["condition_dtype"],
128 input_value1 = create_tensor_data(
131 input_value2 = create_tensor_data(
|
| H A D | unsorted_segment.py | 18 from tensorflow.lite.testing.zip_test_utils import create_tensor_data
141 data_value = create_tensor_data(
147 data_value = create_tensor_data(
149 segment_id_value = create_tensor_data(
|
| H A D | atan2.py | 17 from tensorflow.lite.testing.zip_test_utils import create_tensor_data
45 input_value1 = create_tensor_data(parameters["input_dtype"],
47 input_value2 = create_tensor_data(parameters["input_dtype"],
|
| H A D | depth_to_space.py | 17 from tensorflow.lite.testing.zip_test_utils import create_tensor_data
49 input_values = create_tensor_data(parameters["dtype"],
52 input_values = create_tensor_data(
|
| H A D | maximum.py | 17 from tensorflow.lite.testing.zip_test_utils import create_tensor_data
52 create_tensor_data(
57 create_tensor_data(
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/mlir/testing/op_tests/ |
| H A D | conv_bias_activation.py | 22 from tensorflow.lite.testing.zip_test_utils import create_tensor_data
80 bias_input = create_tensor_data(np.float32, (filter_shape[-1],))
90 filter_input = create_tensor_data(
110 filter_input_2 = create_tensor_data(
131 create_tensor_data(
|
| H A D | identify_dilated_conv.py | 22 from tensorflow.lite.testing.zip_test_utils import create_tensor_data
61 filter_input = create_tensor_data(
85 create_tensor_data(np.float32, input_shape, min_value=-1, max_value=1)
88 values.append(create_tensor_data(np.float32, filter_shape))
|
| H A D | where_v2.py | 21 from tensorflow.lite.testing.zip_test_utils import create_tensor_data
91 input_condition = create_tensor_data(tf.bool,
93 input_value1 = create_tensor_data(parameters["input_dtype"],
95 input_value2 = create_tensor_data(parameters["input_dtype"],
|
| H A D | static_rnn_with_control_flow_v2.py | 21 from tensorflow.lite.testing.zip_test_utils import create_tensor_data
81 kernel_values = create_tensor_data(parameters["dtype"],
84 bias_values = create_tensor_data(parameters["dtype"], [bias.shape[0]], 0,
93 tensor_data = create_tensor_data(parameters["dtype"],
|
| H A D | batchmatmul.py | 21 from tensorflow.lite.testing.zip_test_utils import create_tensor_data
62 data = create_tensor_data(
88 input0_value = create_tensor_data(
97 input1_value = create_tensor_data(
|