Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/
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()
122 MakeOp(.5); in TEST_F()
134 MakeOp(.5); in TEST_F()
157 MakeOp(.5); in TEST_F()
173 MakeOp(1.2); 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 …]
Droll_op_test.cc42 void MakeOp(DataType data_type, DataType index_type) { in MakeOp() function in tensorflow::__anon21f87f230111::RollOpTest
53 MakeOp(DT_FLOAT, DT_INT32); in TEST_F()
68 MakeOp(DT_STRING, DT_INT32); in TEST_F()
83 MakeOp(DT_COMPLEX64, DT_INT32); in TEST_F()
104 MakeOp(DT_FLOAT, DT_INT32); in TEST_F()
121 MakeOp(DT_STRING, DT_INT32); in TEST_F()
139 MakeOp(DT_FLOAT, DT_INT32); in TEST_F()
155 MakeOp(DT_STRING, DT_INT32); in TEST_F()
173 MakeOp(DT_FLOAT, DT_INT64); in TEST_F()
190 MakeOp(DT_STRING, DT_INT64); in TEST_F()
[all …]
Dsummary_op_test.cc50 void MakeOp(DataType dt) { in MakeOp() function in tensorflow::__anone9fff3f80111::SummaryScalarOpTest
60 MakeOp(DT_FLOAT); in TEST_F()
80 MakeOp(DT_DOUBLE); in TEST_F()
100 MakeOp(DT_HALF); in TEST_F()
120 MakeOp(DT_FLOAT); in TEST_F()
130 MakeOp(DT_FLOAT); in TEST_F()
142 MakeOp(DT_FLOAT); in TEST_F()
158 void MakeOp(DataType dt) { in MakeOp() function in tensorflow::__anone9fff3f80111::SummaryHistoOpTest
168 MakeOp(DT_FLOAT); in TEST_F()
197 MakeOp(DT_DOUBLE); in TEST_F()
[all …]
Dparse_tensor_test.cc36 void MakeOp(const TensorShape& input_shape, std::function<T(int)> functor) { in MakeOp() function in tensorflow::__anon6e7fb4e10111::SerializeTensorOpTest
78 MakeOp<Eigen::half>(TensorShape({10}), [](int x) -> Eigen::half { in TEST_F()
88 MakeOp<float>(TensorShape({1, 10}), in TEST_F()
97 MakeOp<double>(TensorShape({5, 5}), in TEST_F()
106 MakeOp<int64>(TensorShape({2, 3, 4}), in TEST_F()
115 MakeOp<int32>(TensorShape({4, 2}), in TEST_F()
124 MakeOp<int16>(TensorShape({8}), in TEST_F()
133 MakeOp<int8>(TensorShape({2}), in TEST_F()
142 MakeOp<uint16>(TensorShape({1, 3}), in TEST_F()
151 MakeOp<uint8>(TensorShape({2, 1, 1}), in TEST_F()
[all …]
Dscoped_allocator_ops_test.cc40 void MakeOp(const TensorShape& shape, in MakeOp() function in tensorflow::ScopedAllocatorOpTest
66 MakeOp(TensorShape({8}), {TensorShape({8})}, DT_FLOAT, "test", 120, 1); in TEST_F()
67 MakeOp(TensorShape({1024}), {TensorShape({32, 32})}, DT_DOUBLE, "test1", 130, in TEST_F()
69 MakeOp(TensorShape({204}), in TEST_F()
72 MakeOp(TensorShape({1024}), {TensorShape({512}), TensorShape({64, 8})}, in TEST_F()
161 void MakeOp(const TensorShape& shape, DataType dtype, bool reshape, in MakeOp() function in tensorflow::ScopedAllocatorConcatOpTest
204 MakeOp({32}, DT_FLOAT, false, "test", 120, 2); in TEST_F()
209 MakeOp({2, 2, 2}, DT_DOUBLE, false, "test", 120, 2); in TEST_F()
214 MakeOp({3, 3, 3}, DT_HALF, false, "test", 120, 3); in TEST_F()
219 MakeOp({2, 2, 4}, DT_DOUBLE, true, "test", 120, 2); in TEST_F()
[all …]
Dsequence_ops_test.cc33 void MakeOp(DataType input_type) { in MakeOp() function in tensorflow::__anonf4b9c9f60111::RangeOpTest
45 void MakeOp(DataType input_type, DataType index_type) { in MakeOp() function in tensorflow::__anonf4b9c9f60111::LinSpaceOpTest
56 MakeOp(DT_INT32); in TEST_F()
71 MakeOp(DT_FLOAT); in TEST_F()
86 MakeOp(DT_DOUBLE); in TEST_F()
103 MakeOp(DT_FLOAT, DT_INT32); in TEST_F()
118 MakeOp(DT_FLOAT, DT_INT32); in TEST_F()
139 MakeOp(DT_FLOAT, DT_INT64); in TEST_F()
154 MakeOp(DT_DOUBLE, DT_INT32); in TEST_F()
Ddynamic_stitch_op_test.cc39 void MakeOp(int n, DataType dt) { in MakeOp() function in tensorflow::__anon7557f2dd0111::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()
Dscatter_nd_op_test.cc41 void MakeOp(DataType variable_ref_type, DataType index_type) { in MakeOp() function in tensorflow::__anonb0382b110111::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()
225 MakeOp(DT_FLOAT_REF, DT_INT32); in TEST_F()
Dgather_op_test.cc43 void MakeOp(DataType data_type, DataType index_type) { in MakeOp() function in tensorflow::__anon5d091fe50111::GatherOpTest
54 MakeOp(DT_FLOAT, DT_INT32); in TEST_F()
69 MakeOp(DT_COMPLEX64, DT_INT32); in TEST_F()
88 MakeOp(DT_FLOAT, DT_INT32); in TEST_F()
104 MakeOp(DT_FLOAT, DT_INT32); in TEST_F()
121 MakeOp(DT_FLOAT, DT_INT32); in TEST_F()
135 MakeOp(DT_FLOAT, DT_INT64); in TEST_F()
151 MakeOp(DT_FLOAT, DT_INT32); in TEST_F()
166 MakeOp(DT_FLOAT, DT_INT32); in TEST_F()
Dscatter_op_test.cc41 void MakeOp(DataType variable_ref_type, DataType index_type) { in MakeOp() function in tensorflow::__anon826d87600111::ScatterUpdateOpTest
52 MakeOp(DT_STRING_REF, DT_INT32); in TEST_F()
65 MakeOp(DT_BOOL_REF, DT_INT32); in TEST_F()
78 MakeOp(DT_FLOAT_REF, DT_INT32); in TEST_F()
97 MakeOp(DT_FLOAT_REF, DT_INT64); in TEST_F()
116 MakeOp(DT_FLOAT_REF, DT_INT32); in TEST_F()
132 MakeOp(DT_FLOAT_REF, DT_INT32); in TEST_F()
148 MakeOp(DT_FLOAT_REF, DT_INT32); in TEST_F()
164 MakeOp(DT_FLOAT_REF, DT_INT32); in TEST_F()
179 MakeOp(DT_FLOAT_REF, DT_INT32); in TEST_F()
[all …]
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); \
Dsparse_to_dense_op_test.cc41 void MakeOp(int dim, DataType index_type, DataType value_type) { in MakeOp() function in tensorflow::__anon48929ca80111::SparseToDenseTest
53 MakeOp(1, DT_INT32, DT_FLOAT); in TEST_F()
72 MakeOp(1, DT_INT64, DT_DOUBLE); in TEST_F()
91 MakeOp(1, DT_INT32, DT_FLOAT); in TEST_F()
110 MakeOp(2, DT_INT32, DT_FLOAT); in TEST_F()
132 MakeOp(2, DT_INT32, DT_FLOAT); in TEST_F()
154 MakeOp(3, DT_INT32, DT_FLOAT); in TEST_F()
176 MakeOp(3, DT_INT32, DT_FLOAT); in TEST_F()
Dsparse_dense_binary_op_shared_test.cc43 void MakeOp() { in MakeOp() function in tensorflow::__anonbc9c9b600111::SparseDenseCDivTest
59 void MakeOp() { in MakeOp() function in tensorflow::__anonbc9c9b600111::SparseDenseCMulTest
73 MakeOp<float>(); in TEST_F()
84 MakeOp<float>(); in TEST_F()
95 MakeOp<float>(); in TEST_F()
124 MakeOp<float>(); in TEST_F()
151 MakeOp<float>(); in TEST_F()
178 MakeOp<float>(); in TEST_F()
Dsummary_image_op_test.cc51 void MakeOp(int max_images) { in MakeOp() function in tensorflow::__anon6b20283a0111::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_reduce_sum_op_test.cc33 void MakeOp() { in MakeOp() function in tensorflow::__anonf151b2230111::SparseReduceSumOpTest
47 MakeOp<float>(); in TEST_F()
74 void MakeOp() { in MakeOp() function in tensorflow::__anonf151b2230111::SparseReduceSumSparseOpTest
89 MakeOp<float>(); in TEST_F()
Ddynamic_partition_op_test.cc42 void MakeOp() { in MakeOp() function in tensorflow::__anon41ea4a880111::DynamicPartitionOpTest
53 MakeOp(); in TEST_F()
86 MakeOp(); in TEST_F()
119 MakeOp(); in TEST_F()
150 MakeOp(); in TEST_F()
Dsummary_audio_op_test.cc51 void MakeOp(const int max_outputs) { in MakeOp() function in tensorflow::__anon5db2cce90111::SummaryAudioOpTest
81 MakeOp(kMaxOutputs); in TEST_F()
115 MakeOp(kMaxOutputs); in TEST_F()
Dsparse_add_op_test.cc33 void MakeOp() { in MakeOp() function in tensorflow::__anonddac4ea00111::SparseAddOpTest
57 MakeOp<float>(); in TEST_F()
100 MakeOp<VALTYPE>(); \
158 MakeOp<VALTYPE>(); \
Dsummary_tensor_op_test.cc51 void MakeOp() { in MakeOp() function in tensorflow::__anoned5e2ef20111::SummaryTensorOpV2Test
62 MakeOp(); in TEST_F()
Dreverse_op_test.cc41 void MakeOp(DataType data_type) { in MakeOp() function in tensorflow::__anon8d8e2a460111::ReverseOpTest
52 MakeOp(DataTypeToEnum<T>::value); in Reverse_0()
65 MakeOp(DataTypeToEnum<T>::value); in Reverse_234()
91 MakeOp(DataTypeToEnum<T>::value); in Reverse_1234()
Dgather_nd_op_test.cc60 void MakeOp(DataType index_type) { in MakeOp() function in tensorflow::__anon9c95b1760111::GatherNdOpTest
70 MakeOp(DT_INT32); in TEST_F()
Dmerge_v2_checkpoints_op_test.cc55 void MakeOp(bool delete_old_dirs) { in MakeOp() function in tensorflow::__anon03c6635b0111::MergeV2CheckpointsOpTest
80 MakeOp(delete_old_dirs); in RunMergeTest()
Dsave_op_test.cc45 void MakeOp() { in MakeOp() function in tensorflow::__anon4acb97790111::SaveOpTest
66 MakeOp(); in TEST_F()
396 void MakeOp() { in MakeOp() function in tensorflow::__anon4acb97790111::SaveSlicesOpTest
426 MakeOp(); in TEST_F()
555 void MakeOp() { in MakeOp() function in tensorflow::__anon4acb97790111::SaveOpSlices2Test
578 MakeOp(); in TEST_F()
/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::__anon0dcd93d60111::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::__anon0dcd93d60111::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::__anon0dcd93d60111::CropAndResizeGradTest
331 MakeOp<X_T>(x_data, boxes, {0}, {1, 1}, &x, &y); in TestCropAndResize()

12