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

Lines Matching refs:op
17 #include "tensorflow/core/framework/op.h"
25   ShapeInferenceTestOp op("CTCLoss");
30   INFER_ERROR("must be rank 3", op, "[];?;?;?");  // inputs
31   INFER_ERROR("must be rank 2", op, "?;[];?;?");  // labels_indices
32   INFER_ERROR("must be rank 1", op, "?;?;[];?");  // labels_values
33   INFER_ERROR("must be rank 1", op, "?;?;?;[]");  // sequence_length
36   INFER_ERROR("must be equal", op, "?;[1,?];[2];?");
41   INFER_OK(op, "[?,?,?];?;?;[?]", "[d0_1|d3_0];[d0_0,d0_1|d3_0,d0_2]");
42   INFER_OK(op, "[?,1,?];?;?;[1]", "[d0_1|d3_0];[d0_0,d0_1|d3_0,d0_2]");
43   INFER_OK(op, "[?,?,?];?;?;[1]", "[d3_0];[d0_0,d3_0,d0_2]");
44   INFER_OK(op, "[?,1,?];?;?;[?]", "[d0_1];[d0_0,d0_1,d0_2]");
45   INFER_ERROR("must be equal", op, "[?,1,?];?;?;[2]");
49   ShapeInferenceTestOp op("CTCGreedyDecoder");
54   INFER_ERROR("must be rank 3", op, "[];?");  // inputs
55   INFER_ERROR("must be rank 1", op, "?;[]");  // sequence_length
59   INFER_OK(op, "[?,?,?];[?]", "[?,2];[?];[2];[d0_1|d1_0,1]");
60   INFER_OK(op, "[?,1,?];[1]", "[?,2];[?];[2];[d0_1|d1_0,1]");
61   INFER_OK(op, "[?,?,?];[1]", "[?,2];[?];[2];[d1_0,1]");
62   INFER_OK(op, "[?,1,?];[?]", "[?,2];[?];[2];[d0_1,1]");
63   INFER_ERROR("must be equal", op, "[?,1,?];[2]");
67   ShapeInferenceTestOp op("CTCBeamSearchDecoder");
68   auto set_top_paths = [&op](int top_paths) {
73                      .Finalize(&op.node_def));
80   INFER_ERROR("must be rank 3", op, "[];?");  // inputs
81   INFER_ERROR("must be rank 1", op, "?;[]");  // sequence_length
85   INFER_OK(op, "[?,?,?];[?]", "[?,2];[?];[2];[d0_1|d1_0,1]");
86   INFER_OK(op, "[?,1,?];[1]", "[?,2];[?];[2];[d0_1|d1_0,1]");
87   INFER_OK(op, "[?,?,?];[1]", "[?,2];[?];[2];[d1_0,1]");
88   INFER_OK(op, "[?,1,?];[?]", "[?,2];[?];[2];[d0_1,1]");
89   INFER_ERROR("must be equal", op, "[?,1,?];[2]");
94   INFER_OK(op, "?;?", "[?,2];[?,2];[?];[?];[2];[2];[?,2]");