Home
last modified time | relevance | path

Searched refs:MakeOp (Results 1 – 25 of 35) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dstring_ngrams_op_test.cc39 void MakeOp(string separator, std::vector<int> ngram_width, string left_pad, in MakeOp() function in tensorflow::text::NgramKernelTest
71 MakeOp("|", {3}, "LP", "RP", -1, false); in TEST_F()
89 MakeOp("|", {2, 3}, "LP", "RP", -1, false); in TEST_F()
108 MakeOp("|", {2}, "LP", "RP", -1, false); in TEST_F()
126 MakeOp("|", {2}, "LP", "RP", 4, false); in TEST_F()
144 MakeOp("|", {1, 2}, "LP", "RP", -1, false); in TEST_F()
164 MakeOp("|", {3}, "LP", "RP", -1, false); in TEST_F()
184 MakeOp("|", {3}, "LP", "RP", -1, false); in TEST_F()
207 MakeOp("|", {5}, "LP", "RP", -1, false); in TEST_F()
224 MakeOp("|", {3}, "", "", 0, false); in TEST_F()
[all …]
Droll_op_test.cc43 void MakeOp(DataType data_type, DataType index_type) { in MakeOp() function in tensorflow::__anon076712010111::RollOpTest
54 MakeOp(DT_FLOAT, DT_INT32); in TEST_F()
69 MakeOp(DT_STRING, DT_INT32); in TEST_F()
84 MakeOp(DT_COMPLEX64, DT_INT32); in TEST_F()
105 MakeOp(DT_FLOAT, DT_INT32); in TEST_F()
122 MakeOp(DT_STRING, DT_INT32); in TEST_F()
140 MakeOp(DT_FLOAT, DT_INT32); in TEST_F()
156 MakeOp(DT_STRING, DT_INT32); in TEST_F()
174 MakeOp(DT_FLOAT, DT_INT64); in TEST_F()
191 MakeOp(DT_STRING, DT_INT64); in TEST_F()
[all …]
Dparse_tensor_test.cc39 void MakeOp(const TensorShape& input_shape, std::function<T(int)> functor) { in MakeOp() function in tensorflow::__anon7fb2c27f0111::SerializeTensorOpTest
81 MakeOp<Eigen::half>(TensorShape({10}), [](int x) -> Eigen::half { in TEST_F()
91 MakeOp<float>(TensorShape({1, 10}), in TEST_F()
100 MakeOp<double>(TensorShape({5, 5}), in TEST_F()
109 MakeOp<int64>(TensorShape({2, 3, 4}), in TEST_F()
118 MakeOp<int32>(TensorShape({4, 2}), in TEST_F()
127 MakeOp<int16>(TensorShape({8}), in TEST_F()
136 MakeOp<int8>(TensorShape({2}), in TEST_F()
145 MakeOp<uint16>(TensorShape({1, 3}), in TEST_F()
154 MakeOp<uint8>(TensorShape({2, 1, 1}), in TEST_F()
[all …]
Dscoped_allocator_ops_test.cc41 void MakeOp(const TensorShape& shape, in MakeOp() function in tensorflow::ScopedAllocatorOpTest
67 MakeOp(TensorShape({8}), {TensorShape({8})}, DT_FLOAT, "test", 120, 1); in TEST_F()
68 MakeOp(TensorShape({1024}), {TensorShape({32, 32})}, DT_DOUBLE, "test1", 130, in TEST_F()
70 MakeOp(TensorShape({204}), in TEST_F()
73 MakeOp(TensorShape({1024}), {TensorShape({512}), TensorShape({64, 8})}, in TEST_F()
162 void MakeOp(const TensorShape& shape, DataType dtype, bool reshape, in MakeOp() function in tensorflow::ScopedAllocatorConcatOpTest
205 MakeOp({32}, DT_FLOAT, false, "test", 120, 2); in TEST_F()
210 MakeOp({2, 2, 2}, DT_DOUBLE, false, "test", 120, 2); in TEST_F()
215 MakeOp({3, 3, 3}, DT_HALF, false, "test", 120, 3); in TEST_F()
220 MakeOp({2, 2, 4}, DT_DOUBLE, true, "test", 120, 2); in TEST_F()
[all …]
Dsequence_ops_test.cc34 void MakeOp(DataType input_type) { in MakeOp() function in tensorflow::__anon05ecd7940111::RangeOpTest
46 void MakeOp(DataType input_type, DataType index_type) { in MakeOp() function in tensorflow::__anon05ecd7940111::LinSpaceOpTest
57 MakeOp(DT_INT32); in TEST_F()
72 MakeOp(DT_FLOAT); in TEST_F()
87 MakeOp(DT_DOUBLE); in TEST_F()
104 MakeOp(DT_FLOAT, DT_INT32); in TEST_F()
119 MakeOp(DT_FLOAT, DT_INT32); in TEST_F()
140 MakeOp(DT_FLOAT, DT_INT64); in TEST_F()
155 MakeOp(DT_DOUBLE, DT_INT32); in TEST_F()
Ddynamic_stitch_op_test.cc39 void MakeOp(int n, DataType dt) { in MakeOp() function in tensorflow::__anon020a333b0111::DynamicStitchOpTest
49 MakeOp(2, DT_FLOAT); in TEST_F()
65 MakeOp(3, DT_FLOAT); in TEST_F()
84 MakeOp(2, DT_FLOAT); in TEST_F()
99 MakeOp(2, DT_FLOAT); in TEST_F()
114 MakeOp(2, DT_FLOAT); in TEST_F()
131 MakeOp(2, DT_FLOAT); in TEST_F()
Dsummary_op_test.cc53 void MakeOp(DataType dt) { in MakeOp() function in tensorflow::__anon513702160111::SummaryHistoOpTest
63 MakeOp(DT_FLOAT); in TEST_F()
92 MakeOp(DT_DOUBLE); in TEST_F()
120 MakeOp(DT_HALF); in TEST_F()
149 MakeOp(DT_FLOAT); in TEST_F()
159 MakeOp(DT_FLOAT); in TEST_F()
173 void MakeOp(int num_inputs) { in MakeOp() function in tensorflow::__anon513702160111::SummaryMergeOpTest
182 MakeOp(1); in TEST_F()
216 MakeOp(3); in TEST_F()
250 MakeOp(1); in TEST_F()
Dgather_op_test.cc43 void MakeOp(DataType data_type, DataType index_type, int batch_dims = 0) { in MakeOp() function in tensorflow::__anon5867e2430111::GatherOpTest
55 MakeOp(DT_FLOAT, DT_INT32); in TEST_F()
70 MakeOp(DT_COMPLEX64, DT_INT32); in TEST_F()
89 MakeOp(DT_FLOAT, DT_INT32); in TEST_F()
105 MakeOp(DT_FLOAT, DT_INT32); in TEST_F()
122 MakeOp(DT_FLOAT, DT_INT32); in TEST_F()
136 MakeOp(DT_FLOAT, DT_INT64); in TEST_F()
152 MakeOp(DT_FLOAT, DT_INT32); in TEST_F()
167 MakeOp(DT_FLOAT, DT_INT32); in TEST_F()
181 MakeOp(DT_FLOAT, DT_INT32, 10); in TEST_F()
Dscatter_nd_op_test.cc41 void MakeOp(DataType variable_ref_type, DataType index_type) { in MakeOp() function in tensorflow::__anone7ccec6f0111::ScatterNdUpdateOpTest
79 MakeOp(DT_FLOAT_REF, DT_INT32); in TEST_F()
98 MakeOp(DT_FLOAT_REF, DT_INT64); in TEST_F()
130 MakeOp(DT_FLOAT_REF, DT_INT32); in TEST_F()
146 MakeOp(DT_FLOAT_REF, DT_INT32); in TEST_F()
162 MakeOp(DT_FLOAT_REF, DT_INT32); in TEST_F()
178 MakeOp(DT_FLOAT_REF, DT_INT32); in TEST_F()
193 MakeOp(DT_FLOAT_REF, DT_INT32); in TEST_F()
209 MakeOp(DT_FLOAT_REF, DT_INT32); in TEST_F()
227 MakeOp(DT_FLOAT_REF, DT_INT32); in TEST_F()
Dscatter_op_test.cc41 void MakeOp(DataType variable_ref_type, DataType index_type) { in MakeOp() function in tensorflow::__anone9a4957e0111::ScatterUpdateOpTest
52 void MakeOp(DataType variable_ref_type, DataType index_type) { in MakeOp() function in tensorflow::__anone9a4957e0111::ScatterSubOpTest
63 MakeOp(DT_STRING_REF, DT_INT32); in TEST_F()
76 MakeOp(DT_BOOL_REF, DT_INT32); in TEST_F()
89 MakeOp(DT_FLOAT_REF, DT_INT32); in TEST_F()
108 MakeOp(DT_FLOAT_REF, DT_INT64); in TEST_F()
127 MakeOp(DT_FLOAT_REF, DT_INT32); in TEST_F()
143 MakeOp(DT_FLOAT_REF, DT_INT32); in TEST_F()
159 MakeOp(DT_FLOAT_REF, DT_INT32); in TEST_F()
175 MakeOp(DT_FLOAT_REF, DT_INT32); in TEST_F()
[all …]
Disotonic_regression_op_test.cc39 void MakeOp(DataType type) { in MakeOp() function in tensorflow::__anon88361e500111::IsotonicRegressionOpTest
61 MakeOp(DT_FLOAT_REF); in TEST_F()
73 MakeOp(DT_FLOAT_REF); in TEST_F()
91 MakeOp(DT_FLOAT_REF); in TEST_F()
117 helper.MakeOp(DT_FLOAT_REF); in BM_IncreasingSequence()
Dsparse_to_dense_op_test.cc43 void MakeOp(int dim, DataType index_type, DataType value_type) { in MakeOp() function in tensorflow::__anon6b8ce8c60111::SparseToDenseTest
55 MakeOp(1, DT_INT32, DT_FLOAT); in TEST_F()
74 MakeOp(1, DT_INT64, DT_DOUBLE); in TEST_F()
93 MakeOp(1, DT_INT32, DT_FLOAT); in TEST_F()
112 MakeOp(2, DT_INT32, DT_FLOAT); in TEST_F()
134 MakeOp(2, DT_INT32, DT_FLOAT); in TEST_F()
156 MakeOp(3, DT_INT32, DT_FLOAT); in TEST_F()
178 MakeOp(3, DT_INT32, DT_FLOAT); in TEST_F()
Dsummary_image_op_test.cc51 void MakeOp(int max_images) { in MakeOp() function in tensorflow::__anond43cd4d80111::SummaryImageOpTest
78 MakeOp(3 /* max images */); in TEST_F()
101 MakeOp(1 /* max images */); in TEST_F()
121 MakeOp(1 /* max images */); in TEST_F()
Dsparse_dense_binary_op_shared_test.cc44 void MakeOp() { in MakeOp() function in tensorflow::__anoncb9458be0111::SparseDenseCDivTest
60 void MakeOp() { in MakeOp() function in tensorflow::__anoncb9458be0111::SparseDenseCMulTest
74 MakeOp<float>(); in TEST_F()
85 MakeOp<float>(); in TEST_F()
96 MakeOp<float>(); in TEST_F()
125 MakeOp<float>(); in TEST_F()
152 MakeOp<float>(); in TEST_F()
179 MakeOp<float>(); in TEST_F()
Dsparse_reduce_sum_op_test.cc34 void MakeOp() { in MakeOp() function in tensorflow::__anonbc0f7dc10111::SparseReduceSumOpTest
48 MakeOp<float>(); in TEST_F()
75 void MakeOp() { in MakeOp() function in tensorflow::__anonbc0f7dc10111::SparseReduceSumSparseOpTest
90 MakeOp<float>(); in TEST_F()
Ddynamic_partition_op_test.cc42 void MakeOp() { in MakeOp() function in tensorflow::__anon0ca816260111::DynamicPartitionOpTest
53 MakeOp(); in TEST_F()
86 MakeOp(); in TEST_F()
119 MakeOp(); in TEST_F()
150 MakeOp(); in TEST_F()
Dgather_nd_op_test.cc60 void MakeOp(DataType param_type, DataType index_type) { in MakeOp() function in tensorflow::__anonadc8bf140111::GatherNdOpTest
70 MakeOp(DT_FLOAT, DT_INT32); in TEST_F()
84 MakeOp(DT_QUINT8, DT_INT32); in TEST_F()
98 MakeOp(DT_QINT8, DT_INT32); in TEST_F()
Dsummary_audio_op_test.cc51 void MakeOp(const int max_outputs) { in MakeOp() function in tensorflow::__anonc6cf79870111::SummaryAudioOpTest
81 MakeOp(kMaxOutputs); in TEST_F()
115 MakeOp(kMaxOutputs); in TEST_F()
Dsparse_add_op_test.cc34 void MakeOp() { in MakeOp() function in tensorflow::__anon15410ffe0111::SparseAddOpTest
58 MakeOp<float>(); in TEST_F()
101 MakeOp<VALTYPE>(); \
159 MakeOp<VALTYPE>(); \
/external/tensorflow/tensorflow/core/kernels/image/
Dnon_max_suppression_op_gpu_test.cc44 void MakeOp() { in MakeOp() function in tensorflow::NonMaxSuppressionV2GPUOpTest
61 MakeOp(); in TEST_F()
78 MakeOp(); in TEST_F()
99 MakeOp(); in TEST_F()
116 MakeOp(); in TEST_F()
132 MakeOp(); in TEST_F()
145 MakeOp(); in TEST_F()
169 MakeOp(); in TEST_F()
186 MakeOp(); in TEST_F()
200 MakeOp(); in TEST_F()
[all …]
Dnon_max_suppression_op_test.cc34 void MakeOp(float iou_threshold) { in MakeOp() function in tensorflow::NonMaxSuppressionOpTest
46 MakeOp(.5); in TEST_F()
61 MakeOp(.5); in TEST_F()
75 MakeOp(.5); in TEST_F()
90 MakeOp(.5); in TEST_F()
107 MakeOp(.5); in TEST_F()
120 MakeOp(.5); in TEST_F()
135 MakeOp(.5); in TEST_F()
147 MakeOp(.5); in TEST_F()
170 MakeOp(.5); in TEST_F()
[all …]
Dcrop_and_resize_op_test.cc37 void MakeOp(float extrapolation_value, const string& method) { in MakeOp() function in tensorflow::CropAndResizeOpTest
52 MakeOp<T>(0, "bilinear"); \
65 MakeOp<T>(0, "nearest"); \
89 MakeOp<uint8>(0, "bilinear"); in REGISTER_TEST()
105 MakeOp<uint8>(0, "nearest"); in TEST_F()
121 MakeOp<float>(0, "bilinear"); in TEST_F()
137 MakeOp<float>(0, "nearest"); in TEST_F()
153 MakeOp<float>(0, "bilinear"); in TEST_F()
174 MakeOp<float>(0, "nearest"); in TEST_F()
195 MakeOp<float>(0, "bilinear"); in TEST_F()
[all …]
Dmirror_pad_op_test.cc37 void MakeOp(const string& mode) { in MakeOp() function in tensorflow::MirrorPadOpTest
49 MakeOp<T>("REFLECT"); \
63 MakeOp<T>("SYMMETRIC"); \
89 MakeOp<float>("REFLECT"); in REGISTER_TEST()
123 MakeOp<float>("SYMMETRIC"); in TEST_F()
159 void MakeOp(const string& mode) { in MakeOp() function in tensorflow::MirrorPadGradOpTest
171 MakeOp<T>("REFLECT"); \
183 MakeOp<T>("SYMMETRIC"); \
Dcolorspace_op_test.cc34 void MakeOp(DataType data_type) { in MakeOp() function in tensorflow::RGBToHSVOpTest
130 void MakeOp(DataType data_type) { in MakeOp() function in tensorflow::HSVToRGBOpTest
229 MakeOp(dt); \
233 MakeOp(dt); \
237 MakeOp(dt); \
241 MakeOp(dt); \
245 MakeOp(dt); \
249 MakeOp(dt); \
253 MakeOp(dt); \
/external/tensorflow/tensorflow/cc/gradients/
Dimage_grad_test.cc54 void MakeOp(const OpType op_type, const Tensor& x_data, const Input& y_shape, in MakeOp() function in tensorflow::__anon7504a1f40111::ImageGradTest
84 MakeOp<T>(op_type, x_data, {4, 6}, align_corners, half_pixel_centers, &x, in TestResizedShapeForType()
117 MakeOp<X_T>(op_type, x_data, {2, 3}, align_corners, half_pixel_centers, &x, in TestResizeToSmallerAndAlign()
132 MakeOp<X_T>(op_type, x_data, {4, 6}, align_corners, half_pixel_centers, &x, in TestResizeToLargerAndAlign()
199 void MakeOp(const Tensor& x_data, const Input& y_shape, Input scale, in MakeOp() function in tensorflow::__anon7504a1f40111::ScaleAndTranslateGradTest
217 MakeOp<X_T>(x_data, {out_height, out_width}, scale, translation, in TestScaleAndTranslate()
316 void MakeOp(const Tensor& x_data, const Input& boxes, const Input& box_ind, in MakeOp() function in tensorflow::__anon7504a1f40111::CropAndResizeGradTest
331 MakeOp<X_T>(x_data, boxes, {0}, {1, 1}, &x, &y); in TestCropAndResize()

12