Searched refs:operands_sharded_at_contracting_dims (Results 1 – 2 of 2) sorted by relevance
101 bool operands_sharded_at_contracting_dims; member600 const bool operands_sharded_at_contracting_dims = in PartitionBaseCase() local601 einsum_config.operands_sharded_at_contracting_dims; in PartitionBaseCase()611 !operands_sharded_at_contracting_dims) { in PartitionBaseCase()625 if (operands_sharded_at_contracting_dims) { in PartitionBaseCase()634 operands_sharded_at_contracting_dims) in PartitionBaseCase()640 !operands_sharded_at_contracting_dims) { in PartitionBaseCase()677 if (operands_sharded_at_contracting_dims) { in PartitionBaseCase()704 if (operands_sharded_at_contracting_dims) { in PartitionBaseCase()729 if (operands_sharded_at_contracting_dims) { in PartitionBaseCase()[all …]
509 bool operands_sharded_at_contracting_dims; member