Searched defs:kInputWithBatch (Results 1 – 6 of 6) sorted by relevance
53 const int kInputWithBatch = 3; in SparseMatrixMatMulCheckShape() local138 const int kInputWithBatch = 3; in SparseMatrixMatMulInferShape() local
46 const int64_t kInputWithBatch = 3; in OrgqrInferShape() local
99 const int kInputWithBatch = 3; in FrontendSparseMatrixSparseMatMulInferShape() local189 const int kInputWithBatch = 3; in SparseMatrixSparseMatMulInferShape() local
48 const int kInputWithBatch = 3; in SparseMatrixNNZInferShape() local
57 const int64_t kInputWithBatch = 3; in SparseMatrixTransposeInferShape() local
58 const int kInputWithBatch = 3; in Resize() local