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

Lines Matching refs:op
18 #include "tensorflow/core/framework/op.h"
27   ShapeInferenceTestOp op("TopK");
28   auto set_k = [&op](int k) {
32                      .Finalize(&op.node_def));
37   INFER_OK(op, "?", "?;?");
39   INFER_OK(op, "[20]", "[20];[20]");
40   INFER_OK(op, "[21]", "[20];[20]");
43   INFER_OK(op, "[1,?,21]", "[d0_0,d0_1,20];[d0_0,d0_1,20]");
45   INFER_OK(op, "[1,?,21,?]", "[d0_0,d0_1,d0_2,20];[d0_0,d0_1,d0_2,20]");
47   INFER_ERROR("Shape must be at least rank 1 but is rank 0", op, "[]");
48   INFER_ERROR("input must have last dimension >= k = 20 but is 1", op, "[1]");
49   INFER_ERROR("input must have last dimension >= k = 20 but is 4", op,
52   INFER_ERROR("Need k >= 0, got -1", op, "[1,2,3,4]");
56   ShapeInferenceTestOp op("TopKV2");
57   op.input_tensors.resize(2);
60   op.input_tensors[1] = &k_t;
64   INFER_OK(op, "?;[]", "?;?");
66   INFER_OK(op, "[20];[]", "[20];[20]");
69   INFER_OK(op, "[1,?,21];[]", "[d0_0,d0_1,20];[d0_0,d0_1,20]");
71   INFER_OK(op, "[1,?,21,?];[]", "[d0_0,d0_1,d0_2,20];[d0_0,d0_1,d0_2,20]");
73   INFER_ERROR("Shape must be at least rank 1 but is rank 0", op, "[];[]");
74   INFER_ERROR("input must have last dimension >= k = 20 but is 1", op,
76   INFER_ERROR("input must have last dimension >= k = 20 but is 4", op,
81       op, "[1,2,3,4];[]");
85   ShapeInferenceTestOp op("NthElement");
86   op.input_tensors.resize(2);
89   op.input_tensors[1] = &n_t;
92   INFER_OK(op, "?;[]", "?");
93   INFER_OK(op, "[21];[]", "[]");
94   INFER_OK(op, "[2,?,?];[]", "[d0_0,d0_1]");
95   INFER_OK(op, "[?,3,?,21];[]", "[d0_0,d0_1,d0_2]");
97   INFER_ERROR("Shape must be at least rank 1 but is rank 0", op, "[];[]");
98   INFER_ERROR("Input must have last dimension > n = 20 but is 1", op, "[1];[]");
99   INFER_ERROR("Input must have last dimension > n = 20 but is 20", op,
104       op, "[1,2,3,4];[]");
108   ShapeInferenceTestOp op("BatchNormWithGlobalNormalization");
111   INFER_ERROR("Shape must be rank 4 but is rank 3", op, "[1,2,3];?;?;?;?");
112   INFER_ERROR("Shape must be rank 1 but is rank 3", op, "?;[1,2,3];?;?;?");
113   INFER_ERROR("Shape must be rank 1 but is rank 3", op, "?;?;[1,2,3];?;?");
114   INFER_ERROR("Shape must be rank 1 but is rank 3", op, "?;?;?;[1,2,3];?");
115   INFER_ERROR("Shape must be rank 1 but is rank 3", op, "?;?;?;?;[1,2,3]");
118   INFER_OK(op, "?;?;?;?;?", "[?,?,?,?]");
119   INFER_OK(op, "?;[1];?;?;?", "[?,?,?,d1_0]");
120   INFER_OK(op, "?;?;[1];?;?", "[?,?,?,d2_0]");
121   INFER_OK(op, "?;?;?;[1];?", "[?,?,?,d3_0]");
122   INFER_OK(op, "?;?;?;?;[1]", "[?,?,?,d4_0]");
123   INFER_OK(op, "[1,2,3,4];[4];[4];[4];[4]",
131   ShapeInferenceTestOp op("QuantizedBatchNormWithGlobalNormalization");
134   INFER_ERROR("Shape must be rank 4 but is rank 3", op,
136   INFER_ERROR("Shape must be rank 1 but is rank 3", op,
138   INFER_ERROR("Shape must be rank 1 but is rank 3", op,
140   INFER_ERROR("Shape must be rank 1 but is rank 3", op,
142   INFER_ERROR("Shape must be rank 1 but is rank 3", op,
146   INFER_OK(op, "?;[];[];?;[];[];?;[];[];?;[];[];?;[];[]", "[?,?,?,?];[];[]");
147   INFER_OK(op, "?;[];[];[1];[];[];?;[];[];?;[];[];?;[];[]",
149   INFER_OK(op, "?;[];[];?;[];[];[1];[];[];?;[];[];?;[];[]",
151   INFER_OK(op, "?;[];[];?;[];[];?;[];[];[1];[];[];?;[];[]",
153   INFER_OK(op, "?;[];[];?;[];[];?;[];[];?;[];[];[1];[];[]",
155   INFER_OK(op, "[1,2,3,4];[];[];[4];[];[];[4];[];[];[4];[];[];[4];[];[]",
160   ShapeInferenceTestOp op("BatchNormWithGlobalNormalizationGrad");
163   INFER_ERROR("Shape must be rank 4 but is rank 3", op, "[1,2,3];?;?;?;?");
164   INFER_ERROR("Shape must be rank 1 but is rank 3", op, "?;[1,2,3];?;?;?");
165   INFER_ERROR("Shape must be rank 1 but is rank 3", op, "?;?;[1,2,3];?;?");
166   INFER_ERROR("Shape must be rank 1 but is rank 3", op, "?;?;?;[1,2,3];?");
167   INFER_ERROR("Shapes must be equal rank, but are 4 and 3", op,
173   INFER_OK(op, "?;?;?;?;?", "[?,?,?,?];[?];[?];[?];[?]");
174   INFER_OK(op, "?;[1];?;?;?", "[?,?,?,d1_0];[d1_0];[d1_0];[d1_0];[d1_0]");
175   INFER_OK(op, "?;?;[1];?;?", "[?,?,?,d2_0];[d2_0];[d2_0];[d2_0];[d2_0]");
176   INFER_OK(op, "?;?;?;[1];?", "[?,?,?,d3_0];[d3_0];[d3_0];[d3_0];[d3_0]");
177   INFER_OK(op, "[1,?,3,?];[?];[?];[?];[?,2,?,4]",
182   ShapeInferenceTestOp op("FusedBatchNorm");
184   auto set_op = [&op](bool is_training, string data_format) {
193                      .Finalize(&op.node_def));
198   INFER_ERROR("Shape must be rank 4 but is rank 3", op, "[1,2,3];?;?;?;?");
199   INFER_ERROR("Shape must be rank 1 but is rank 3", op, "?;[1,2,3];?;?;?");
200   INFER_ERROR("Shape must be rank 1 but is rank 3", op, "?;?;[1,2,3];?;?");
202   INFER_OK(op, "?;?;?;?;?", "[?,?,?,?];[?];[?];[?];[?]");
203   INFER_OK(op, "?;[1];?;?;?", "[?,?,?,d1_0];[d1_0];[d1_0];[d1_0];[d1_0]");
204   INFER_OK(op, "?;?;[1];?;?", "[?,?,?,d2_0];[d2_0];[d2_0];[d2_0];[d2_0]");
205   INFER_OK(op, "[1,2,3,4];[4];[4];?;?",
212   INFER_ERROR("Shape must be rank 4 but is rank 3", op, "[1,2,3];?;?;?;?");
213   INFER_ERROR("Shape must be rank 1 but is rank 3", op, "?;[1,2,3];?;?;?");
214   INFER_ERROR("Shape must be rank 1 but is rank 3", op, "?;?;[1,2,3];?;?");
216   INFER_OK(op, "?;?;?;?;?", "[?,?,?,?];[?];[?];[?];[?]");
217   INFER_OK(op, "?;[1];?;?;?", "[?,d1_0,?,?];[d1_0];[d1_0];[d1_0];[d1_0]");
218   INFER_OK(op, "?;?;[1];?;?", "[?,d2_0,?,?];[d2_0];[d2_0];[d2_0];[d2_0]");
219   INFER_OK(op, "[1,4,2,3];[4];[4];?;?",
226   INFER_ERROR("Shape must be rank 4 but is rank 3", op, "[1,2,3];?;?;?;?");
227   INFER_ERROR("Shape must be rank 1 but is rank 3", op, "?;[1,2,3];?;?;?");
228   INFER_ERROR("Shape must be rank 1 but is rank 3", op, "?;?;[1,2,3];?;?");
229   INFER_ERROR("Shape must be rank 1 but is rank 3", op, "?;?;?;[1,2,3];?");
230   INFER_ERROR("Shape must be rank 1 but is rank 3", op, "?;?;?;?;[1,2,3]");
232   INFER_OK(op, "?;?;?;?;?", "[?,?,?,?];[?];[?];[?];[?]");
233   INFER_OK(op, "?;[1];?;?;?", "[?,?,?,d1_0];[d1_0];[d1_0];[d1_0];[d1_0]");
234   INFER_OK(op, "?;?;[1];?;?", "[?,?,?,d2_0];[d2_0];[d2_0];[d2_0];[d2_0]");
235   INFER_OK(op, "?;?;?;[1];?", "[?,?,?,d3_0];[d3_0];[d3_0];[d3_0];[d3_0]");
236   INFER_OK(op, "?;?;?;?;[1]", "[?,?,?,d4_0];[d4_0];[d4_0];[d4_0];[d4_0]");
237   INFER_OK(op, "[1,2,3,4];[4];[4];[4];[4]",
244   INFER_ERROR("Shape must be rank 4 but is rank 3", op, "[1,2,3];?;?;?;?");
245   INFER_ERROR("Shape must be rank 1 but is rank 3", op, "?;[1,2,3];?;?;?");
246   INFER_ERROR("Shape must be rank 1 but is rank 3", op, "?;?;[1,2,3];?;?");
247   INFER_ERROR("Shape must be rank 1 but is rank 3", op, "?;?;?;[1,2,3];?");
248   INFER_ERROR("Shape must be rank 1 but is rank 3", op, "?;?;?;?;[1,2,3]");
250   INFER_OK(op, "?;?;?;?;?", "[?,?,?,?];[?];[?];[?];[?]");
251   INFER_OK(op, "?;[1];?;?;?", "[?,d1_0,?,?];[d1_0];[d1_0];[d1_0];[d1_0]");
252   INFER_OK(op, "?;?;[1];?;?", "[?,d2_0,?,?];[d2_0];[d2_0];[d2_0];[d2_0]");
253   INFER_OK(op, "?;?;?;[1];?", "[?,d3_0,?,?];[d3_0];[d3_0];[d3_0];[d3_0]");
254   INFER_OK(op, "?;?;?;?;[1]", "[?,d4_0,?,?];[d4_0];[d4_0];[d4_0];[d4_0]");
255   INFER_OK(op, "[1,4,2,3];[4];[4];[4];[4]",
262   ShapeInferenceTestOp op("FusedBatchNormGrad");
263   auto set_op = [&op](string data_format) {
271                      .Finalize(&op.node_def));
276   INFER_ERROR("Shape must be rank 4 but is rank 3", op, "[1,2,3];?;?;?;?");
277   INFER_ERROR("Shape must be rank 4 but is rank 3", op, "?;[1,2,3];?;?;?");
278   INFER_ERROR("Shape must be rank 1 but is rank 3", op, "?;?;[1,2,3];?;?");
279   INFER_ERROR("Shape must be rank 1 but is rank 3", op, "?;?;?;[1,2,3];?");
280   INFER_ERROR("Shape must be rank 1 but is rank 3", op, "?;?;?;?;[1,2,3]");
282   INFER_OK(op, "?;?;?;?;?", "[?,?,?,?];[?];[?];[0];[0]");
283   INFER_OK(op, "?;?;[1];?;?", "[?,?,?,d2_0];[d2_0];[d2_0];[0];[0]");
284   INFER_OK(op, "?;?;?;[1];?", "[?,?,?,d3_0];[d3_0];[d3_0];[0];[0]");
285   INFER_OK(op, "?;?;?;?;[1]", "[?,?,?,d4_0];[d4_0];[d4_0];[0];[0]");
286   INFER_OK(op, "[1,2,3,4];[1,2,3,4];[4];[4];[4]",
292   INFER_ERROR("Shape must be rank 4 but is rank 3", op, "[1,2,3];?;?;?;?");
293   INFER_ERROR("Shape must be rank 4 but is rank 3", op, "?;[1,2,3];?;?;?");
294   INFER_ERROR("Shape must be rank 1 but is rank 3", op, "?;?;[1,2,3];?;?");
295   INFER_ERROR("Shape must be rank 1 but is rank 3", op, "?;?;?;[1,2,3];?");
296   INFER_ERROR("Shape must be rank 1 but is rank 3", op, "?;?;?;?;[1,2,3]");
298   INFER_OK(op, "?;?;?;?;?", "[?,?,?,?];[?];[?];[0];[0]");
299   INFER_OK(op, "?;?;[1];?;?", "[?,d2_0,?,?];[d2_0];[d2_0];[0];[0]");
300   INFER_OK(op, "?;?;?;[1];?", "[?,d3_0,?,?];[d3_0];[d3_0];[0];[0]");
301   INFER_OK(op, "?;?;?;?;[1]", "[?,d4_0,?,?];[d4_0];[d4_0];[0];[0]");
302   INFER_OK(op, "[1,4,2,3];[1,4,2,3];[4];[4];[4]",
308   ShapeInferenceTestOp op("Conv3DBackpropInput");
311   INFER_ERROR("Shape must be rank 5 but is rank 3", op, "[1,2,3];?;?");
314   INFER_OK(op, "?;?;?", "[?,?,?,?,?]");
315   INFER_OK(op, "[?,?,?,?,?];?;?", "in0");
316   INFER_OK(op, "[?,2,?,4,?];?;?", "in0");
320   ShapeInferenceTestOp op("Conv3DBackpropFilter");
323   INFER_ERROR("Shape must be rank 5 but is rank 3", op, "?;[1,2,3];?");
326   INFER_OK(op, "?;?;?", "[?,?,?,?,?]");
327   INFER_OK(op, "?;[?,?,?,?,?];?", "in1");
328   INFER_OK(op, "?;[?,2,?,4,?];?", "in1");
332   ShapeInferenceTestOp op("MaxPool3DGrad");
335   INFER_ERROR("Shape must be rank 5 but is rank 3", op, "[1,2,3];?;?");
338   INFER_OK(op, "?;?;?", "[?,?,?,?,?]");
339   INFER_OK(op, "[?,?,?,?,?];?;?", "in0");
340   INFER_OK(op, "[?,2,?,4,?];?;?", "in0");
344   ShapeInferenceTestOp op("LRNGrad");
347   INFER_OK(op, "[1,?,?,4];[?,2,?,?];[?,?,3,?]", "[d0_0,d1_1,d2_2,d0_3]");
350   INFER_ERROR("Shape must be rank 4 but is rank 3", op, "[1,2,3];?;?");
351   INFER_ERROR("Shapes must be equal rank, but are 4 and 3", op, "?;[1,2,3];?");
352   INFER_ERROR("Shapes must be equal rank, but are 4 and 3", op, "?;?;[1,2,3]");
357     ShapeInferenceTestOp op(op_name);
360     INFER_ERROR("Shape must be rank 4 but is rank 3", op, "[1,2,3];?;?");
363     INFER_OK(op, "?;?;?", "[?,?,?,?]");
364     INFER_OK(op, "[?,?,?,?];?;?", "in0");
365     INFER_OK(op, "[?,2,?,4];?;?", "in0");
370   ShapeInferenceTestOp op("Dilation2DBackpropInput");
373   INFER_OK(op, "?;?;?", "in0");
374   INFER_OK(op, "?;[?,?,?,?,?];?", "in0");
375   INFER_OK(op, "?;[?,2,?,4,?];?", "in0");
379   ShapeInferenceTestOp op("Dilation2DBackpropFilter");
382   INFER_OK(op, "?;?;?", "in1");
383   INFER_OK(op, "?;[?,?,?,?,?];?", "in1");
384   INFER_OK(op, "?;[?,2,?,4,?];?", "in1");
390     ShapeInferenceTestOp op(op_name);
392     INFER_OK(op, "?;?", "in0|in1");
393     INFER_OK(op, "?;[1,?,3]", "in1");
394     INFER_OK(op, "[1,?,3];?", "in0");
395     INFER_OK(op, "[1,?];[?,2]", "[d0_0,d1_1]");
396     INFER_ERROR("Dimension 1 in both shapes must be equal, but are 3 and 2", op,
402   ShapeInferenceTestOp op("SoftmaxCrossEntropyWithLogits");
407   INFER_OK(op, "?;?", "[?];[?,?]");
408   INFER_OK(op, "[?,?];[?,?]", "[d0_0|d1_0];in0|in1");
409   INFER_OK(op, "[1,2];[?,2]", "[d0_0];in0");
410   INFER_OK(op, "[1,?];[?,2]", "[d0_0];[d0_0,d0_1|d1_1]");
411   INFER_OK(op, "[?,2];[1,2]", "[d1_0];in1");
413   INFER_ERROR("Dimension 0 in both shapes must be equal, but are 1 and 2", op,
415   INFER_ERROR("Shape must be rank 2 but is rank 3", op, "[1,2,3];?");
416   INFER_ERROR("Shapes must be equal rank, but are 2 and 3", op, "?;[1,2,3]");
420   ShapeInferenceTestOp op("SparseSoftmaxCrossEntropyWithLogits");
424   INFER_OK(op, "?;?", "[?];[?,?]");
425   INFER_OK(op, "[?,?];[?]", "[d0_0|d1_0];[d0_0|d1_0,d0_1]");
426   INFER_OK(op, "[1,2];[1]", "[d0_0|d1_0];[d0_0|d1_0,d0_1]");
427   INFER_OK(op, "[?,2];[1]", "[d1_0];[d1_0,d0_1]");
429   INFER_ERROR("Dimensions must be equal, but are 1 and 2", op, "[1,?];[2]");
430   INFER_ERROR("Shape must be rank 2 but is rank 3", op, "[1,2,3];?");
431   INFER_ERROR("Shape must be rank 1 but is rank 2", op, "?;[1,2]");
435   ShapeInferenceTestOp op("InTopK");
438   INFER_OK(op, "?;?", "[?]");
439   INFER_OK(op, "[?,?];[?]", "[d0_0|d1_0]");
440   INFER_OK(op, "[1,2];[1]", "[d0_0|d1_0]");
441   INFER_OK(op, "[?,2];[1]", "[d1_0]");
443   INFER_ERROR("Dimensions must be equal, but are 1 and 2", op, "[1,?];[2]");
444   INFER_ERROR("Shape must be rank 2 but is rank 3", op, "[1,2,3];?");
445   INFER_ERROR("Shape must be rank 1 but is rank 2", op, "?;[1,2]");
449   ShapeInferenceTestOp op("Dilation2D");
450   auto set_op = [&op](const std::vector<int32>& strides,
458                      .Finalize(&op.node_def));
464   INFER_OK(op, "[1,2,2,2];[1,1,2]", "[d0_0,2,2,d1_2]");
470   INFER_OK(op, "[1,7,7,2];[2,2,2]", "[d0_0,5,5,d1_2]");
475     ShapeInferenceTestOp op(op_name);
476     auto set_op = [&op, op_name](const std::vector<float>& pooling_ratio) {
480                        .Finalize(&op.node_def));
486     INFER_ERROR("must be rank 4", op, "[?,?,?]");
489     INFER_OK(op, "?", "[?,?,?,?];[?];[?]");
490     INFER_OK(op, "[?,?,?,?]", "[?,?,?,?];[?];[?]");
492     INFER_OK(op, "[10,20,30,40]", "[5,20,45,80];[20];[45]");
493     INFER_OK(op, "[?,20,30,40]", "[?,20,45,80];[20];[45]");
494     INFER_OK(op, "[10,?,30,40]", "[5,?,45,80];[?];[45]");
495     INFER_OK(op, "[10,20,?,40]", "[5,20,?,80];[20];[?]");
496     INFER_OK(op, "[10,20,30,?]", "[5,20,45,?];[20];[45]");
500     INFER_ERROR("pooling_ratio field", op, "?");
502     INFER_ERROR("pooling_ratio field", op, "?");
506     INFER_ERROR("is negative", op, "[1,2,3,4]");
511   ShapeInferenceTestOp op("FractionalMaxPoolGrad");
514   INFER_ERROR("must be rank 4", op, "[?,?,?];?;?;?;?");
515   INFER_OK(op, "?;?;?;?;?", "[?,?,?,?]");
516   INFER_OK(op, "[?,?,3,4];?;?;?;?", "in0");
520   ShapeInferenceTestOp op("FractionalAvgPoolGrad");
521   op.input_tensors.resize(1);
524   INFER_OK(op, "?;?;?;?", "[?,?,?,?]");
529   op.input_tensors[0] = &shape_t;
530   INFER_OK(op, "[5];?;?;?", "[1,2,3,4]");