Searched refs:partition_outer_dimension (Results 1 – 2 of 2) sorted by relevance
513 st2 = st1.partition_outer_dimension(517 st3 = st2.partition_outer_dimension(523 st4 = st1.partition_outer_dimension(543 struct_2 = struct.partition_outer_dimension(545 struct_3 = struct_2.partition_outer_dimension(554 st.partition_outer_dimension(partition)558 st.partition_outer_dimension(10)967 struct_2 = struct.partition_outer_dimension(969 struct_3 = struct_2.partition_outer_dimension(1293 result = struct.partition_outer_dimension(
951 def partition_outer_dimension(self, row_partition): member in StructuredTensor