Lines Matching refs:op
16 #include "tensorflow/core/framework/op.h"
23 static void TestGradAndIndicesErrorHandling(const ShapeInferenceTestOp& op,
33 INFER_ERROR("Dimension 1 in both shapes must be equal", op,
36 INFER_ERROR("Dimensions must be equal, but are 1 and 2", op,
39 INFER_ERROR("must be equal rank", op, shape_spec("[1]", "[?,2];[?]"));
41 INFER_ERROR("Shape must be rank 1 but is rank 2", op,
46 ShapeInferenceTestOp op("ApplyGradientDescent");
49 INFER_OK(op, "[1,?];[];[?,2]", "[d0_0,d2_1]");
50 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
54 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;[?];?");
58 ShapeInferenceTestOp op("ApplyProximalGradientDescent");
61 INFER_OK(op, "[1,?];[];[];[];[?,2]", "[d0_0,d4_1]");
62 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
66 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;[?];?;?;?");
67 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;[?];?;?");
68 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;[?];?");
72 ShapeInferenceTestOp op("SparseApplyProximalGradientDescent");
75 INFER_OK(op, "[1,?];[];[];[];[?,2];[3]", "[d0_0,d4_1]");
77 TestGradAndIndicesErrorHandling(op, "[];[];[]");
80 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;[?];?;?;?;?");
81 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;[?];?;?;?");
82 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;[?];?;?");
86 ShapeInferenceTestOp op("ApplyAdadelta");
90 INFER_OK(op, "[1,?,?,?];[?,2,?,?];[?,?,3,?];[];[];[];[?,?,?,4]",
92 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
94 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
96 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
100 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;[?];?;?;?");
101 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;[?];?;?");
102 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;?;[?];?");
106 ShapeInferenceTestOp op("SparseApplyAdadelta");
110 INFER_OK(op, "[1,?,?,?];[?,2,?,?];[?,?,3,?];[];[];[];[?,?,?,4];?",
112 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
114 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
116 INFER_ERROR("Dimension 1 in both shapes must be equal, but are 1 and 2", op,
119 TestGradAndIndicesErrorHandling(op, "?;?;?;?;?");
122 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;[?];?;?;?;?");
123 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;[?];?;?;?");
124 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;?;[?];?;?");
128 ShapeInferenceTestOp op("ApplyAdagrad");
131 INFER_OK(op, "[1,?,?];[?,2,?];[];[?,?,3]", "[d0_0,d1_1,d3_2]");
132 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
134 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
138 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;[?];?");
142 ShapeInferenceTestOp op("SparseApplyAdagrad");
146 INFER_OK(op, "[1,?,?];[?,2,?];[];[?,?,3];?", "[d0_0,d1_1,d3_2]");
147 INFER_ERROR("Dimension 1 in both shapes must be equal, but are 1 and 2", op,
149 INFER_ERROR("Shapes must be equal rank, but are 2 and 3", op,
152 TestGradAndIndicesErrorHandling(op, "?;?");
155 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;[?];?;?");
159 ShapeInferenceTestOp op("ApplyProximalAdagrad");
162 INFER_OK(op, "[1,?,?];[?,2,?];[];[];[];[?,?,3]", "[d0_0,d1_1,d5_2]");
163 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
165 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
169 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;[?];?;?;?");
170 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;[?];?;?");
171 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;[?];?");
175 ShapeInferenceTestOp op("SparseApplyProximalAdagrad");
179 INFER_OK(op, "[1,?,?];[?,2,?];[];[];[];[?,?,3];?", "[d0_0,d1_1,d5_2]");
180 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
182 INFER_ERROR("Dimension 1 in both shapes must be equal, but are 1 and 2", op,
185 TestGradAndIndicesErrorHandling(op, "?;?;?;?");
188 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;[?];?;?;?;?");
189 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;[?];?;?;?");
190 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;[?];?;?");
194 ShapeInferenceTestOp op("ApplyFtrl");
197 INFER_OK(op, "[1,?,?,?];[?,2,?,?];[?,?,3,?];[?,?,?,4];[];[];[];[]",
199 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
201 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
203 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
207 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;[?];?;?;?");
208 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;?;[?];?;?");
209 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;?;?;[?];?");
210 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;?;?;?;[?]");
214 ShapeInferenceTestOp op("SparseApplyFtrl");
218 INFER_OK(op, "[1,?,?,?];[?,2,?,?];[?,?,3,?];[?,?,?,4];?;[];[];[];[]",
220 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
222 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
224 INFER_ERROR("Dimension 1 in both shapes must be equal, but are 1 and 2", op,
227 TestGradAndIndicesErrorHandling(op, "?;?", ";?;?;?;?");
230 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;?;[?];?;?;?");
231 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;?;?;[?];?;?");
232 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;?;?;?;[?];?");
233 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;?;?;?;?;[?]");
237 ShapeInferenceTestOp op("ApplyMomentum");
240 INFER_OK(op, "[1,?,?];[?,2,?];[];[?,?,3];[]", "[d0_0,d1_1,d3_2]");
241 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
243 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
247 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;[?];?;?");
248 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;[?]");
252 ShapeInferenceTestOp op("SparseApplyMomentum");
256 INFER_OK(op, "[1,?,?];[?,2,?];[];[?,?,3];?;[]", "[d0_0,d1_1,d3_2]");
257 INFER_ERROR("Dimension 1 in both shapes must be equal, but are 1 and 2", op,
259 INFER_ERROR("Dimension 1 in both shapes must be equal, but are 1 and 2", op,
262 TestGradAndIndicesErrorHandling(op, "?;?", ";?");
265 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;[?];?;?;?");
266 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;?;[?]");
270 ShapeInferenceTestOp op("ApplyAdam");
273 INFER_OK(op, "[1,?,?,?];[?,2,?,?];[?,?,3,?];[];[];[];[];[];[];[?,?,?,4]",
275 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
277 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
279 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
284 INFER_ERROR(err, op, "?;?;?;[?];?;?;?;?;?;?");
285 INFER_ERROR(err, op, "?;?;?;?;[?];?;?;?;?;?");
286 INFER_ERROR(err, op, "?;?;?;?;?;[?];?;?;?;?");
287 INFER_ERROR(err, op, "?;?;?;?;?;?;[?];?;?;?");
288 INFER_ERROR(err, op, "?;?;?;?;?;?;?;[?];?;?");
289 INFER_ERROR(err, op, "?;?;?;?;?;?;?;?;[?];?");
293 ShapeInferenceTestOp op("ApplyRMSProp");
296 INFER_OK(op, "[1,?,?,?];[?,2,?,?];[?,?,3,?];[];[];[];[];[?,?,?,4]",
298 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
300 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
302 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
306 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;[?];?;?;?;?");
307 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;[?];?;?;?");
308 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;?;[?];?;?");
309 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;?;?;[?];?");
313 ShapeInferenceTestOp op("SparseApplyRMSProp");
317 INFER_OK(op, "[1,?,?,?];[?,2,?,?];[?,?,3,?];[];[];[];[];[?,?,?,4];?",
319 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
321 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
323 INFER_ERROR("Dimension 1 in both shapes must be equal, but are 1 and 2", op,
326 TestGradAndIndicesErrorHandling(op, "?;?;?;?;?;?");
329 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;[?];?;?;?;?;?");
330 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;[?];?;?;?;?");
331 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;?;[?];?;?;?");
332 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;?;?;[?];?;?");
336 ShapeInferenceTestOp op("ApplyAddSign");
339 INFER_OK(op, "[1,?,?];[?,2,?];[];[];[];[];[?,?,2]", "[d0_0,d1_1,d6_2]");
340 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
342 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
346 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;[?];?;?;?;?");
347 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;[?];?;?;?");
348 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;[?];?;?");
349 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;?;[?];?");
353 ShapeInferenceTestOp op("ApplyPowerSign");
356 INFER_OK(op, "[1,?,?];[?,2,?];[];[];[];[];[?,?,2]", "[d0_0,d1_1,d6_2]");
357 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
359 INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
363 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;[?];?;?;?;?");
364 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;[?];?;?;?");
365 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;[?];?;?");
366 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;?;?;?;[?];?");