/external/tensorflow/tensorflow/core/kernels/ |
D | non_max_suppression_op_test.cc | 46 AddInputFromArray<float>( in TEST_F() 50 AddInputFromArray<float>(TensorShape({6}), {.9f, .75f, .6f, .95f, .5f, .3f}); in TEST_F() 51 AddInputFromArray<int>(TensorShape({}), {3}); in TEST_F() 61 AddInputFromArray<float>(TensorShape({6, 4}), in TEST_F() 64 AddInputFromArray<float>(TensorShape({6}), {.9f, .75f, .6f, .95f, .5f, .3f}); in TEST_F() 65 AddInputFromArray<int>(TensorShape({}), {3}); in TEST_F() 75 AddInputFromArray<float>( in TEST_F() 79 AddInputFromArray<float>(TensorShape({6}), {.9f, .75f, .6f, .95f, .5f, .3f}); in TEST_F() 80 AddInputFromArray<int>(TensorShape({}), {2}); in TEST_F() 90 AddInputFromArray<float>( in TEST_F() [all …]
|
D | crop_and_resize_op_test.cc | 51 AddInputFromArray<T>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); \ 52 AddInputFromArray<float>(TensorShape({1, 4}), {0, 0, 1, 1}); \ 53 AddInputFromArray<int32>(TensorShape({1}), {0}); \ 54 AddInputFromArray<int32>(TensorShape({2}), {1, 1}); \ 78 AddInputFromArray<uint8>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in REGISTER_TEST() 79 AddInputFromArray<float>(TensorShape({1, 4}), {0, 0, 1, 1}); in REGISTER_TEST() 80 AddInputFromArray<int32>(TensorShape({1}), {0}); in REGISTER_TEST() 81 AddInputFromArray<int32>(TensorShape({2}), {1, 1}); in REGISTER_TEST() 94 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_F() 95 AddInputFromArray<float>(TensorShape({1, 4}), {1, 1, 0, 0}); in TEST_F() [all …]
|
D | roll_op_test.cc | 55 AddInputFromArray<float>(TensorShape({5}), {0, 1, 2, 3, 4}); in TEST_F() 56 AddInputFromArray<int32>(TensorShape({}), {3}); in TEST_F() 57 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F() 70 AddInputFromArray<string>(TensorShape({5}), {"a", "b", "c", "d", "e"}); in TEST_F() 71 AddInputFromArray<int32>(TensorShape({}), {3}); in TEST_F() 72 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F() 85 AddInputFromArray<std::complex<float>>( in TEST_F() 89 AddInputFromArray<int32>(TensorShape({}), {3}); in TEST_F() 90 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F() 106 AddInputFromArray<float>(TensorShape({3, 5}), in TEST_F() [all …]
|
D | dynamic_stitch_op_test.cc | 51 AddInputFromArray<int32>(TensorShape({3}), {0, 4, 7}); in TEST_F() 52 AddInputFromArray<int32>(TensorShape({5}), {1, 6, 2, 3, 5}); in TEST_F() 53 AddInputFromArray<float>(TensorShape({3}), {0, 40, 70}); in TEST_F() 54 AddInputFromArray<float>(TensorShape({5}), {10, 60, 20, 30, 50}); in TEST_F() 67 AddInputFromArray<int32>(TensorShape({3}), {0, 4, 7}); in TEST_F() 68 AddInputFromArray<int32>(TensorShape({2}), {1, 6}); in TEST_F() 69 AddInputFromArray<int32>(TensorShape({3}), {2, 3, 5}); in TEST_F() 70 AddInputFromArray<float>(TensorShape({3, 2}), {0, 1, 40, 41, 70, 71}); in TEST_F() 71 AddInputFromArray<float>(TensorShape({2, 2}), {10, 11, 60, 61}); in TEST_F() 72 AddInputFromArray<float>(TensorShape({3, 2}), {20, 21, 30, 31, 50, 51}); in TEST_F() [all …]
|
D | quantize_and_dequantize_op_test.cc | 51 AddInputFromArray<float>(TensorShape({1}), {-3.5}); in TEST_F() 52 AddInputFromArray<float>(TensorShape({}), {0.0}); // Min in TEST_F() 53 AddInputFromArray<float>(TensorShape({}), {0.0}); // Max in TEST_F() 76 AddInputFromArray<float>(TensorShape({1}), {-3.5}); in TEST_F() 77 AddInputFromArray<float>(TensorShape({}), {0.0}); // Min in TEST_F() 78 AddInputFromArray<float>(TensorShape({}), {0.0}); // Max in TEST_F() 79 AddInputFromArray<int32>(TensorShape({}), {8}); // num_bits in TEST_F() 103 AddInputFromArray<float>(TensorShape({6}), {-1, -0.5, 0, 0.3, 0.8, 0.555}); in TEST_F() 104 AddInputFromArray<float>(TensorShape({}), {0.0}); // Min in TEST_F() 105 AddInputFromArray<float>(TensorShape({}), {0.0}); // Max in TEST_F() [all …]
|
D | scatter_op_test.cc | 52 AddInputFromArray<string>(TensorShape({1}), {"Brain"}); in TEST_F() 53 AddInputFromArray<int32>(TensorShape({1}), {0}); in TEST_F() 54 AddInputFromArray<string>(TensorShape({1}), {"TensorFlow"}); in TEST_F() 65 AddInputFromArray<bool>(TensorShape({1}), {false}); in TEST_F() 66 AddInputFromArray<int32>(TensorShape({1}), {0}); in TEST_F() 67 AddInputFromArray<bool>(TensorShape({1}), {true}); in TEST_F() 80 AddInputFromArray<float>(TensorShape({5, 3}), in TEST_F() 82 AddInputFromArray<int32>(TensorShape({3}), {0, 4, 2}); in TEST_F() 83 AddInputFromArray<float>(TensorShape({3, 3}), in TEST_F() 99 AddInputFromArray<float>(TensorShape({5, 3}), in TEST_F() [all …]
|
D | quantized_matmul_op_test.cc | 58 AddInputFromArray<quint8>(TensorShape({2, 3}), {1, 2, 3, 4, 5, 6}); in TEST_F() 63 AddInputFromArray<quint8>(TensorShape({3, 4}), in TEST_F() 65 AddInputFromArray<float>(TensorShape({1}), {0}); in TEST_F() 66 AddInputFromArray<float>(TensorShape({1}), {255.0f}); in TEST_F() 67 AddInputFromArray<float>(TensorShape({1}), {0}); in TEST_F() 68 AddInputFromArray<float>(TensorShape({1}), {255.0f}); in TEST_F() 116 AddInputFromArray<quint8>(TensorShape({a_rows, a_cols}), {11}); in TEST_F() 120 AddInputFromArray<quint8>(TensorShape({b_rows, b_cols}), {0}); in TEST_F() 121 AddInputFromArray<float>(TensorShape({1}), {-12.0f}); in TEST_F() 122 AddInputFromArray<float>(TensorShape({1}), {243.0f}); in TEST_F() [all …]
|
D | resize_nearest_neighbor_op_test.cc | 62 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_F() 63 AddInputFromArray<int32>(TensorShape({2}), {1, 1}); in TEST_F() 80 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_F() 81 AddInputFromArray<int32>(TensorShape({2}), {1, 1}); in TEST_F() 97 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_F() 98 AddInputFromArray<int32>(TensorShape({2}), {3, 3}); in TEST_F() 118 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_F() 119 AddInputFromArray<int32>(TensorShape({2}), {3, 3}); in TEST_F() 139 AddInputFromArray<float>(TensorShape({1, 3, 3, 1}), in TEST_F() 141 AddInputFromArray<int32>(TensorShape({2}), {2, 2}); in TEST_F() [all …]
|
D | scatter_nd_op_test.cc | 81 AddInputFromArray<float>(TensorShape({5, 3}), in TEST_F() 83 AddInputFromArray<int32>(TensorShape({3, 1}), {0, 4, 2}); in TEST_F() 84 AddInputFromArray<float>(TensorShape({3, 3}), in TEST_F() 100 AddInputFromArray<float>(TensorShape({5, 3}), in TEST_F() 102 AddInputFromArray<int64>(TensorShape({3, 1}), {0, 4, 2}); in TEST_F() 103 AddInputFromArray<float>(TensorShape({3, 3}), in TEST_F() 132 AddInputFromArray<float>(TensorShape({5}), {0, 0, 0, 0, 0}); in TEST_F() 133 AddInputFromArray<int32>(TensorShape({1}), {3}); in TEST_F() 134 AddInputFromArray<float>(TensorShape({1}), {101}); in TEST_F() 148 AddInputFromArray<float>(TensorShape({5}), {0, 0, 0, 0, 0}); in TEST_F() [all …]
|
D | requantize_op_test.cc | 54 AddInputFromArray<qint32>(TensorShape({value_count}), in TEST_F() 56 AddInputFromArray<float>(TensorShape({1}), {-256.0f}); in TEST_F() 57 AddInputFromArray<float>(TensorShape({1}), {256.0f}); in TEST_F() 58 AddInputFromArray<float>(TensorShape({1}), {-1.0f}); in TEST_F() 59 AddInputFromArray<float>(TensorShape({1}), {1.0f}); in TEST_F() 72 AddInputFromArray<qint32>(TensorShape({value_count}), in TEST_F() 74 AddInputFromArray<float>(TensorShape({1}), {-256.0f}); in TEST_F() 75 AddInputFromArray<float>(TensorShape({1}), {256.0f}); in TEST_F() 76 AddInputFromArray<float>(TensorShape({1}), {0.01f}); in TEST_F() 77 AddInputFromArray<float>(TensorShape({1}), {1.0f}); in TEST_F() [all …]
|
D | sparse_to_dense_op_test.cc | 56 AddInputFromArray<int32>(TensorShape({3}), {1, 3, 4}); in TEST_F() 58 AddInputFromArray<int32>(TensorShape({1}), {5}); in TEST_F() 60 AddInputFromArray<float>(TensorShape({}), {2}); in TEST_F() 62 AddInputFromArray<float>(TensorShape({}), {-2}); in TEST_F() 75 AddInputFromArray<int64>(TensorShape({3}), {1, 3, 4}); in TEST_F() 77 AddInputFromArray<int64>(TensorShape({1}), {5}); in TEST_F() 79 AddInputFromArray<double>(TensorShape({}), {2}); in TEST_F() 81 AddInputFromArray<double>(TensorShape({}), {-2}); in TEST_F() 94 AddInputFromArray<int32>({3}, {1, 3, 4}); in TEST_F() 96 AddInputFromArray<int32>({1}, {5}); in TEST_F() [all …]
|
D | quantized_batch_norm_op_test.cc | 98 AddInputFromArray<quint8>(input_quantized.shape(), in TEST_F() 100 AddInputFromArray<float>(TensorShape({1}), {input_min}); in TEST_F() 101 AddInputFromArray<float>(TensorShape({1}), {input_max}); in TEST_F() 102 AddInputFromArray<quint8>(mean_quantized.shape(), in TEST_F() 104 AddInputFromArray<float>(TensorShape({1}), {mean_min}); in TEST_F() 105 AddInputFromArray<float>(TensorShape({1}), {mean_max}); in TEST_F() 106 AddInputFromArray<quint8>(variance_quantized.shape(), in TEST_F() 108 AddInputFromArray<float>(TensorShape({1}), {variance_min}); in TEST_F() 109 AddInputFromArray<float>(TensorShape({1}), {variance_max}); in TEST_F() 110 AddInputFromArray<quint8>(beta_quantized.shape(), in TEST_F() [all …]
|
D | quantized_conv_ops_test.cc | 90 AddInputFromArray<quint8>(image_quantized.shape(), in TEST_F() 92 AddInputFromArray<quint8>(filter_quantized.shape(), in TEST_F() 94 AddInputFromArray<float>(TensorShape({1}), {image_min}); in TEST_F() 95 AddInputFromArray<float>(TensorShape({1}), {image_max}); in TEST_F() 96 AddInputFromArray<float>(TensorShape({1}), {filter_min}); in TEST_F() 97 AddInputFromArray<float>(TensorShape({1}), {filter_max}); in TEST_F() 153 AddInputFromArray<quint8>( in TEST_F() 158 AddInputFromArray<quint8>( in TEST_F() 161 AddInputFromArray<float>(TensorShape({1}), {0}); in TEST_F() 162 AddInputFromArray<float>(TensorShape({1}), {255.0f}); in TEST_F() [all …]
|
D | gather_op_test.cc | 56 AddInputFromArray<float>(TensorShape({5}), {0, 1, 2, 3, 4}); in TEST_F() 57 AddInputFromArray<int32>(TensorShape({}), {3}); in TEST_F() 58 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F() 71 AddInputFromArray<std::complex<float>>( in TEST_F() 75 AddInputFromArray<int32>(TensorShape({}), {3}); in TEST_F() 76 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F() 90 AddInputFromArray<float>(TensorShape({5, 3}), in TEST_F() 92 AddInputFromArray<int32>(TensorShape({4}), {0, 4, 0, 2}); in TEST_F() 93 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F() 106 AddInputFromArray<float>(TensorShape({5, 3}), in TEST_F() [all …]
|
D | quantize_op_test.cc | 40 AddInputFromArray<float>(TensorShape({7}), in TEST_F() 43 AddInputFromArray<float>(TensorShape({1}), {0}); in TEST_F() 45 AddInputFromArray<float>(TensorShape({1}), {255.0f}); in TEST_F() 63 AddInputFromArray<float>(TensorShape({8}), in TEST_F() 65 AddInputFromArray<float>(TensorShape({1}), {-255.0f}); in TEST_F() 66 AddInputFromArray<float>(TensorShape({1}), {127.0f}); in TEST_F() 95 AddInputFromArray<float>(TensorShape({3}), {-1.0, 0.0, 2.0}); in TEST_F() 96 AddInputFromArray<float>(TensorShape({1}), {-1.0f}); in TEST_F() 97 AddInputFromArray<float>(TensorShape({1}), {2.0f}); in TEST_F() 125 AddInputFromArray<float>(TensorShape({7}), in TEST_F() [all …]
|
D | sequence_ops_test.cc | 59 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F() 60 AddInputFromArray<int32>(TensorShape({}), {10}); in TEST_F() 61 AddInputFromArray<int32>(TensorShape({}), {2}); in TEST_F() 74 AddInputFromArray<float>(TensorShape({}), {0.5}); in TEST_F() 75 AddInputFromArray<float>(TensorShape({}), {2}); in TEST_F() 76 AddInputFromArray<float>(TensorShape({}), {0.3}); in TEST_F() 89 AddInputFromArray<double>(TensorShape({}), {0.0}); in TEST_F() 90 AddInputFromArray<double>(TensorShape({}), {10000}); in TEST_F() 91 AddInputFromArray<double>(TensorShape({}), {0.5}); in TEST_F() 106 AddInputFromArray<float>(TensorShape({}), {3.0}); in TEST_F() [all …]
|
D | resize_bilinear_op_test.cc | 115 AddInputFromArray<int32>(TensorShape({2}), {output_width, output_height}); in TestResize() 171 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_F() 172 AddInputFromArray<int32>(TensorShape({2}), {1, 1}); in TEST_F() 184 AddInputFromArray<int32>(TensorShape({2}), {1, 1}); in TEST_F() 203 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_F() 204 AddInputFromArray<int32>(TensorShape({2}), {1, 1}); in TEST_F() 218 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_F() 219 AddInputFromArray<int32>(TensorShape({2}), {3, 3}); in TEST_F() 238 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_F() 239 AddInputFromArray<int32>(TensorShape({2}), {3, 3}); in TEST_F() [all …]
|
D | logging_ops_test.cc | 45 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6}); in TEST_F() 46 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6}); in TEST_F() 47 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6}); in TEST_F() 56 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6}); in TEST_F() 57 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6}); in TEST_F() 58 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6}); in TEST_F() 67 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6}); in TEST_F() 68 AddInputFromArray<string>(TensorShape({}), {"foo"}); in TEST_F() 69 AddInputFromArray<string>(TensorShape({}), {"bar"}); in TEST_F() 78 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6}); in TEST_F() [all …]
|
D | fused_batch_norm_op_test.cc | 43 AddInputFromArray<float>(TensorShape({1, 1, 6, 2}), in TEST_F() 45 AddInputFromArray<float>(TensorShape({2}), {4.0, 4.0}); in TEST_F() 46 AddInputFromArray<float>(TensorShape({2}), {2.0, 2.0}); in TEST_F() 47 AddInputFromArray<float>(TensorShape({0}), {}); in TEST_F() 48 AddInputFromArray<float>(TensorShape({0}), {}); in TEST_F() 77 AddInputFromArray<float>(TensorShape({1, 1, 6, 2}), in TEST_F() 79 AddInputFromArray<float>(TensorShape({2}), {4.0, 4.0}); in TEST_F() 80 AddInputFromArray<float>(TensorShape({2}), {2.0, 2.0}); in TEST_F() 81 AddInputFromArray<float>(TensorShape({2}), {10, 10}); in TEST_F() 82 AddInputFromArray<float>(TensorShape({2}), {11.67f, 11.67f}); in TEST_F() [all …]
|
D | sparse_add_op_test.cc | 70 AddInputFromArray<int64>(indices_shape, indices); \ in TEST_F() 71 AddInputFromArray<float>(TensorShape({4}), {1, 2, 3, 4}); \ in TEST_F() 72 AddInputFromArray<int64>(TensorShape({2}), shape); in TEST_F() 76 AddInputFromArray<float>(TensorShape({}), {0.0}); in TEST_F() 109 AddInputFromArray<int64>(indices_shape, indices); \ 110 AddInputFromArray<VALTYPE>(TensorShape({4}), {1, 2, 3, 4}); \ 111 AddInputFromArray<int64>(TensorShape({2}), shape); \ 113 AddInputFromArray<int64>(TensorShape({2, 2}), {0, 0, 1, 1}); \ 114 AddInputFromArray<VALTYPE>(TensorShape({2}), {5, 6}); \ 115 AddInputFromArray<int64>(TensorShape({2}), shape); \ [all …]
|
D | fake_quant_ops_test.cc | 66 AddInputFromArray<float>(shape, data); in RunTestFakeQuantWithMinMaxArgs() 91 AddInputFromArray<float>(shape, data); in RunTestFakeQuantWithMinMaxVars() 93 AddInputFromArray<float>(TensorShape({}), {min}); in RunTestFakeQuantWithMinMaxVars() 95 AddInputFromArray<float>(TensorShape({}), {max}); in RunTestFakeQuantWithMinMaxVars() 121 AddInputFromArray<float>(shape, data); in RunTestFakeQuantWithMinMaxVarsPerChannel() 123 AddInputFromArray<float>(minmax_shape, min); in RunTestFakeQuantWithMinMaxVarsPerChannel() 125 AddInputFromArray<float>(minmax_shape, max); in RunTestFakeQuantWithMinMaxVarsPerChannel() 372 AddInputFromArray<float>(TensorShape({2, 3}), in TEST_F() 402 AddInputFromArray<float>(TensorShape({2, 3}), in TEST_F() 433 AddInputFromArray<float>(TensorShape({2, 3}), in TEST_F() [all …]
|
D | sparse_dense_binary_op_shared_test.cc | 74 AddInputFromArray<int64>(TensorShape({1, 1}), {0}); // indices in TEST_F() 75 AddInputFromArray<float>(TensorShape({1}), {1618}); // values in TEST_F() 76 AddInputFromArray<int64>(TensorShape({1}), {1}); // shape in TEST_F() 77 AddInputFromArray<float>(TensorShape({2, 1}), {17, 19}); // dense in TEST_F() 85 AddInputFromArray<int64>(TensorShape({1, 2}), {0, 0}); in TEST_F() 86 AddInputFromArray<float>(TensorShape({1}), {1618}); in TEST_F() 87 AddInputFromArray<int64>(TensorShape({2}), {1, 1}); in TEST_F() 88 AddInputFromArray<float>(TensorShape({2, 1}), {17, 19}); in TEST_F() 109 AddInputFromArray<int64>(indices_shape, indices); in TEST_F() 110 AddInputFromArray<float>(TensorShape({4}), {1, 2, 3, 4}); in TEST_F() [all …]
|
D | summary_op_test.cc | 62 AddInputFromArray<string>(TensorShape({3}), {"tag1", "tag2", "tag3"}); in TEST_F() 63 AddInputFromArray<float>(TensorShape({3}), {1.0f, -0.73f, 10000.0f}); in TEST_F() 82 AddInputFromArray<string>(TensorShape({3}), {"tag1", "tag2", "tag3"}); in TEST_F() 83 AddInputFromArray<double>(TensorShape({3}), {1.0, -0.73, 10000.0}); in TEST_F() 122 AddInputFromArray<string>(TensorShape({2}), {"tag1", "tag2"}); in TEST_F() 123 AddInputFromArray<float>(TensorShape({3}), {1.0f, -0.73f, 10000.0f}); in TEST_F() 132 AddInputFromArray<string>(TensorShape({2, 1}), {"tag1", "tag2"}); in TEST_F() 133 AddInputFromArray<float>(TensorShape({2}), {1.0f, -0.73f}); in TEST_F() 144 AddInputFromArray<string>(TensorShape({2}), {"tag1", "tag2"}); in TEST_F() 145 AddInputFromArray<float>(TensorShape({2, 1}), {1.0f, -0.73f}); in TEST_F() [all …]
|
D | quantized_concat_op_test.cc | 100 AddInputFromArray<int32>(TensorShape({}), {0}); in TestSmall8Bit() 101 AddInputFromArray<quint8>(first_quantized.shape(), in TestSmall8Bit() 103 AddInputFromArray<quint8>(second_quantized.shape(), in TestSmall8Bit() 105 AddInputFromArray<float>(TensorShape({}), {first_min}); in TestSmall8Bit() 106 AddInputFromArray<float>(TensorShape({}), {second_min}); in TestSmall8Bit() 107 AddInputFromArray<float>(TensorShape({}), {first_max}); in TestSmall8Bit() 108 AddInputFromArray<float>(TensorShape({}), {second_max}); in TestSmall8Bit() 166 AddInputFromArray<int32>(TensorShape({}), {0}); in TestSmall32Bit() 167 AddInputFromArray<qint32>(first_quantized.shape(), in TestSmall32Bit() 169 AddInputFromArray<qint32>(second_quantized.shape(), in TestSmall32Bit() [all …]
|
D | quantized_bias_add_op_test.cc | 73 AddInputFromArray<quint8>(input_quantized.shape(), in TEST_F() 75 AddInputFromArray<quint8>(bias_quantized.shape(), in TEST_F() 77 AddInputFromArray<float>(TensorShape({1}), {input_min}); in TEST_F() 78 AddInputFromArray<float>(TensorShape({1}), {input_max}); in TEST_F() 79 AddInputFromArray<float>(TensorShape({1}), {bias_min}); in TEST_F() 80 AddInputFromArray<float>(TensorShape({1}), {bias_max}); in TEST_F() 155 AddInputFromArray<quint8>(input_quantized.shape(), in TEST_F() 157 AddInputFromArray<quint8>(bias_quantized.shape(), in TEST_F() 159 AddInputFromArray<float>(TensorShape({1}), {input_min}); in TEST_F() 160 AddInputFromArray<float>(TensorShape({1}), {input_max}); in TEST_F() [all …]
|