Searched refs:GetLeadingDimForTensorList (Results 1 – 3 of 3) sorted by relevance
70 Status GetLeadingDimForTensorList(xla::XlaOp list, int64* leading_dim,
205 Status GetLeadingDimForTensorList(xla::XlaOp list, int64* leading_dim, in GetLeadingDimForTensorList() function302 TF_RETURN_IF_ERROR(GetLeadingDimForTensorList( in GetInitializedTensorListForElement()
95 OP_REQUIRES_OK(ctx, GetLeadingDimForTensorList(ctx->Input(0), &leading_dim, in Compile()