Searched refs:LongToUlong (Results 1 – 14 of 14) sorted by relevance
145 parallel_strategy_map.set_current_stage(UlongToUint(LongToUlong(++current_stage_))); in Save()153 parallel_strategys->set_stage(UlongToUint(LongToUlong(node_stra.second->GetInputStage()))); in Save()158 parallel_strategy->add_dim(UlongToUint(LongToUlong(stra_dim))); in Save()172 dev_matrix->add_dim(UlongToUint(LongToUlong(dev_dim))); in Save()
110 Args args = {dev_mat_.GetDimByReverseIdx(LongToUlong(out_dim)), UlongToLong(index), out_dim}; in InferSplitByAxis()138 int64_t dev_dim = in_tensor_map_.GetDimByIdx(LongToUlong(cat_dim)); in InferPermuteByAxis()139 …Args args_alltoall = {dev_mat_.GetDimByReverseIdx(LongToUlong(dev_dim)), UlongToLong(index), cat_d… in InferPermuteByAxis()
196 re_map_expand_list.push_back((*expand_list_ptr)[LongToUlong(tensor_map_.GetIndexByValue(i))]); in ComputeArrangementByExpandedShape()330 int64_t divisor = device_arrangement_.GetDimByReverseIdx(LongToUlong(dim)); in slice_shape()
45 if ((LongToUlong(device_num) & LongToUlong(device_num - 1)) != 0) { in InitDevice()
43 auto unsigned_length = LongToUlong(length); in Log2Ceil()
105 tensor_map_index_output[i] = tensor_map_index_input[LongToUlong(axis_v_[i])]; in InferTensorMap()
108 if ((LongToUlong(strategy_value) & LongToUlong(strategy_value - 1)) != 0) { in CheckStrategyValue()
116 if (output_num_ > IntToSize(input_shape_[LongToUlong(axis_)])) { in CheckParam()
117 inline uint64_t LongToUlong(int64_t u) { in LongToUlong() function
155 start_timestamp_ = LongToUlong(CURRENT_TIME_MILLI.count()); in SetIterationRunning()168 complete_timestamp_ = LongToUlong(CURRENT_TIME_MILLI.count()); in SetIterationCompleted()
37 iter_next_req_timestamp_ = LongToUlong(CURRENT_TIME_MILLI.count()) + iteration_time_window_; in InitKernel()
847 if (!partition->at(LongToUlong(server_id)).first) { in RoundRobinPartitioner()848 partition->at(LongToUlong(server_id)).first = true; in RoundRobinPartitioner()851 KVMessage &server_kv_pairs = partition->at(LongToUlong(server_id)).second; in RoundRobinPartitioner()
845 uint64_t axis_value_pos = LongToUlong(GetPositiveAxis(axis_value, LongToSize(rank))); in InferImplSplit()
1556 *index = LongToUlong(GetValue<int64_t>(GetValueNode(index_node))); in TraceTupleGetItem()