zzy34407230
/
mindspore2022
Not watched
1
0
Fork 0
Code
Releases 22 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
14423 Commits
18 Branches
584 MB
67 Download
 
 
 
 
 
 
Tree: dfd894c190
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'dfd894c190'
${ noResults }
mindspore2022/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays
mindspore-ci-bot 0241aa8797 !9106 Fix bot warning 5 years ago
..
argmax_gpu_kernel.cc mindspore path adjust 5 years ago
argmax_gpu_kernel.h support int64 shape 5 years ago
argmaxwithvalue_gpu_kernel.cc mindspore path adjust 5 years ago
argmaxwithvalue_gpu_kernel.h support int64 shape 5 years ago
array_reduce_gpu_kernel.cc add gpu cumsum 5 years ago
array_reduce_gpu_kernel.h Optimize log level to DEBUG in array reduce gpu kernel 5 years ago
broadcast_to_gpu_kernel.cc Gpu support BroadcastTo kernel 5 years ago
broadcast_to_gpu_kernel.h change gpu kernel shape to size_t 5 years ago
cast_gpu_kernel.cc cast gpu kernel 5 years ago
cast_gpu_kernel.h Cast/ReLU dynamic shapelsgpu op supports int32 and int64 5 years ago
concatv2_gpu_kernel.cc allow unsigned char types during registration, convert char to unsigned char 5 years ago
concatv2_gpu_kernel.h support int64 shape 5 years ago
dynamic_shape_gpu_kernel.cc int64 support and typo fix 5 years ago
dynamic_shape_gpu_kernel.h int64 support and typo fix 5 years ago
gather_gpu_kernel.cc fix some bug in gather and gathergrad op 5 years ago
gather_gpu_kernel.h fix some bug in gather and gathergrad op 5 years ago
gather_grad_gpu_kernel.cc fix some bug in gather and gathergrad op 5 years ago
gather_grad_gpu_kernel.h fix some bug in gather and gathergrad op 5 years ago
gathernd_gpu_kernel.cc allow unsigned char types during registration, convert char to unsigned char 5 years ago
gathernd_gpu_kernel.h add iou ops. 5 years ago
gatherv2_gpu_kernel.cc fix dynamic shape and add testcase for SparseGatherV2 GPU 5 years ago
gatherv2_gpu_kernel.h GatherUpdate 5 years ago
one_hot_gpu_kernel.cc mindspore path adjust 5 years ago
one_hot_gpu_kernel.h support int64 shape 5 years ago
oneslike_gpu_kernel.cc add gpu oneslike kernel 5 years ago
oneslike_gpu_kernel.h Unified code style 5 years ago
pack_gpu_kernel.cc add supports to op pack on gpu 5 years ago
pack_gpu_kernel.h fix-bot-warning 5 years ago
range_gpu_kernel.cc support categorical log_prob 5 years ago
range_gpu_kernel.h support categorical log_prob 5 years ago
resize_nearest_neighbor_gpu_kernel.cc add GPU support to ResizeNearestNeighbor 5 years ago
resize_nearest_neighbor_gpu_kernel.h fix ResizeNearestNeighbor & add util.cuh to support atomicadd for half 5 years ago
resize_nearest_neighbor_grad_gpu_kernel.cc add GPU support to ResizeNearestNeighbor 5 years ago
resize_nearest_neighbor_grad_gpu_kernel.h fix ResizeNearestNeighbor & add util.cuh to support atomicadd for half 5 years ago
scatter_add_gpu_kernel.cc refine GPU-Scatteradd 5 years ago
scatter_add_gpu_kernel.h Remove use_locking in GPU ScatterAdd 5 years ago
scatter_nd_gpu_kernel.cc added type support for atomic add and scatternd 5 years ago
scatter_nd_gpu_kernel.h support int64 shape 5 years ago
scatter_update_gpu_kernel.cc add int8 and uint8 support to scatteradd and fix doc example 5 years ago
scatter_update_gpu_kernel.h Adapt changes to GPU framework for ScatterAdd/Update and SparseFtrl 5 years ago
select_gpu_kernel.cc mindspore path adjust 5 years ago
select_gpu_kernel.h Unified code style 5 years ago
slice_gpu_kernel.cc gpu reshepe add type 5 years ago
slice_gpu_kernel.h support int64 shape 5 years ago
slice_grad_gpu_kernel.cc gpu reshepe add type 5 years ago
slice_grad_gpu_kernel.h support int64 shape 5 years ago
split_gpu_kernel.cc add gpu split and restructure gpu concat 5 years ago
split_gpu_kernel.h change Split op to extend PrimitiveWithCheck 5 years ago
strided_slice_gpu_kernel.cc allow unsigned char types during registration, convert char to unsigned char 5 years ago
strided_slice_gpu_kernel.h strided slice 5 years ago
strided_slice_grad_gpu_kernel.cc allow unsigned char types during registration, convert char to unsigned char 5 years ago
strided_slice_grad_gpu_kernel.h strided slice 5 years ago
topk_gpu_kernel.cc gpu support TopK kernel 5 years ago
topk_gpu_kernel.h code warning 5 years ago
transpose_gpu_kernel.cc added type support for transpose and maxgrad 5 years ago
transpose_gpu_kernel.h fix and add testcase for dynamic shape scatteradd/update transpose 5 years ago
unique_gpu_kernel.cc gpu unique op kernel and st test 5 years ago
unique_gpu_kernel.h gpu unique op kernel and st test 5 years ago
unpack_gpu_kernel.cc add supports to op unpack on gpu 5 years ago
unpack_gpu_kernel.h fix-bot-warning 5 years ago
unsorted_segment_max_gpu_kernel.cc First Commit SegMax changes + typo in SegSum InferImpl 5 years ago
unsorted_segment_max_gpu_kernel.h First Commit SegMax changes + typo in SegSum InferImpl 5 years ago
unsorted_segment_sum_gpu_kernel.cc DynamicShape_Support_SegmentSum 5 years ago
unsorted_segment_sum_gpu_kernel.h DynamicShape_Support_SegmentSum 5 years ago
Delete task

Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.

cancel
confirm

阿对对队

C++ Python Text C Unity3D Asset other