| /external/tensorflow/tensorflow/core/kernels/ |
| dynamic_stitch_op_test.cc | 51 AddInputFromArray<int32>(TensorShape({3}), {0, 4, 7});
52 AddInputFromArray<int32>(TensorShape({5}), {1, 6, 2, 3, 5});
53 AddInputFromArray<float>(TensorShape({3}), {0, 40, 70});
54 AddInputFromArray<float>(TensorShape({5}), {10, 60, 20, 30, 50});
67 AddInputFromArray<int32>(TensorShape({3}), {0, 4, 7});
68 AddInputFromArray<int32>(TensorShape({2}), {1, 6});
69 AddInputFromArray<int32>(TensorShape({3}), {2, 3, 5});
70 AddInputFromArray<float>(TensorShape({3, 2}), {0, 1, 40, 41, 70, 71});
71 AddInputFromArray<float>(TensorShape({2, 2}), {10, 11, 60, 61});
72 AddInputFromArray<float>(TensorShape({3, 2}), {20, 21, 30, 31, 50, 51})
[all...] |
| 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});
79 AddInputFromArray<float>(TensorShape({1, 4}), {0, 0, 1, 1});
80 AddInputFromArray<int32>(TensorShape({1}), {0});
81 AddInputFromArray<int32>(TensorShape({2}), {1, 1});
94 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4});
95 AddInputFromArray<float>(TensorShape({1, 4}), {1, 1, 0, 0})
[all...] |
| requantize_op_test.cc | 54 AddInputFromArray<qint32>(TensorShape({value_count}),
56 AddInputFromArray<float>(TensorShape({1}), {-256.0f});
57 AddInputFromArray<float>(TensorShape({1}), {256.0f});
58 AddInputFromArray<float>(TensorShape({1}), {-1.0f});
59 AddInputFromArray<float>(TensorShape({1}), {1.0f});
72 AddInputFromArray<qint32>(TensorShape({value_count}),
74 AddInputFromArray<float>(TensorShape({1}), {-256.0f});
75 AddInputFromArray<float>(TensorShape({1}), {256.0f});
76 AddInputFromArray<float>(TensorShape({1}), {0.01f});
77 AddInputFromArray<float>(TensorShape({1}), {1.0f})
[all...] |
| roll_op_test.cc | 55 AddInputFromArray<float>(TensorShape({5}), {0, 1, 2, 3, 4});
56 AddInputFromArray<int32>(TensorShape({}), {3});
57 AddInputFromArray<int32>(TensorShape({}), {0});
70 AddInputFromArray<string>(TensorShape({5}), {"a", "b", "c", "d", "e"});
71 AddInputFromArray<int32>(TensorShape({}), {3});
72 AddInputFromArray<int32>(TensorShape({}), {0});
85 AddInputFromArray<std::complex<float>>(
89 AddInputFromArray<int32>(TensorShape({}), {3});
90 AddInputFromArray<int32>(TensorShape({}), {0});
106 AddInputFromArray<float>(TensorShape({3, 5})
[all...] |
| sequence_ops_test.cc | 59 AddInputFromArray<int32>(TensorShape({}), {0});
60 AddInputFromArray<int32>(TensorShape({}), {10});
61 AddInputFromArray<int32>(TensorShape({}), {2});
74 AddInputFromArray<float>(TensorShape({}), {0.5});
75 AddInputFromArray<float>(TensorShape({}), {2});
76 AddInputFromArray<float>(TensorShape({}), {0.3});
89 AddInputFromArray<double>(TensorShape({}), {0.0});
90 AddInputFromArray<double>(TensorShape({}), {10000});
91 AddInputFromArray<double>(TensorShape({}), {0.5});
106 AddInputFromArray<float>(TensorShape({}), {3.0})
[all...] |
| resize_nearest_neighbor_op_test.cc | 62 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4});
63 AddInputFromArray<int32>(TensorShape({2}), {1, 1});
80 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4});
81 AddInputFromArray<int32>(TensorShape({2}), {1, 1});
97 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4});
98 AddInputFromArray<int32>(TensorShape({2}), {3, 3});
118 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4});
119 AddInputFromArray<int32>(TensorShape({2}), {3, 3});
139 AddInputFromArray<float>(TensorShape({1, 3, 3, 1}),
141 AddInputFromArray<int32>(TensorShape({2}), {2, 2})
[all...] |
| non_max_suppression_op_test.cc | 46 AddInputFromArray<float>(
50 AddInputFromArray<float>(TensorShape({6}), {.9f, .75f, .6f, .95f, .5f, .3f});
51 AddInputFromArray<int>(TensorShape({}), {3});
61 AddInputFromArray<float>(TensorShape({6, 4}),
64 AddInputFromArray<float>(TensorShape({6}), {.9f, .75f, .6f, .95f, .5f, .3f});
65 AddInputFromArray<int>(TensorShape({}), {3});
75 AddInputFromArray<float>(
79 AddInputFromArray<float>(TensorShape({6}), {.9f, .75f, .6f, .95f, .5f, .3f});
80 AddInputFromArray<int>(TensorShape({}), {2});
90 AddInputFromArray<float>
[all...] |
| fused_batch_norm_op_test.cc | 43 AddInputFromArray<float>(TensorShape({1, 1, 6, 2}),
45 AddInputFromArray<float>(TensorShape({2}), {4.0, 4.0});
46 AddInputFromArray<float>(TensorShape({2}), {2.0, 2.0});
47 AddInputFromArray<float>(TensorShape({0}), {});
48 AddInputFromArray<float>(TensorShape({0}), {});
77 AddInputFromArray<float>(TensorShape({1, 1, 6, 2}),
79 AddInputFromArray<float>(TensorShape({2}), {4.0, 4.0});
80 AddInputFromArray<float>(TensorShape({2}), {2.0, 2.0});
81 AddInputFromArray<float>(TensorShape({2}), {10, 10});
82 AddInputFromArray<float>(TensorShape({2}), {11.67f, 11.67f})
[all...] |
| logging_ops_test.cc | 45 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6});
46 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6});
47 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6});
56 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6});
57 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6});
58 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6});
67 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6});
68 AddInputFromArray<string>(TensorShape({}), {"foo"});
69 AddInputFromArray<string>(TensorShape({}), {"bar"});
78 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6})
[all...] |
| quantized_matmul_op_test.cc | 58 AddInputFromArray<quint8>(TensorShape({2, 3}), {1, 2, 3, 4, 5, 6});
63 AddInputFromArray<quint8>(TensorShape({3, 4}),
65 AddInputFromArray<float>(TensorShape({1}), {0});
66 AddInputFromArray<float>(TensorShape({1}), {255.0f});
67 AddInputFromArray<float>(TensorShape({1}), {0});
68 AddInputFromArray<float>(TensorShape({1}), {255.0f});
116 AddInputFromArray<quint8>(TensorShape({a_rows, a_cols}), {11});
120 AddInputFromArray<quint8>(TensorShape({b_rows, b_cols}), {0});
121 AddInputFromArray<float>(TensorShape({1}), {-12.0f});
122 AddInputFromArray<float>(TensorShape({1}), {243.0f})
[all...] |
| scatter_nd_op_test.cc | 53 // AddInputFromArray<string>(TensorShape({1}), {"Brain"});
54 // AddInputFromArray<int32>(TensorShape({1}), {0});
55 // AddInputFromArray<string>(TensorShape({1}), {"TensorFlow"});
66 // AddInputFromArray<bool>(TensorShape({1}), {false});
67 // AddInputFromArray<int32>(TensorShape({1}), {0});
68 // AddInputFromArray<bool>(TensorShape({1}), {true});
81 AddInputFromArray<float>(TensorShape({5, 3}),
83 AddInputFromArray<int32>(TensorShape({3, 1}), {0, 4, 2});
84 AddInputFromArray<float>(TensorShape({3, 3}),
100 AddInputFromArray<float>(TensorShape({5, 3})
[all...] |
| sparse_to_dense_op_test.cc | 56 AddInputFromArray<int32>(TensorShape({3}), {1, 3, 4});
58 AddInputFromArray<int32>(TensorShape({1}), {5});
60 AddInputFromArray<float>(TensorShape({}), {2});
62 AddInputFromArray<float>(TensorShape({}), {-2});
75 AddInputFromArray<int64>(TensorShape({3}), {1, 3, 4});
77 AddInputFromArray<int64>(TensorShape({1}), {5});
79 AddInputFromArray<double>(TensorShape({}), {2});
81 AddInputFromArray<double>(TensorShape({}), {-2});
94 AddInputFromArray<int32>({3}, {1, 3, 4});
96 AddInputFromArray<int32>({1}, {5})
[all...] |
| quantize_and_dequantize_op_test.cc | 51 AddInputFromArray<float>(TensorShape({1}), {-3.5});
52 AddInputFromArray<float>(TensorShape({}), {0.0}); // Min
53 AddInputFromArray<float>(TensorShape({}), {0.0}); // Max
76 AddInputFromArray<float>(TensorShape({1}), {-3.5});
77 AddInputFromArray<float>(TensorShape({}), {0.0}); // Min
78 AddInputFromArray<float>(TensorShape({}), {0.0}); // Max
79 AddInputFromArray<int32>(TensorShape({}), {8}); // num_bits
103 AddInputFromArray<float>(TensorShape({6}), {-1, -0.5, 0, 0.3, 0.8, 0.555});
104 AddInputFromArray<float>(TensorShape({}), {0.0}); // Min
105 AddInputFromArray<float>(TensorShape({}), {0.0}); // Ma
[all...] |
| scatter_op_test.cc | 52 AddInputFromArray<string>(TensorShape({1}), {"Brain"});
53 AddInputFromArray<int32>(TensorShape({1}), {0});
54 AddInputFromArray<string>(TensorShape({1}), {"TensorFlow"});
65 AddInputFromArray<bool>(TensorShape({1}), {false});
66 AddInputFromArray<int32>(TensorShape({1}), {0});
67 AddInputFromArray<bool>(TensorShape({1}), {true});
80 AddInputFromArray<float>(TensorShape({5, 3}),
82 AddInputFromArray<int32>(TensorShape({3}), {0, 4, 2});
83 AddInputFromArray<float>(TensorShape({3, 3}),
99 AddInputFromArray<float>(TensorShape({5, 3})
[all...] |
| cross_op_test.cc | 43 AddInputFromArray<float>(TensorShape({3}), {0, 0, 0});
44 AddInputFromArray<float>(TensorShape({3}), {0, 0, 0});
53 AddInputFromArray<float>(TensorShape({2, 3}), {1, 0, 0, /**/ 0, 1, 0});
54 AddInputFromArray<float>(TensorShape({2, 3}), {0, 1, 0, /**/ 1, 0, 0});
69 AddInputFromArray<float>(TensorShape({3}), {u1, u2, u3});
70 AddInputFromArray<float>(TensorShape({3}), {v1, v2, v3});
90 AddInputFromArray<int>(TensorShape({2, 3}), {2, 0, 0, /**/ 0, 2, 0});
91 AddInputFromArray<int>(TensorShape({2, 3}), {0, 2, 0, /**/ 2, 0, 0});
|
| adjust_contrast_op_test.cc | 40 AddInputFromArray<float>(TensorShape({1, 1, 1, 3}), {-1, 2, 3});
41 AddInputFromArray<float>(TensorShape({}), {1.0});
55 AddInputFromArray<float>(TensorShape({1, 2, 2, 3}),
57 AddInputFromArray<float>(TensorShape({}), {0.2f});
79 AddInputFromArray<float>(TensorShape({1, 99, 99, 3}), values);
80 AddInputFromArray<float>(TensorShape({}), {0.2f});
|
| gather_op_test.cc | 56 AddInputFromArray<float>(TensorShape({5}), {0, 1, 2, 3, 4});
57 AddInputFromArray<int32>(TensorShape({}), {3});
58 AddInputFromArray<int32>(TensorShape({}), {0});
71 AddInputFromArray<std::complex<float>>(
75 AddInputFromArray<int32>(TensorShape({}), {3});
76 AddInputFromArray<int32>(TensorShape({}), {0});
90 AddInputFromArray<float>(TensorShape({5, 3}),
92 AddInputFromArray<int32>(TensorShape({4}), {0, 4, 0, 2});
93 AddInputFromArray<int32>(TensorShape({}), {0});
106 AddInputFromArray<float>(TensorShape({5, 3})
[all...] |
| quantized_batch_norm_op_test.cc | 98 AddInputFromArray<quint8>(input_quantized.shape(),
100 AddInputFromArray<float>(TensorShape({1}), {input_min});
101 AddInputFromArray<float>(TensorShape({1}), {input_max});
102 AddInputFromArray<quint8>(mean_quantized.shape(),
104 AddInputFromArray<float>(TensorShape({1}), {mean_min});
105 AddInputFromArray<float>(TensorShape({1}), {mean_max});
106 AddInputFromArray<quint8>(variance_quantized.shape(),
108 AddInputFromArray<float>(TensorShape({1}), {variance_min});
109 AddInputFromArray<float>(TensorShape({1}), {variance_max});
110 AddInputFromArray<quint8>(beta_quantized.shape()
[all...] |
| quantized_conv_ops_test.cc | 90 AddInputFromArray<quint8>(image_quantized.shape(),
92 AddInputFromArray<quint8>(filter_quantized.shape(),
94 AddInputFromArray<float>(TensorShape({1}), {image_min});
95 AddInputFromArray<float>(TensorShape({1}), {image_max});
96 AddInputFromArray<float>(TensorShape({1}), {filter_min});
97 AddInputFromArray<float>(TensorShape({1}), {filter_max});
153 AddInputFromArray<quint8>(
158 AddInputFromArray<quint8>(
161 AddInputFromArray<float>(TensorShape({1}), {0});
162 AddInputFromArray<float>(TensorShape({1}), {255.0f})
[all...] |
| quantize_op_test.cc | 40 AddInputFromArray<float>(TensorShape({7}),
43 AddInputFromArray<float>(TensorShape({1}), {0});
45 AddInputFromArray<float>(TensorShape({1}), {255.0f});
63 AddInputFromArray<float>(TensorShape({8}),
65 AddInputFromArray<float>(TensorShape({1}), {-255.0f});
66 AddInputFromArray<float>(TensorShape({1}), {127.0f});
95 AddInputFromArray<float>(TensorShape({3}), {-1.0, 0.0, 2.0});
96 AddInputFromArray<float>(TensorShape({1}), {-1.0f});
97 AddInputFromArray<float>(TensorShape({1}), {2.0f});
125 AddInputFromArray<float>(TensorShape({7})
[all...] |
| sparse_add_op_test.cc | 70 AddInputFromArray<int64>(indices_shape, indices); \
71 AddInputFromArray<float>(TensorShape({4}), {1, 2, 3, 4}); \
72 AddInputFromArray<int64>(TensorShape({2}), shape);
76 AddInputFromArray<float>(TensorShape({}), {0.0});
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...] |
| resize_bilinear_op_test.cc | 115 AddInputFromArray<int32>(TensorShape({2}), {output_width, output_height});
171 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4});
172 AddInputFromArray<int32>(TensorShape({2}), {1, 1});
184 AddInputFromArray<int32>(TensorShape({2}), {1, 1});
203 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4});
204 AddInputFromArray<int32>(TensorShape({2}), {1, 1});
218 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4});
219 AddInputFromArray<int32>(TensorShape({2}), {3, 3});
238 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4});
239 AddInputFromArray<int32>(TensorShape({2}), {3, 3})
[all...] |
| quantized_bias_add_op_test.cc | 73 AddInputFromArray<quint8>(input_quantized.shape(),
75 AddInputFromArray<quint8>(bias_quantized.shape(),
77 AddInputFromArray<float>(TensorShape({1}), {input_min});
78 AddInputFromArray<float>(TensorShape({1}), {input_max});
79 AddInputFromArray<float>(TensorShape({1}), {bias_min});
80 AddInputFromArray<float>(TensorShape({1}), {bias_max});
155 AddInputFromArray<quint8>(input_quantized.shape(),
157 AddInputFromArray<quint8>(bias_quantized.shape(),
159 AddInputFromArray<float>(TensorShape({1}), {input_min});
160 AddInputFromArray<float>(TensorShape({1}), {input_max})
[all...] |
| quantized_reshape_op_test.cc | 58 AddInputFromArray<quint8>(input.shape(), input.flat<quint8>());
60 AddInputFromArray<float>(TensorShape({1}), {-10});
61 AddInputFromArray<float>(TensorShape({1}), {20});
|
| identity_n_op_test.cc | 41 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6});
42 AddInputFromArray<double>(TensorShape({6}),
55 AddInputFromArray<int32>(TensorShape({2, 3}), {1, 2, 3, 4, 5, 6});
56 AddInputFromArray<int32>(TensorShape({2, 3}), {7, 8, 9, 10, 11, 12});
67 AddInputFromArray<string>(TensorShape({6}), {"A", "b", "C", "d", "E", "f"});
68 AddInputFromArray<int32>(TensorShape({8}), {1, 3, 5, 7, 9, 11, 13, 15});
|
