Searched defs:source_dim (Results 1 – 4 of 4) sorted by relevance
186 int64_t source_dim() const { return source_dim_; } in source_dim() function195 int64_t source_dim, in ScalarIndexedArray()226 int64_t source_dim, in ScalarIndexedConstantArray()338 Array* source, Array* indices, int64_t source_dim, in ConstructScalarIndexedArray()
193 ScalarIndexedArray* source, Array* indices, int64_t source_dim, in FoldGatherOfGather()295 int64_t source_dim = dim_numbers.start_index_map(0); in ComputeArrayForGather() local
1931 const int64_t source_dim = std::distance(dimensions.begin(), it); in InferShardingFromOperands() local
1495 int64_t source_dim = source_target_dims[0].first; in ReshardWithAllToAll() local