Searched defs:dev_dim (Results 1 – 4 of 4) sorted by relevance
138 int64_t dev_dim = in_tensor_map_.GetDimByIdx(LongToUlong(cat_dim)); in InferPermuteByAxis() local262 int64_t dev_dim = args[TRANSFER_CONCAT_DEV_DIM_INDEX]; in TransferConcatByAxis() local
96 int64_t dev_dim = args[TRANSFER_PERMUTE_CONCAT_DIM_INDEX]; in StridedSliceOP() local139 Status ConstructOperator::AllGatherOP(int64_t dev_dim) { in AllGatherOP()204 int64_t dev_dim = args[TRANSFER_PERMUTE_DEV_DIM_INDEX]; in AlltoAllOP() local
115 int64_t dev_dim = 2; in TEST_F() local133 int64_t dev_dim = 3; in TEST_F() local
171 for (auto dev_dim : tensor_layout->device_arrangement().array()) { in Save() local