Lines Matching refs:op
17 #include "tensorflow/core/framework/op.h"
25 ShapeInferenceTestOp op("Multinomial");
26 op.input_tensors.resize(2);
28 INFER_OK(op, "?;?", "[?,?]");
29 INFER_ERROR("Shape must be rank 2 but is rank 1", op, "[?];?");
30 INFER_OK(op, "[?,?];?", "[d0_0,?]");
31 INFER_OK(op, "[2,?];?", "[d0_0,?]");
32 INFER_OK(op, "[2,1];?", "[d0_0,?]");
34 op.input_tensors[1] = &num_samples;
35 INFER_OK(op, "[2,1];[]", "[d0_0,3]");
37 INFER_ERROR("Shape must be rank 0 but is rank 1", op, "[2,1];[3]");
41 ShapeInferenceTestOp op("RandomGamma");
42 op.input_tensors.resize(2);
44 INFER_OK(op, "?;?", "?");
45 INFER_OK(op, "?;[3]", "?");
46 INFER_OK(op, "[1];?", "?");
47 INFER_ERROR("Shape must be rank 1 but is rank 2", op, "[1,2];[3,4]");
49 op.input_tensors[0] = &shape;
50 INFER_OK(op, "[3];[4,?]", "[1,2,3,d1_0,d1_1]");
51 INFER_OK(op, "[3];[4,5]", "[1,2,3,d1_0,d1_1]");
52 INFER_OK(op, "[3];[]", "[1,2,3]");
56 ShapeInferenceTestOp op("RandomPoisson");
57 op.input_tensors.resize(2);
59 INFER_OK(op, "?;?", "?");
60 INFER_OK(op, "?;[3]", "?");
61 INFER_OK(op, "[1];?", "?");
62 INFER_ERROR("Shape must be rank 1 but is rank 2", op, "[1,2];[3,4]");
64 op.input_tensors[0] = &shape;
65 INFER_OK(op, "[3];[4,?]", "[1,2,3,d1_0,d1_1]");
66 INFER_OK(op, "[3];[4,5]", "[1,2,3,d1_0,d1_1]");
67 INFER_OK(op, "[3];[]", "[1,2,3]");