Searched defs:dev_dim (Results 1 – 4 of 4) sorted by relevance
141 int64_t dev_dim = args[TRANSFER_PERMUTE_CONCAT_DIM_INDEX]; in ReplaceStridedSliceOpToSplitOp() local188 int64_t dev_dim = args[TRANSFER_PERMUTE_CONCAT_DIM_INDEX]; in StridedSliceOP() local236 Status ConstructOperator::AllGatherOP(int64_t dev_dim) { in AllGatherOP()307 int64_t dev_dim = args[TRANSFER_PERMUTE_DEV_DIM_INDEX]; in AlltoAllOP() local
181 int64_t dev_dim = in_tensor_map_.GetDimByIdx(LongToUlong(cat_dim)); in InferPermuteByAxis() local305 int64_t dev_dim = args[TRANSFER_CONCAT_DEV_DIM_INDEX]; in TransferConcatByAxis() local
115 int64_t dev_dim = 2; in TEST_F() local133 int64_t dev_dim = 3; in TEST_F() local
154 for (auto dev_dim : tensor_layout->device_arrangement().array()) { in to_protobuf() local