Searched refs:GetRowPartitionTypes (Results 1 – 3 of 3) sorted by relevance
35 Status GetRowPartitionTypes( in GetRowPartitionTypes() function
252 TF_RETURN_IF_ERROR(GetRowPartitionTypes(c, &row_partition_types)); in RaggedTensorToTensorShapeFn()
63 OP_REQUIRES_OK(context, GetRowPartitionTypes<OpKernelConstruction>( in RaggedTensorToTensorBaseOp()