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.
 
 
 
 
 
 
Peilin Wang 91242dcf53 gpu sort op initial commit 4 years ago
..
argmax_gpu_kernel.cc fix cpu/gpu argmax op 5 years ago
argmax_gpu_kernel.h fix cpu/gpu argmax op 5 years ago
argmaxwithvalue_gpu_kernel.cc add float64 support for ops for end of version 1.1 5 years ago
argmaxwithvalue_gpu_kernel.h Rewrite sequence_mask as a composite op 5 years ago
array_reduce_gpu_kernel.cc add float64 support to ReduceMax and ReduceSum 5 years ago
array_reduce_gpu_kernel.h addtensor size limitation to 2G 5 years ago
broadcast_to_gpu_kernel.cc add float64 support for ops for end of version 1.1 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 add float64 support to concat GPU 5 years ago
concatv2_gpu_kernel.h add float64 support to concat GPU 5 years ago
depthtospace_gpu_kernel.cc Add GPU ops:depthtospace and spacetodepth 4 years ago
depthtospace_gpu_kernel.h fix depthtospace and spacetodepth ops bugs 4 years ago
dynamic_range_gpu_kernel.cc range past segfault addressed comments 5 years ago
dynamic_range_gpu_kernel.h fix GPUKernelMod about the using of shard_ptr 5 years ago
dynamic_shape_gpu_kernel.cc int64 support and typo fix 5 years ago
dynamic_shape_gpu_kernel.h add trace for gpu error/excpt log 5 years ago
gather_gpu_kernel.cc fix some bug in gather and gathergrad op 5 years ago
gather_gpu_kernel.h fix bug in gatherd and gatherdgrad 5 years ago
gather_grad_gpu_kernel.cc add more dtypes support for gatherdgrad and other bugfix 5 years ago
gather_grad_gpu_kernel.h fix bug in gatherd and gatherdgrad 5 years ago
gathernd_gpu_kernel.cc add float64 support to GatherNd 5 years ago
gathernd_gpu_kernel.h add float64 support to GatherNd 5 years ago
gatherv2_gpu_kernel.cc add bool support to gpu gather 4 years ago
gatherv2_gpu_kernel.h add float64 support to gather gpu 5 years ago
one_hot_gpu_kernel.cc initial commit 5 years ago
one_hot_gpu_kernel.h add one_hot and one_like int64_t requirement 5 years ago
oneslike_gpu_kernel.cc add gpu oneslike kernel 5 years ago
oneslike_gpu_kernel.h add one_hot and one_like int64_t requirement 5 years ago
pack_gpu_kernel.cc modify pack to stack 5 years ago
pack_gpu_kernel.h !9492 Fix implement bug in pack and unpack 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 add additional type support to ScatterAdd 5 years ago
scatter_add_gpu_kernel.h Update tensor shape from int to size_t in scatterop 5 years ago
scatter_nd_gpu_kernel.cc add float64 support to gather grad and gatherd grad 5 years ago
scatter_nd_gpu_kernel.h add trace for gpu error/excpt log 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 Update tensor shape from int to size_t in scatterop 5 years ago
select_gpu_kernel.cc add bool support to gpu select 4 years ago
select_gpu_kernel.h Unified code style 5 years ago
slice_gpu_kernel.cc Add float64 support to slice ops 5 years ago
slice_gpu_kernel.h Add float64 support to slice ops 5 years ago
slice_grad_gpu_kernel.cc Add float64 support to slice ops 5 years ago
slice_grad_gpu_kernel.h Add float64 support to slice ops 5 years ago
sort_gpu_kernel.cc gpu sort op initial commit 4 years ago
sort_gpu_kernel.h gpu sort op initial commit 4 years ago
spacetodepth_gpu_kernel.cc Add GPU ops:depthtospace and spacetodepth 4 years ago
spacetodepth_gpu_kernel.h fix depthtospace and spacetodepth ops bugs 4 years ago
split_gpu_kernel.cc initial commit 5 years ago
split_gpu_kernel.h add trace for gpu error/excpt log 5 years ago
strided_slice_gpu_kernel.cc Add float64 support to slice ops 5 years ago
strided_slice_gpu_kernel.h Add float64 support to slice ops 5 years ago
strided_slice_grad_gpu_kernel.cc Add float64 support to slice ops 5 years ago
strided_slice_grad_gpu_kernel.h Add float64 support to slice ops 5 years ago
tensor_scatter_update_gpu_kernel.cc add tensor scatter update support different type inputs 4 years ago
tensor_scatter_update_gpu_kernel.h tensor_scatter_update new op quick initial commit 5 years ago
tile_gpu_kernel.cc add gpu Tile kernel 5 years ago
tile_gpu_kernel.h add gpu Tile kernel 5 years ago
topk_gpu_kernel.cc gpu sort op initial commit 4 years ago
topk_gpu_kernel.h fix GPUKernelMod about the using of shard_ptr 5 years ago
transpose_gpu_kernel.cc add float64 support to gather grad and gatherd grad 5 years ago
transpose_gpu_kernel.h 1.Add xception model && scripts on gpu. 5 years ago
unique_gpu_kernel.cc unique op int64 support 5 years ago
unique_gpu_kernel.h fix GPUKernelMod about the using of shard_ptr 5 years ago
unpack_gpu_kernel.cc modify pack to stack 5 years ago
unpack_gpu_kernel.h !9492 Fix implement bug in pack and unpack 5 years ago
unsorted_segment_max_gpu_kernel.cc UnsortedSegMax_Int64_support 5 years ago
unsorted_segment_max_gpu_kernel.h UnsortedSegMin/Max output_shape validation fix 5 years ago
unsorted_segment_min_gpu_kernel.cc Adding int32 scalar support + validation fix + alarm fix 5 years ago
unsorted_segment_min_gpu_kernel.h UnsortedSegMin/Max output_shape validation fix 5 years ago
unsorted_segment_sum_gpu_kernel.cc add float64 support to gather grad and gatherd grad 5 years ago
unsorted_segment_sum_gpu_kernel.h add trace for gpu error/excpt log 5 years ago
zeroslike_gpu_kernel.cc removed zeroslike from math/unary_op_gpu_kernel and added float64 support 5 years ago
zeroslike_gpu_kernel.h fix GPUKernelMod about the using of shard_ptr 5 years ago