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

Lines Matching refs:op
17 #include "tensorflow/core/framework/op.h"
25   ShapeInferenceTestOp op("DenseToDenseSetOperation");
26   op.input_tensors.resize(3);
27   INFER_ERROR("Wrong number of inputs passed", op, "?;?;?");
31   ShapeInferenceTestOp op("DenseToDenseSetOperation");
34   INFER_OK(op, "?;?", "[?,?];[?];[?]");
37   INFER_ERROR("Shape must be at least rank 2 but is rank 1", op, "[?];?");
38   INFER_ERROR("Shape must be at least rank 2 but is rank 1", op, "?;[?]");
39   INFER_ERROR("Shape must be at least rank 2 but is rank 1", op, "[2];?");
40   INFER_ERROR("Shape must be at least rank 2 but is rank 1", op, "?;[2]");
43   INFER_ERROR("Shape must be rank 2 but is rank 3", op, "[?,?];[?,?,?]");
44   INFER_ERROR("Shape must be rank 3 but is rank 2", op, "[?,?,?];[?,?]");
45   INFER_ERROR("Shape must be rank 2 but is rank 3", op, "[2,1];[2,1,2]");
46   INFER_ERROR("Shape must be rank 3 but is rank 2", op, "[2,1,2];[2,1]");
49   INFER_OK(op, "[?,?];?", "[?,2];[?];[2]");
50   INFER_OK(op, "?;[?,?]", "[?,2];[?];[2]");
51   INFER_OK(op, "[?,?];[?,?]", "[?,2];[?];[2]");
54   INFER_OK(op, "[?,?,?,?];?", "[?,4];[?];[4]");
55   INFER_OK(op, "?;[?,?,?,?]", "[?,4];[?];[4]");
56   INFER_OK(op, "[?,?,?,?];[?,?,?,?]", "[?,4];[?];[4]");
59   INFER_OK(op, "[5,3,2,1];?", "[?,4];[?];[4]");
60   INFER_OK(op, "?;[5,3,2,1]", "[?,4];[?];[4]");
61   INFER_OK(op, "[5,3,2,1];[?,?,?,?]", "[?,4];[?];[4]");
62   INFER_OK(op, "[?,?,?,?];[5,3,2,1]", "[?,4];[?];[4]");
63   INFER_OK(op, "[5,3,2,1];[?,?,?,?]", "[?,4];[?];[4]");
66   INFER_ERROR("Dimension 0 in both shapes must be equal", op,
68   INFER_ERROR("Dimension 2 in both shapes must be equal", op,
72   INFER_OK(op, "[4,5,6,7];[?,?,?,?]", "[?,4];[?];[4]");
73   INFER_OK(op, "[4,5,6,7];[?,?,?,4]", "[?,4];[?];[4]");
74   INFER_OK(op, "[?,?,?,?];[4,5,6,7]", "[?,4];[?];[4]");
75   INFER_OK(op, "[4,?,2,?];[?,1,?,5]", "[?,4];[?];[4]");
76   INFER_OK(op, "[4,5,6,7];[4,?,6,?]", "[?,4];[?];[4]");
77   INFER_OK(op, "[4,5,6,7];[4,5,6,4]", "[?,4];[?];[4]");
81   ShapeInferenceTestOp op("DenseToSparseSetOperation");
82   op.input_tensors.resize(5);
83   INFER_ERROR("Wrong number of inputs passed", op, "?;?;?;?;?");
87   ShapeInferenceTestOp op("DenseToSparseSetOperation");
88   INFER_OK(op, "?;?;?;?", "[?,?];[?];[?]");
91   INFER_OK(op, "?;?;?;?", "[?,?];[?];[?]");
92   INFER_OK(op, "?;[?,?];[?];[?]", "[?,?];[?];[?]");
95   INFER_ERROR("Shape must be at least rank 2 but is rank 1", op, "[?];?;?;?");
96   INFER_ERROR("Shape must be at least rank 2 but is rank 1", op,
98   INFER_ERROR("Shape must be at least rank 2 but is rank 1", op,
100   INFER_ERROR("Shape must be at least rank 2 but is rank 1", op, "[2];?;?;?");
101   INFER_ERROR("Shape must be at least rank 2 but is rank 1", op,
103   INFER_ERROR("Shape must be at least rank 2 but is rank 1", op,
107   INFER_OK(op, "[?,?];?;?;?", "[?,2];[?];[2]");
108   INFER_OK(op, "[?,?];[?,?];[?];[?]", "[?,2];[?];[2]");
111   INFER_OK(op, "?;[?,2];[?];[2]", "[?,d3_0];[?];[d3_0]");
112   INFER_OK(op, "?;[5,2];[5];[2]", "[?,d3_0];[?];[d3_0]");
115   INFER_OK(op, "[?,?];[5,2];[5];[2]", "[?,2];[?];[2]");
116   INFER_OK(op, "[4,3];[5,2];[5];[2]", "[?,2];[?];[2]");
119   INFER_ERROR("elements in index (5) and values (6) do not match", op,
121   INFER_ERROR("rank (3) and shape rank (4) do not match", op,
126   ShapeInferenceTestOp op("SparseToSparseSetOperation");
127   op.input_tensors.resize(7);
128   INFER_ERROR("Wrong number of inputs passed", op, "?;?;?;?;?;?;?");
132   ShapeInferenceTestOp op("SparseToSparseSetOperation");
135   INFER_OK(op, "?;?;?;?;?;?", "[?,?];[?];[?]");
136   INFER_OK(op, "[?,?];[?];[?];[?,?];[?];[?]", "[?,?];[?];[?]");
137   INFER_OK(op, "?;?;?;[?,?];[?];[?]", "[?,?];[?];[?]");
138   INFER_OK(op, "[?,?];[?];[?];?;?;?", "[?,?];[?];[?]");
141   INFER_OK(op, "[?,2];[?];[2];?;?;?", "[?,d2_0];[?];[d2_0]");
142   INFER_OK(op, "?;?;?;[?,2];[?];[2]", "[?,d5_0];[?];[d5_0]");
143   INFER_OK(op, "[?,2];[?];[2];[?,?];[?];[?]", "[?,d2_0];[?];[d2_0]");
144   INFER_OK(op, "[?,?];[?];[?];[?,2];[?];[2]", "[?,d5_0];[?];[d5_0]");
147   INFER_OK(op, "[?,2];[?];[2];[?,2];[?];[2]", "[?,d2_0];[?];[d2_0]");