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

Lines Matching refs:op
17 #include "tensorflow/core/framework/op.h"
25   ShapeInferenceTestOp op("MatrixDeterminant");
26   INFER_OK(op, "?", "?");
27   INFER_ERROR("Shape must be at least rank 2 but is rank 1", op, "[1]");
28   INFER_ERROR("Dimensions must be equal, but are 2 and 1", op, "[1,?,3,4,1,2]");
30   INFER_OK(op, "[?,?]", "[]");
31   INFER_OK(op, "[1,?]", "[]");
32   INFER_OK(op, "[?,1]", "[]");
35   INFER_OK(op, "[1,?,3,4,?,?]", "[d0_0,d0_1,d0_2,d0_3]");
36   INFER_OK(op, "[1,?,3,4,1,?]", "[d0_0,d0_1,d0_2,d0_3]");
37   INFER_OK(op, "[1,?,3,4,?,1]", "[d0_0,d0_1,d0_2,d0_3]");
42     ShapeInferenceTestOp op(op_name);
44     const string extra_shape = (op.name == "CholeskyGrad" ? ";?" : "");
46     INFER_OK(op, "?" + extra_shape, "?");
47     INFER_ERROR("Shape must be at least rank 2 but is rank 1", op,
49     INFER_ERROR("Dimensions must be equal, but are 1 and 2", op,
52     INFER_OK(op, "[?,?]" + extra_shape, "[d0_0|d0_1,d0_0|d0_1]");
53     INFER_OK(op, "[1,?]" + extra_shape, "[d0_0,d0_0]");
54     INFER_OK(op, "[?,1]" + extra_shape, "[d0_1,d0_1]");
57     INFER_OK(op, "[5,?,7,?,?]" + extra_shape,
59     INFER_OK(op, "[5,?,7,1,?]" + extra_shape, "[d0_0,d0_1,d0_2,d0_3,d0_3]");
60     INFER_OK(op, "[5,?,7,?,1]" + extra_shape, "[d0_0,d0_1,d0_2,d0_4,d0_4]");
65   ShapeInferenceTestOp op("SelfAdjointEig");
66   INFER_OK(op, "?", "?");
67   INFER_ERROR("Shape must be at least rank 2 but is rank 1", op, "[1]");
68   INFER_ERROR("Dimensions must be equal, but are 1 and 2", op, "[1,2]");
70   INFER_OK(op, "[?,?]", "[?,d0_0|d0_1]");
71   INFER_OK(op, "[1,?]", "[2,d0_0]");
72   INFER_OK(op, "[?,1]", "[2,d0_1]");
75   INFER_OK(op, "[5,?,7,?,?]", "[d0_0,d0_1,d0_2,?,d0_3|d0_4]");
76   INFER_OK(op, "[5,?,7,1,?]", "[d0_0,d0_1,d0_2,2,d0_3]");
77   INFER_OK(op, "[5,?,7,?,1]", "[d0_0,d0_1,d0_2,2,d0_4]");
81   ShapeInferenceTestOp op("SelfAdjointEigV2");
82   auto set_compute_v = [&op](bool compute_v) {
86                      .Finalize(&op.node_def));
90   INFER_ERROR("Shape must be at least rank 2 but is rank 1", op, "[1]");
91   INFER_ERROR("Dimensions must be equal, but are 1 and 2", op, "[1,2]");
92   INFER_ERROR("Dimensions must be equal, but are 1 and 2", op, "[3,1,2]");
94   INFER_OK(op, "?", "?;[0]");
95   INFER_OK(op, "[?,?]", "[d0_0|d0_1];[0]");
96   INFER_OK(op, "[1,?]", "[d0_0|d0_1];[0]");
97   INFER_OK(op, "[?,1]", "[d0_0|d0_1];[0]");
100   INFER_OK(op, "[5,?,7,?,?]", "[d0_0,d0_1,d0_2,d0_3|d0_4];[0]");
101   INFER_OK(op, "[5,?,7,1,?]", "[d0_0,d0_1,d0_2,d0_3|d0_4];[0]");
102   INFER_OK(op, "[5,?,7,?,1]", "[d0_0,d0_1,d0_2,d0_3|d0_4];[0]");
105   INFER_OK(op, "?", "?;?");
106   INFER_OK(op, "[?,?]", "[d0_0|d0_1];[d0_0|d0_1,d0_0|d0_1]");
107   INFER_OK(op, "[1,?]", "[d0_0|d0_1];[d0_0|d0_1,d0_0|d0_1]");
108   INFER_OK(op, "[?,1]", "[d0_0|d0_1];[d0_0|d0_1,d0_0|d0_1]");
111   INFER_OK(op, "[5,?,7,?,?]",
113   INFER_OK(op, "[5,?,7,1,?]",
115   INFER_OK(op, "[5,?,7,?,1]",
121     ShapeInferenceTestOp op(op_name);
122     INFER_OK(op, "?;?", "?");
123     INFER_ERROR("Shape must be at least rank 2 but is rank 1", op, "[1];?");
124     INFER_ERROR("Dimensions must be equal, but are 1 and 2", op, "[1,2];?");
125     INFER_ERROR("Shape must be at least rank 2 but is rank 1", op,
127     INFER_ERROR("Shapes must be equal rank, but are 0 and 1", op,
130     INFER_OK(op, "[?,?];?", "[d0_0|d0_1,?]");
134     INFER_OK(op, "[?,?];[?,?]", "[d0_0,d1_1]");
135     INFER_OK(op, "[?,?];[1,?]", "[d1_0,d1_1]");
136     INFER_OK(op, "[1,?];[1,?]", "[d0_0|d1_0,d1_1]");
137     INFER_OK(op, "[?,1];[1,?]", "[d0_1|d1_0,d1_1]");
138     INFER_OK(op, "[1,1];[?,?]", "[d0_0,d1_1]");
139     INFER_OK(op, "[1,1];[1,?]", "[d0_0|d0_1|d1_0,d1_1]");
141     INFER_OK(op, "[10,?,?,?];[?,20,1,?]", "[d0_0,d1_1,d1_2,d1_3]");
142     INFER_OK(op, "[10,?,1,?];[?,20,1,?]", "[d0_0,d1_1,d0_2|d1_2,d1_3]");
143     INFER_OK(op, "[10,?,?,1];[?,20,1,?]", "[d0_0,d1_1,d0_3|d1_2,d1_3]");
144     INFER_OK(op, "[10,?,1,1];[?,20,?,?]", "[d0_0,d1_1,d0_2,d1_3]");
145     INFER_OK(op, "[10,?,1,1];[?,20,1,?]", "[d0_0,d1_1,d0_2|d0_3|d1_2,d1_3]");
150   ShapeInferenceTestOp op("MatrixSolveLs");
151   INFER_OK(op, "?;?;?", "?");
152   INFER_OK(op, "?;?;[]", "?");
157   INFER_OK(op, "[1,?];[1,?];?", "[d0_1,d1_1]");
158   INFER_OK(op, "[1,2];[1,3];?", "[d0_1,d1_1]");
159   INFER_ERROR("Dimensions must be equal, but are 5 and 6", op, "[5,?];[6,?];?");
161   INFER_OK(op, "[10,?,1,?];[?,20,1,?];?", "[d0_0,d1_1,d0_3,d1_3]");
162   INFER_OK(op, "[10,20,1,2];[10,20,1,3];?", "[d0_0|d1_0,d0_1|d1_1,d0_3,d1_3]");
163   INFER_ERROR("Dimensions must be equal, but are 5 and 6", op,
165   INFER_ERROR("Dimension 0 in both shapes must be equal, but are 10 and 11", op,
169   INFER_ERROR("Shape must be at least rank 2 but is rank 1", op, "[?];?;?");
170   INFER_ERROR("Shape must be at least rank 2 but is rank 1", op, "?;[?];?");
171   INFER_ERROR("Shape must be rank 0 but is rank 1", op, "?;?;[1]");
175   ShapeInferenceTestOp op("Qr");
176   auto set_attrs = [&op](bool full_matrices) {
180                      .Finalize(&op.node_def));
190   INFER_OK(op, "?", "?;?");
191   INFER_OK(op, "[?,?,?]", "[d0_0,d0_1,?];[d0_0,?,d0_2]");
192   INFER_OK(op, "[4,?,?]", "[d0_0,d0_1,?];[d0_0,?,d0_2]");
193   INFER_OK(op, "[4,2,?]", "[d0_0,d0_1,?];[d0_0,?,d0_2]");
194   INFER_OK(op, "[4,?,2]", "[d0_0,d0_1,?];[d0_0,?,d0_2]");
195   INFER_OK(op, "[?,2,2]", "[d0_0,d0_1,d0_1];[d0_0,d0_1,d0_2]");
196   INFER_OK(op, "[4,2,2]", "[d0_0,d0_1,d0_1];[d0_0,d0_1,d0_2]");
197   INFER_OK(op, "[?,3,2]", "[d0_0,d0_1,d0_2];[d0_0,d0_2,d0_2]");
198   INFER_OK(op, "[4,3,2]", "[d0_0,d0_1,d0_2];[d0_0,d0_2,d0_2]");
199   INFER_OK(op, "[?,2,3]", "[d0_0,d0_1,d0_1];[d0_0,d0_1,d0_2]");
200   INFER_OK(op, "[4,2,3]", "[d0_0,d0_1,d0_1];[d0_0,d0_1,d0_2]");
201   INFER_ERROR("Shape must be at least rank 2 but is rank 1", op, "[1]");
204   INFER_OK(op, "?", "?;?");
205   INFER_OK(op, "[?,?,?]", "[d0_0,d0_1,d0_1];[d0_0,d0_1,d0_2]");
206   INFER_OK(op, "[4,?,?]", "[d0_0,d0_1,d0_1];[d0_0,d0_1,d0_2]");
207   INFER_OK(op, "[4,2,?]", "[d0_0,d0_1,d0_1];[d0_0,d0_1,d0_2]");
208   INFER_OK(op, "[4,?,2]", "[d0_0,d0_1,d0_1];[d0_0,d0_1,d0_2]");
209   INFER_OK(op, "[?,2,2]", "[d0_0,d0_1,d0_1];[d0_0,d0_1,d0_2]");
210   INFER_OK(op, "[4,2,2]", "[d0_0,d0_1,d0_1];[d0_0,d0_1,d0_2]");
211   INFER_OK(op, "[?,3,2]", "[d0_0,d0_1,d0_1];[d0_0,d0_1,d0_2]");
212   INFER_OK(op, "[4,3,2]", "[d0_0,d0_1,d0_1];[d0_0,d0_1,d0_2]");
213   INFER_OK(op, "[?,2,3]", "[d0_0,d0_1,d0_1];[d0_0,d0_1,d0_2]");
214   INFER_OK(op, "[4,2,3]", "[d0_0,d0_1,d0_1];[d0_0,d0_1,d0_2]");
215   INFER_ERROR("Shape must be at least rank 2 but is rank 1", op, "[1]");
219   ShapeInferenceTestOp op("Svd");
220   auto set_attrs = [&op](bool compute_uv, bool full_matrices) {
225                      .Finalize(&op.node_def));
235   INFER_OK(op, "?", "?;[0];[0]");
236   INFER_OK(op, "[?,?,?]", "[d0_0,?];[0];[0]");
237   INFER_OK(op, "[4,?,?]", "[d0_0,?];[0];[0]");
238   INFER_OK(op, "[4,2,?]", "[d0_0,?];[0];[0]");
239   INFER_OK(op, "[4,?,2]", "[d0_0,?];[0];[0]");
240   INFER_OK(op, "[?,2,2]", "[d0_0,d0_1];[0];[0]");
241   INFER_OK(op, "[4,2,2]", "[d0_0,d0_1];[0];[0]");
242   INFER_OK(op, "[?,3,2]", "[d0_0,d0_2];[0];[0]");
243   INFER_OK(op, "[4,3,2]", "[d0_0,d0_2];[0];[0]");
244   INFER_OK(op, "[?,2,3]", "[d0_0,d0_1];[0];[0]");
245   INFER_OK(op, "[4,2,3]", "[d0_0,d0_1];[0];[0]");
246   INFER_ERROR("Shape must be at least rank 2 but is rank 1", op, "[1]");
249   INFER_OK(op, "?", "?;?;?");
250   INFER_OK(op, "[?,?,?]", "[d0_0,?];[d0_0,d0_1,?];[d0_0,d0_2,?]");
251   INFER_OK(op, "[4,?,?]", "[d0_0,?];[d0_0,d0_1,?];[d0_0,d0_2,?]");
252   INFER_OK(op, "[4,2,?]", "[d0_0,?];[d0_0,d0_1,?];[d0_0,d0_2,?]");
253   INFER_OK(op, "[4,?,2]", "[d0_0,?];[d0_0,d0_1,?];[d0_0,d0_2,?]");
254   INFER_OK(op, "[?,2,2]", "[d0_0,d0_1];[d0_0,d0_1,d0_1];[d0_0,d0_2,d0_1]");
255   INFER_OK(op, "[4,2,2]", "[d0_0,d0_1];[d0_0,d0_1,d0_1];[d0_0,d0_2,d0_1]");
256   INFER_OK(op, "[?,3,2]", "[d0_0,d0_2];[d0_0,d0_1,d0_2];[d0_0,d0_2,d0_2]");
257   INFER_OK(op, "[4,3,2]", "[d0_0,d0_2];[d0_0,d0_1,d0_2];[d0_0,d0_2,d0_2]");
258   INFER_OK(op, "[?,2,3]", "[d0_0,d0_1];[d0_0,d0_1,d0_1];[d0_0,d0_2,d0_1]");
259   INFER_OK(op, "[4,2,3]", "[d0_0,d0_1];[d0_0,d0_1,d0_1];[d0_0,d0_2,d0_1]");
260   INFER_ERROR("Shape must be at least rank 2 but is rank 1", op, "[1]");
263   INFER_OK(op, "?", "?;?;?");
264   INFER_OK(op, "[?,?,?]", "[d0_0,?];[d0_0,d0_1,d0_1];[d0_0,d0_2,d0_2]");
265   INFER_OK(op, "[4,?,?]", "[d0_0,?];[d0_0,d0_1,d0_1];[d0_0,d0_2,d0_2]");
266   INFER_OK(op, "[4,2,?]", "[d0_0,?];[d0_0,d0_1,d0_1];[d0_0,d0_2,d0_2]");
267   INFER_OK(op, "[4,?,2]", "[d0_0,?];[d0_0,d0_1,d0_1];[d0_0,d0_2,d0_2]");
268   INFER_OK(op, "[?,2,2]", "[d0_0,d0_1];[d0_0,d0_1,d0_1];[d0_0,d0_2,d0_2]");
269   INFER_OK(op, "[4,2,2]", "[d0_0,d0_1];[d0_0,d0_1,d0_1];[d0_0,d0_2,d0_2]");
270   INFER_OK(op, "[?,3,2]", "[d0_0,d0_2];[d0_0,d0_1,d0_1];[d0_0,d0_2,d0_2]");
271   INFER_OK(op, "[4,3,2]", "[d0_0,d0_2];[d0_0,d0_1,d0_1];[d0_0,d0_2,d0_2]");
272   INFER_OK(op, "[?,2,3]", "[d0_0,d0_1];[d0_0,d0_1,d0_1];[d0_0,d0_2,d0_2]");
273   INFER_OK(op, "[4,2,3]", "[d0_0,d0_1];[d0_0,d0_1,d0_1];[d0_0,d0_2,d0_2]");
274   INFER_ERROR("Shape must be at least rank 2 but is rank 1", op, "[1]");