Searched defs:leading_dim (Results 1 – 4 of 4) sorted by relevance
205 Status GetLeadingDimForTensorList(xla::XlaOp list, int64* leading_dim, in GetLeadingDimForTensorList()226 const xla::Shape& element_tensor_list_shape, int64 leading_dim, in GetTensorListShapeFromElementTensorListShape()248 int64 leading_dim, in GetTensorListShapeFromElementShape()299 int64 leading_dim; in GetInitializedTensorListForElement() local
92 int64 leading_dim; in Compile() local
388 int64 leading_dim = 0; in Compute() local
441 Value leading_dim = GetNumElements(op, operands, &rewriter); in matchAndRewrite() local