Cross Reference: /external/tensorflow/tensorflow/core/ops/spectral_ops

Lines Matching refs:op
24     ShapeInferenceTestOp op(op_name);
25     INFER_OK(op, "?", "in0");
26     INFER_ERROR("Shape must be at least rank 1 but is rank 0", op, "[]");
27     INFER_OK(op, "[?]", "in0");
28     INFER_OK(op, "[1]", "in0");
29     INFER_OK(op, "[1,2,3,4,5,6,7]", "in0");
33     ShapeInferenceTestOp op(op_name);
34     INFER_OK(op, "?", "in0");
35     INFER_ERROR("Shape must be at least rank 2 but is rank 1", op, "[1]");
36     INFER_OK(op, "[?,1]", "in0");
37     INFER_OK(op, "[1,2]", "in0");
38     INFER_OK(op, "[1,2,3,4,5,6,7]", "in0");
42     ShapeInferenceTestOp op(op_name);
43     INFER_OK(op, "?", "in0");
44     INFER_ERROR("Shape must be at least rank 3 but is rank 2", op, "[1,2]");
45     INFER_OK(op, "[?,1,?]", "in0");
46     INFER_OK(op, "[1,2,3]", "in0");
47     INFER_OK(op, "[1,2,3,4,5,6,7]", "in0");
54     ShapeInferenceTestOp op(forward ? "RFFT" : "IRFFT");
57     INFER_OK(op, "?;?", "?");
58     INFER_OK(op, "?;[1]", "?");
62     INFER_OK(op, "[1];?", "[?]");
63     INFER_OK(op, "[1];[1]", "[?]");
64     INFER_OK(op, "[?];[1]", "[?]");
67     INFER_OK(op, "[1,2,3,4];[1]", "[d0_0,d0_1,d0_2,?]");
69     INFER_ERROR("Shape must be at least rank 1 but is rank 0", op, "[];?");
70     INFER_ERROR("Shape must be rank 1 but is rank 2", op, "[1];[1,1]");
71     INFER_ERROR("Dimension must be 1 but is 2", op, "[1];[2]");
74     op.input_tensors.resize(2);
76     op.input_tensors[1] = &fft_length;
80       INFER_OK(op, "[?];[1]", "[6]");
81       INFER_OK(op, "[1];[1]", "[6]");
82       INFER_OK(op, "[1,1];[1]", "[d0_0,6]");
84       INFER_OK(op, "[?];[1]", "[10]");
85       INFER_OK(op, "[1];[1]", "[10]");
86       INFER_OK(op, "[1,1];[1]", "[d0_0,10]");
91       INFER_OK(op, "[?];[1]", "[6]");
92       INFER_OK(op, "[1];[1]", "[6]");
93       INFER_OK(op, "[1,1];[1]", "[d0_0,6]");
95       INFER_OK(op, "[?];[1]", "[11]");
96       INFER_OK(op, "[1];[1]", "[11]");
97       INFER_OK(op, "[1,1];[1]", "[d0_0,11]");
102       INFER_OK(op, "[?];[1]", "[7]");
103       INFER_OK(op, "[1];[1]", "[7]");
104       INFER_OK(op, "[1,1];[1]", "[d0_0,7]");
106       INFER_OK(op, "[?];[1]", "[12]");
107       INFER_OK(op, "[1];[1]", "[12]");
108       INFER_OK(op, "[1,1];[1]", "[d0_0,12]");
114     ShapeInferenceTestOp op(forward ? "RFFT2D" : "IRFFT2D");
117     INFER_OK(op, "?;?", "?");
118     INFER_OK(op, "?;[2]", "?");
122     INFER_OK(op, "[1,1];?", "[?,?]");
123     INFER_OK(op, "[1,1];[2]", "[?,?]");
124     INFER_OK(op, "[?,?];[2]", "[?,?]");
127     INFER_OK(op, "[1,2,3,4];[2]", "[d0_0,d0_1,?,?]");
129     INFER_ERROR("Shape must be at least rank 2 but is rank 0", op, "[];?");
130     INFER_ERROR("Shape must be rank 1 but is rank 2", op, "[1,1];[1,1]");
131     INFER_ERROR("Dimension must be 2 but is 3", op, "[1,1];[3]");
134     op.input_tensors.resize(2);
136     op.input_tensors[1] = &fft_length;
140       INFER_OK(op, "[?,?];[2]", "[9,6]");
141       INFER_OK(op, "[1,1];[2]", "[9,6]");
142       INFER_OK(op, "[1,1,1];[2]", "[d0_0,9,6]");
144       INFER_OK(op, "[?,?];[2]", "[9,10]");
145       INFER_OK(op, "[1,1];[2]", "[9,10]");
146       INFER_OK(op, "[1,1,1];[2]", "[d0_0,9,10]");
151       INFER_OK(op, "[?,?];[2]", "[10,6]");
152       INFER_OK(op, "[1,1];[2]", "[10,6]");
153       INFER_OK(op, "[1,1,1];[2]", "[d0_0,10,6]");
155       INFER_OK(op, "[?,?];[2]", "[10,11]");
156       INFER_OK(op, "[1,1];[2]", "[10,11]");
157       INFER_OK(op, "[1,1,1];[2]", "[d0_0,10,11]");
162       INFER_OK(op, "[?,?];[2]", "[11,7]");
163       INFER_OK(op, "[1,1];[2]", "[11,7]");
164       INFER_OK(op, "[1,1,1];[2]", "[d0_0,11,7]");
166       INFER_OK(op, "[?,?];[2]", "[11,12]");
167       INFER_OK(op, "[1,1];[2]", "[11,12]");
168       INFER_OK(op, "[1,1,1];[2]", "[d0_0,11,12]");
174     ShapeInferenceTestOp op(forward ? "RFFT3D" : "IRFFT3D");
177     INFER_OK(op, "?;?", "?");
178     INFER_OK(op, "?;[3]", "?");
182     INFER_OK(op, "[1,1,1];?", "[?,?,?]");
183     INFER_OK(op, "[1,1,1];[3]", "[?,?,?]");
184     INFER_OK(op, "[?,?,?];[3]", "[?,?,?]");
187     INFER_OK(op, "[1,2,3,4];[3]", "[d0_0,?,?,?]");
189     INFER_ERROR("Shape must be at least rank 3 but is rank 0", op, "[];?");
190     INFER_ERROR("Shape must be rank 1 but is rank 2", op, "[1,1,1];[1,1]");
191     INFER_ERROR("Dimension must be 3 but is 4", op, "[1,1,1];[4]");
194     op.input_tensors.resize(2);
196     op.input_tensors[1] = &fft_length;
200       INFER_OK(op, "[?,?,?];[3]", "[10,11,7]");
201       INFER_OK(op, "[1,1,1];[3]", "[10,11,7]");
202       INFER_OK(op, "[1,1,1,1];[3]", "[d0_0,10,11,7]");
204       INFER_OK(op, "[?,?,?];[3]", "[10,11,12]");
205       INFER_OK(op, "[1,1,1];[3]", "[10,11,12]");
206       INFER_OK(op, "[1,1,1,1];[3]", "[d0_0,10,11,12]");
211       INFER_OK(op, "[?,?,?];[3]", "[11,12,7]");
212       INFER_OK(op, "[1,1,1];[3]", "[11,12,7]");
213       INFER_OK(op, "[1,1,1,1];[3]", "[d0_0,11,12,7]");
215       INFER_OK(op, "[?,?,?];[3]", "[11,12,13]");
216       INFER_OK(op, "[1,1,1];[3]", "[11,12,13]");
217       INFER_OK(op, "[1,1,1,1];[3]", "[d0_0,11,12,13]");
222       INFER_OK(op, "[?,?,?];[3]", "[12,13,8]");
223       INFER_OK(op, "[1,1,1];[3]", "[12,13,8]");
224       INFER_OK(op, "[1,1,1,1];[3]", "[d0_0,12,13,8]");
226       INFER_OK(op, "[?,?,?];[3]", "[12,13,14]");
227       INFER_OK(op, "[1,1,1];[3]", "[12,13,14]");
228       INFER_OK(op, "[1,1,1,1];[3]", "[d0_0,12,13,14]");