Searched defs:unspecified_dims (Results 1 – 6 of 6) sorted by relevance
209 unspecified_dims=None): argument330 unspecified_dims=None): argument374 unspecified_dims=None): argument451 unspecified_dims=None): argument480 unspecified_dims=None): argument563 unspecified_dims=None): argument
31 std::string EncodeAttributes(absl::Span<const int64_t> unspecified_dims) { in EncodeAttributes()40 std::vector<int64_t>* unspecified_dims) { in ParseAttributes()
925 const HloSharding& to_merge, absl::Span<const int64_t> unspecified_dims, in RefineManualAutoShardingFromAuto()997 const HloSharding& to_merge, absl::Span<const int64_t> unspecified_dims, in RefineManualAutoShardingFromManual()1028 absl::Span<const int64_t> unspecified_dims, in InferUnspecifiedDimsFromOperand()1102 absl::Span<const int64_t> unspecified_dims, in InferUnspecifiedDimsFromOneUser()1152 absl::Span<const int64_t> unspecified_dims, in InferUnspecifiedDimsFromUsers()1249 unspecified_dims) { in ProcessShardingInstruction()2312 unspecified_dims; in Run() local
38 std::vector<int32_t> unspecified_dims; in XlaSpmdFullToShardShapeOp() local80 std::vector<int32_t> unspecified_dims; in XlaSpmdShardToFullShapeOp() local
30 std::vector<int32_t> unspecified_dims; in ShardingOp() local
5063 absl::Span<const int64_t> unspecified_dims) { in ConvertSpmdFullToShardShape()5108 absl::Span<const int64_t> unspecified_dims) { in ConvertSpmdShardToFullShape()