Searched refs:outer_dimension_partitions (Results 1 – 4 of 4) sorted by relevance
1368 const std::vector<int64>& outer_dimension_partitions() const { in outer_dimension_partitions() function1372 const std::vector<int64>& outer_dimension_partitions);
3209 const std::vector<int64>& outer_dimension_partitions) { in set_outer_dimension_partitions() argument3210 outer_dimension_partitions_ = outer_dimension_partitions; in set_outer_dimension_partitions()
117 if (!computation->root_instruction()->outer_dimension_partitions().empty()) { in EmitComputation()119 computation->root_instruction()->outer_dimension_partitions().size(); in EmitComputation()2208 if (!computation->root_instruction()->outer_dimension_partitions().empty()) { in HandleCall()2220 call_args, root->shape(), root->outer_dimension_partitions(), &b_, in HandleCall()
1041 CHECK(dot.parent()->root_instruction()->outer_dimension_partitions().empty()); in EmitDotOperation()