Searched refs:UlongToLong (Results 1 – 5 of 5) sorted by relevance
39 mask_shape.push_back(UlongToLong(ceil((input_shape[1] + 31) / 32))); in GetOutputMaskShape()41 mask_shape.push_back(UlongToLong(ceil((input_shape[1] + 15) / 16))); in GetOutputMaskShape()
110 Args args = {dev_mat_.GetDimByReverseIdx(LongToUlong(out_dim)), UlongToLong(index), out_dim}; in InferSplitByAxis()139 …Args args_alltoall = {dev_mat_.GetDimByReverseIdx(LongToUlong(dev_dim)), UlongToLong(index), cat_d… in InferPermuteByAxis()147 Args args_allsplit = {dev_num, UlongToLong(index), out_dim}; in InferPermuteByAxis()
133 inline int64_t UlongToLong(uint64_t u) { in UlongToLong() function
707 attr_proto->set_i(UlongToLong(value->cast<UInt64ImmPtr>()->value())); in SetScalarToAttributeProto_ir()
314 CreatInt64Imm(UlongToLong(i))}; in InsertMakeTuple()2623 for (int64_t i = 0; i < UlongToLong(node_inputs.size()); ++i) { in NodeParameterName()2690 for (int64_t i = 0; i < UlongToLong(param_split_shapes.size()); ++i) { in CheckpointStrategy()