Searched defs:CastKernel (Results 1 – 2 of 2) sorted by relevance
27 __global__ void CastKernel(const int input_size, const S *input_addr, T *output_addr) { in CastKernel() function
356 bool DataTypeTransfer::CastKernel(const TypeIdArgs &args, void *dst, int64_t data_size, DataTypeTra… in CastKernel() function in mindspore::trans::DataTypeTransfer