| /external/tensorflow/tensorflow/core/kernels/ |
| crop_and_resize_op_test.cc | 37 void MakeOp(float extrapolation_value, const string& method) {
52 MakeOp<T>(0, "bilinear"); \
65 MakeOp<T>(0, "nearest"); \
89 MakeOp<uint8>(0, "bilinear");
105 MakeOp<uint8>(0, "nearest");
121 MakeOp<float>(0, "bilinear");
137 MakeOp<float>(0, "nearest");
153 MakeOp<float>(0, "bilinear");
174 MakeOp<float>(0, "nearest");
195 MakeOp<float>(0, "bilinear")
[all...] |
| dynamic_stitch_op_test.cc | 39 void MakeOp(int n, DataType dt) {
49 MakeOp(2, DT_FLOAT);
65 MakeOp(3, DT_FLOAT);
84 MakeOp(2, DT_FLOAT);
99 MakeOp(2, DT_FLOAT);
114 MakeOp(2, DT_FLOAT);
131 MakeOp(2, DT_FLOAT);
|
| roll_op_test.cc | 42 void MakeOp(DataType data_type, DataType index_type) {
53 MakeOp(DT_FLOAT, DT_INT32);
68 MakeOp(DT_STRING, DT_INT32);
83 MakeOp(DT_COMPLEX64, DT_INT32);
104 MakeOp(DT_FLOAT, DT_INT32);
121 MakeOp(DT_STRING, DT_INT32);
139 MakeOp(DT_FLOAT, DT_INT32);
155 MakeOp(DT_STRING, DT_INT32);
173 MakeOp(DT_FLOAT, DT_INT64);
190 MakeOp(DT_STRING, DT_INT64)
[all...] |
| sequence_ops_test.cc | 33 void MakeOp(DataType input_type) {
45 void MakeOp(DataType input_type, DataType index_type) {
56 MakeOp(DT_INT32);
71 MakeOp(DT_FLOAT);
86 MakeOp(DT_DOUBLE);
103 MakeOp(DT_FLOAT, DT_INT32);
118 MakeOp(DT_FLOAT, DT_INT32);
139 MakeOp(DT_FLOAT, DT_INT64);
154 MakeOp(DT_DOUBLE, DT_INT32);
|
| parse_tensor_test.cc | 36 void MakeOp(const TensorShape& input_shape, std::function<T(int)> functor) {
78 MakeOp<Eigen::half>(TensorShape({10}), [](int x) -> Eigen::half {
88 MakeOp<float>(TensorShape({1, 10}),
97 MakeOp<double>(TensorShape({5, 5}),
106 MakeOp<int64>(TensorShape({2, 3, 4}),
115 MakeOp<int32>(TensorShape({4, 2}),
124 MakeOp<int16>(TensorShape({8}),
133 MakeOp<int8>(TensorShape({2}),
142 MakeOp<uint16>(TensorShape({1, 3}),
151 MakeOp<uint8>(TensorShape({2, 1, 1})
[all...] |
| summary_op_test.cc | 50 void MakeOp(DataType dt) {
60 MakeOp(DT_FLOAT);
80 MakeOp(DT_DOUBLE);
100 MakeOp(DT_HALF);
120 MakeOp(DT_FLOAT);
130 MakeOp(DT_FLOAT);
142 MakeOp(DT_FLOAT);
158 void MakeOp(DataType dt) {
168 MakeOp(DT_FLOAT);
197 MakeOp(DT_DOUBLE)
[all...] |
| scoped_allocator_ops_test.cc | 40 void MakeOp(const TensorShape& shape,
66 MakeOp(TensorShape({8}), {TensorShape({8})}, DT_FLOAT, "test", 120, 1);
67 MakeOp(TensorShape({1024}), {TensorShape({32, 32})}, DT_DOUBLE, "test1", 130,
69 MakeOp(TensorShape({204}),
72 MakeOp(TensorShape({1024}), {TensorShape({512}), TensorShape({64, 8})},
161 void MakeOp(const TensorShape& shape, DataType dtype, bool reshape,
204 MakeOp({32}, DT_FLOAT, false, "test", 120, 2);
209 MakeOp({2, 2, 2}, DT_DOUBLE, false, "test", 120, 2);
214 MakeOp({3, 3, 3}, DT_HALF, false, "test", 120, 3);
219 MakeOp({2, 2, 4}, DT_DOUBLE, true, "test", 120, 2)
[all...] |
| scatter_nd_op_test.cc | 41 void MakeOp(DataType variable_ref_type, DataType index_type) {
53 // MakeOp(DT_STRING_REF, DT_INT32);
66 // MakeOp(DT_BOOL_REF, DT_INT32);
79 MakeOp(DT_FLOAT_REF, DT_INT32);
98 MakeOp(DT_FLOAT_REF, DT_INT64);
117 MakeOp(DT_FLOAT_REF, DT_INT32);
130 MakeOp(DT_FLOAT_REF, DT_INT32);
146 MakeOp(DT_FLOAT_REF, DT_INT32);
162 MakeOp(DT_FLOAT_REF, DT_INT32);
178 MakeOp(DT_FLOAT_REF, DT_INT32)
[all...] |
| scatter_op_test.cc | 41 void MakeOp(DataType variable_ref_type, DataType index_type) {
52 MakeOp(DT_STRING_REF, DT_INT32);
65 MakeOp(DT_BOOL_REF, DT_INT32);
78 MakeOp(DT_FLOAT_REF, DT_INT32);
97 MakeOp(DT_FLOAT_REF, DT_INT64);
116 MakeOp(DT_FLOAT_REF, DT_INT32);
132 MakeOp(DT_FLOAT_REF, DT_INT32);
148 MakeOp(DT_FLOAT_REF, DT_INT32);
164 MakeOp(DT_FLOAT_REF, DT_INT32);
179 MakeOp(DT_FLOAT_REF, DT_INT32)
[all...] |
| summary_audio_op_test.cc | 51 void MakeOp(const int max_outputs) {
81 MakeOp(kMaxOutputs);
115 MakeOp(kMaxOutputs);
|
| colorspace_op_test.cc | 34 void MakeOp(DataType data_type) {
130 void MakeOp(DataType data_type) {
229 MakeOp(dt); \
233 MakeOp(dt); \
237 MakeOp(dt); \
241 MakeOp(dt); \
245 MakeOp(dt); \
249 MakeOp(dt); \
253 MakeOp(dt); \
|
| gather_op_test.cc | 43 void MakeOp(DataType data_type, DataType index_type) {
54 MakeOp(DT_FLOAT, DT_INT32);
69 MakeOp(DT_COMPLEX64, DT_INT32);
88 MakeOp(DT_FLOAT, DT_INT32);
104 MakeOp(DT_FLOAT, DT_INT32);
121 MakeOp(DT_FLOAT, DT_INT32);
135 MakeOp(DT_FLOAT, DT_INT64);
151 MakeOp(DT_FLOAT, DT_INT32);
166 MakeOp(DT_FLOAT, DT_INT32);
|
| sparse_reduce_sum_op_test.cc | 33 void MakeOp() {
47 MakeOp<float>();
74 void MakeOp() {
89 MakeOp<float>();
|
| summary_image_op_test.cc | 51 void MakeOp(int max_images) {
78 MakeOp(3 /* max images */);
101 MakeOp(1 /* max images */);
121 MakeOp(1 /* max images */);
|
| non_max_suppression_op_test.cc | 34 void MakeOp(float iou_threshold) {
46 MakeOp(.5);
61 MakeOp(.5);
75 MakeOp(.5);
90 MakeOp(.5);
107 MakeOp(.5);
122 MakeOp(.5);
134 MakeOp(.5);
157 MakeOp(.5);
173 MakeOp(1.2)
[all...] |
| sparse_to_dense_op_test.cc | 41 void MakeOp(int dim, DataType index_type, DataType value_type) {
53 MakeOp(1, DT_INT32, DT_FLOAT);
72 MakeOp(1, DT_INT64, DT_DOUBLE);
91 MakeOp(1, DT_INT32, DT_FLOAT);
110 MakeOp(2, DT_INT32, DT_FLOAT);
132 MakeOp(2, DT_INT32, DT_FLOAT);
154 MakeOp(3, DT_INT32, DT_FLOAT);
176 MakeOp(3, DT_INT32, DT_FLOAT);
|
| summary_tensor_op_test.cc | 51 void MakeOp() {
62 MakeOp();
|
| sparse_dense_binary_op_shared_test.cc | 43 void MakeOp() {
59 void MakeOp() {
73 MakeOp<float>();
84 MakeOp<float>();
95 MakeOp<float>();
124 MakeOp<float>();
151 MakeOp<float>();
178 MakeOp<float>();
|
| dynamic_partition_op_test.cc | 42 void MakeOp() {
53 MakeOp();
86 MakeOp();
119 MakeOp();
150 MakeOp();
|
| gather_nd_op_test.cc | 60 void MakeOp(DataType index_type) {
70 MakeOp(DT_INT32);
|
| merge_v2_checkpoints_op_test.cc | 55 void MakeOp(bool delete_old_dirs) {
80 MakeOp(delete_old_dirs);
|
| sparse_add_op_test.cc | 33 void MakeOp() {
57 MakeOp<float>();
100 MakeOp<VALTYPE>(); \
158 MakeOp<VALTYPE>(); \
|
| reverse_op_test.cc | 41 void MakeOp(DataType data_type) {
52 MakeOp(DataTypeToEnum<T>::value);
65 MakeOp(DataTypeToEnum<T>::value);
91 MakeOp(DataTypeToEnum<T>::value);
|
| save_op_test.cc | 45 void MakeOp() {
66 MakeOp();
396 void MakeOp() {
426 MakeOp();
555 void MakeOp() {
578 MakeOp();
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| image_grad_test.cc | 54 void MakeOp(const OpType op_type, const Tensor& x_data, const Input& y_shape,
84 MakeOp<T>(op_type, x_data, {4, 6}, align_corners, half_pixel_centers, &x,
117 MakeOp<X_T>(op_type, x_data, {2, 3}, align_corners, half_pixel_centers, &x,
132 MakeOp<X_T>(op_type, x_data, {4, 6}, align_corners, half_pixel_centers, &x,
199 void MakeOp(const Tensor& x_data, const Input& y_shape, Input scale,
217 MakeOp<X_T>(x_data, {out_height, out_width}, scale, translation,
316 void MakeOp(const Tensor& x_data, const Input& boxes, const Input& box_ind,
331 MakeOp<X_T>(x_data, boxes, {0}, {1, 1}, &x, &y);
|
