Searched defs:leading_dim (Results 1 – 4 of 4) sorted by relevance
206 Status GetLeadingDimForTensorList(xla::XlaOp list, int64_t* leading_dim, in GetLeadingDimForTensorList()227 const xla::Shape& element_tensor_list_shape, int64_t leading_dim, in GetTensorListShapeFromElementTensorListShape()249 int64_t leading_dim, in GetTensorListShapeFromElementShape()301 int64_t leading_dim; in GetInitializedTensorListForElement() local
94 int64_t leading_dim; in Compile() local
493 int64_t leading_dim = 0; in Compute() local
585 Value leading_dim = GetNumElements(op, adaptor.getOperands(), &rewriter); in matchAndRewrite() local