Searched refs:num_row_partition_tensors (Results 1 – 5 of 5) sorted by relevance
61 int num_row_partition_tensors; in ValidateRowPartitionTypesAndShapes() local63 c->GetAttr("num_row_partition_tensors", &num_row_partition_tensors)); in ValidateRowPartitionTypesAndShapes()64 if (num_row_partition_tensors != row_partition_types.size()) { in ValidateRowPartitionTypesAndShapes()66 "Number of row partition tensors (", num_row_partition_tensors, in ValidateRowPartitionTypesAndShapes()71 for (int i = 0; i < num_row_partition_tensors; ++i) { in ValidateRowPartitionTypesAndShapes()
35081 number_attr: "num_row_partition_tensors"35112 name: "num_row_partition_tensors"
60 int num_row_partition_tensors = row_partition_tensors.size(); in BuildRaggedTensorToTensorGraph() local65 .Attr("num_row_partition_tensors", num_row_partition_tensors) in BuildRaggedTensorToTensorGraph()70 .Input(FakeInput(num_row_partition_tensors, in BuildRaggedTensorToTensorGraph()512 int num_row_partition_tensors) { in SetAttributes() argument517 num_row_partition_tensors); in SetAttributes()
18 number_attr: "num_row_partition_tensors"49 name: "num_row_partition_tensors"