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

Lines Matching refs:op
17 #include "tensorflow/core/framework/op.h"
26   ShapeInferenceTestOp op("SparseTensorDenseAdd");
29   INFER_OK(op, "?;?;?;?", "in3");
33   ShapeInferenceTestOp op("SparseAdd");
35   INFER_OK(op, "?;?;?;?;?;?;?", "[?,?];[?];[?]");
38   INFER_OK(op, "?;?;[?];?;?;?;?", "[?,d2_0];[?];in2");
39   INFER_OK(op, "?;?;[1];?;?;?;?", "[?,d2_0];[?];in2");
43   ShapeInferenceTestOp op("SparseAddGrad");
46   INFER_ERROR("must be rank 2", op, "?;?;[1];?");
47   INFER_ERROR("must be rank 2", op, "?;[1];?;?");
49   INFER_OK(op, "?;?;?;?", "[?];[?]");
52   INFER_OK(op, "?;[?,?];[?,?];?", "[d1_0];[d2_0]");
56   ShapeInferenceTestOp op("SparseReorder");
61   INFER_ERROR("must be rank 2", op, "[1];?;?");
62   INFER_ERROR("must be rank 1", op, "?;[];?");
63   INFER_ERROR("must be rank 1", op, "?;?;[]");
66   INFER_OK(op, "?;?;?", "[?,?];[?]");
69   INFER_OK(op, "[?,?];[?];?", "in0;in1");
73   ShapeInferenceTestOp op("SparseReshape");
78   INFER_ERROR("must be rank 2", op, "[1];?;?");
79   INFER_ERROR("must be rank 1", op, "?;[];?");
80   INFER_ERROR("must be rank 1", op, "?;?;[]");
83   INFER_OK(op, "?;?;?", "[?,?];[?]");
87   INFER_OK(op, "[?,?];?;[?]", "[d0_0,d2_0];in2");
91   ShapeInferenceTestOp op("SparseSplit");
98                    .Finalize(&op.node_def));
105   INFER_OK(op, "?;?;?;?", "[?,?];[?,?];[?];[?];in3;in3");
106   INFER_OK(op, "?;?;?;[5,4,3,2,1]", "[?,120];[?,120];[?];[?];in3;in3");
110   ShapeInferenceTestOp op("SparseToDense");
111   op.input_tensors.resize(4);
114   INFER_OK(op, "?;?;?;?", "?");
115   INFER_OK(op, "?;[?];?;?", "?");
116   INFER_OK(op, "?;[4];?;?", "[?,?,?,?]");
118   op.input_tensors[1] = &in_t;
119   INFER_OK(op, "?;[4];?;?", "[1,2,3,4]");
123   ShapeInferenceTestOp op("SparseReduceSum");
126   INFER_OK(op, "?;?;?;?", "?");
130   ShapeInferenceTestOp op("SerializeSparse");
133   INFER_ERROR("must be rank 2", op, "[1];?;?");
134   INFER_ERROR("must be rank 1", op, "?;[];?");
135   INFER_ERROR("must be rank 1", op, "?;?;[]");
138   INFER_OK(op, "?;?;?", "[3]");
142   ShapeInferenceTestOp op("SerializeManySparse");
145   INFER_ERROR("must be rank 2", op, "[1];?;?");
146   INFER_ERROR("must be rank 1", op, "?;[];?");
147   INFER_ERROR("must be rank 1", op, "?;?;[]");
150   INFER_OK(op, "?;?;?", "[?,3]");
154   ShapeInferenceTestOp op("DeserializeManySparse");
157   INFER_ERROR("must be rank 2", op, "[1]");
158   INFER_ERROR("must be 3", op, "[?,4]");
161   INFER_OK(op, "?", "[?,?];[?];[?]");
162   INFER_OK(op, "[?,3]", "[?,?];[?];[?]");
166   ShapeInferenceTestOp op("SparseTensorDenseMatMul");
167   auto set_adjoints = [&op](bool adjoint_a, bool adjoint_b) {
175                      .Finalize(&op.node_def));
182   INFER_ERROR("must be rank 2", op, "[1];?;?;?");
183   INFER_ERROR("must be rank 1", op, "?;[];?;?");
184   INFER_ERROR("must be rank 1", op, "?;?;[];?");
185   INFER_ERROR("must be rank 2", op, "?;?;[3];?");
186   INFER_ERROR("must be rank 2", op, "?;?;?;[]");
189   INFER_OK(op, "?;?;?;?", "[?,?]");
190   INFER_OK(op, "?;?;?;[?,?]", "[?,d3_1]");    // use d3_1, !adjoint_b.
191   INFER_OK(op, "?;?;?;[1,2]", "[?,d3_1]");    // use d3_1, !adjoint_b.
192   INFER_OK(op, "?;?;[2];[1,2]", "[?,d3_1]");  // use d3_1, !adjoint_b.
195   INFER_OK(op, "?;?;?;[?,?]", "[?,d3_0]");  // use d3_0, adjoint_b.
196   INFER_OK(op, "?;?;?;[1,2]", "[?,d3_0]");  // use d3_0, adjoint_b.
201   op.input_tensors.resize(4);
202   op.input_tensors[2] = &a_shape_t;
206   INFER_OK(op, "?;?;[2];[1,2]", "[3,d3_1]");  // use d3_1, !adjoint_b.
207   INFER_OK(op, "?;?;?;[1,2]", "[3,d3_1]");    // use d3_1, !adjoint_b.
211   INFER_ERROR("must be equal", op, "?;?;[2];[1,2]");  // adjoint_a, !adjoint_b.
215   INFER_ERROR("must be rank 2 but is rank 3", op, "?;?;[3];[1,2]");
219   ShapeInferenceTestOp op("SparseSoftmax");
224   INFER_ERROR("must be rank 2", op, "[1];?;?");
225   INFER_ERROR("must be rank 1", op, "?;[];?");
226   INFER_ERROR("must be rank 1", op, "?;?;[]");
229   INFER_OK(op, "?;?;?", "[?]");
230   INFER_OK(op, "?;[?];?", "in1");
231   INFER_OK(op, "?;[5];?", "in1");
236     ShapeInferenceTestOp op(op_name);
239     INFER_ERROR("must be rank 2", op, "[1];?;?;?;?;?");  // a_indices
240     INFER_ERROR("must be rank 1", op, "?;[];?;?;?;?");   // a_values
241     INFER_ERROR("must be rank 1", op, "?;?;[];?;?;?");   // a_shape
242     INFER_ERROR("must be rank 2", op, "?;?;?;[];?;?");   // b_indices
243     INFER_ERROR("must be rank 1", op, "?;?;?;?;[];?");   // b_values
244     INFER_ERROR("must be rank 1", op, "?;?;?;?;?;[]");   // b_shape
247     INFER_OK(op, "?;?;?;?;?;?", "[?,?];[?]");
248     INFER_OK(op, "?;[?];?;?;?;?", "[?,?];[?]");
249     INFER_OK(op, "?;[5];?;?;?;?", "[?,?];[?]");
254   ShapeInferenceTestOp op("SparseConcat");
264                    .Finalize(&op.node_def));
267   INFER_ERROR("must be rank 2", op, "[1];?;?;?;?;?");  // indices
268   INFER_ERROR("must be rank 2", op, "?;[1];?;?;?;?");  // indices
269   INFER_ERROR("must be rank 1", op, "?;?;[];?;?;?");   // values
270   INFER_ERROR("must be rank 1", op, "?;?;?;[];?;?");   // values
271   INFER_ERROR("must be rank 1", op, "?;?;?;?;[];?");   // shapes
272   INFER_ERROR("must be rank 1", op, "?;?;?;?;?;[]");   // shapes
282   INFER_OK(op, "?;?;?;?;?;?", "[?,?];[?];[?]");
283   INFER_OK(op, "?;?;?;?;[?];[?]", "[?,?];[?];in4|in5");
284   INFER_OK(op, "?;?;?;?;[?];[5]", "[?,?];[?];in5");
287   INFER_OK(op, "[4,5];[3,?];?;?;?;?", "[7,d0_1];[7];[?]");
290   INFER_OK(op, "?;?;[4];[3];?;?", "[7,?];[7];[?]");
293   INFER_OK(op, "[?,2];[3,?];[4];[?];?;?", "[7,d0_1];[7];[?]");
296   INFER_ERROR("but are 100 and 200", op, "[100,?];[?,?];[200];[?];?;?");
297   INFER_ERROR("but are 2 and 3", op, "[?,2];[?,3];[?];[?];?;?");
298   INFER_ERROR("but are 4 and 5", op, "?;?;?;?;[4];[5]");
304     ShapeInferenceTestOp op(op_name);
307     INFER_OK(op, "?;?;?;?", "[?]");
310     INFER_OK(op, "[?,?];?;?;?", "[d0_0]");
313     INFER_ERROR("must be rank 2", op, "[1];?;?;?");
318   ShapeInferenceTestOp op("AddSparseToTensorsMap");
321   INFER_ERROR("must be rank 2", op, "[1];?;?");
322   INFER_ERROR("must be rank 1", op, "?;[];?");
323   INFER_ERROR("must be rank 1", op, "?;?;[]");
326   INFER_OK(op, "?;?;?", "[]");
330   ShapeInferenceTestOp op("AddManySparseToTensorsMap");
333   INFER_ERROR("must be rank 2", op, "[1];?;?");
334   INFER_ERROR("must be rank 1", op, "?;[];?");
335   INFER_ERROR("must be rank 1", op, "?;?;[]");
338   INFER_OK(op, "?;?;?", "[?]");
342   ShapeInferenceTestOp op("TakeManySparseFromTensorsMap");
345   INFER_ERROR("must be rank 1", op, "[?,1]");
348   INFER_OK(op, "?", "[?,?];[?];[?]");
349   INFER_OK(op, "[?]", "[?,?];[?];[?]");