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.
 
 
 
 
 
 
djc b077aa1cab [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
..
argmax_gpu_kernel.cc fix cpu/gpu argmax op 5 years ago
argmax_gpu_kernel.h initial commit: fix 11 dts tickets 4 years ago
argmaxandminwithvalue_gpu_kernel.cc add argminwithvalue oparetor arithmetric 4 years ago
argmaxandminwithvalue_gpu_kernel.h fix bug in gpu/array 4 years ago
array_reduce_gpu_kernel.cc ReduceProd gpu kernel initial commit 4 years ago
array_reduce_gpu_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
broadcast_to_gpu_kernel.cc fix 4 years ago
broadcast_to_gpu_kernel.h initial commit: fix 11 dts tickets 4 years ago
cast_gpu_kernel.cc cast gpu kernel 5 years ago
cast_gpu_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
concatv2_gpu_kernel.cc add float64 support to concat GPU 5 years ago
concatv2_gpu_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
crop_and_resize_gpu_kernel.cc firstCommit gpu CropAndResizeSupport 4 years ago
crop_and_resize_gpu_kernel.h firstCommit gpu CropAndResizeSupport 4 years ago
depthtospace_gpu_kernel.cc Add GPU ops:depthtospace and spacetodepth 4 years ago
depthtospace_gpu_kernel.h fix the exception when occur error and replace magic number with const value. 4 years ago
dynamic_range_gpu_kernel.cc range past segfault addressed comments 5 years ago
dynamic_range_gpu_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
dynamic_shape_gpu_kernel.cc int64 support and typo fix 5 years ago
dynamic_shape_gpu_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
embedding_lookup_gpu_kernel.cc add EmbeddingLookup kernel 4 years ago
embedding_lookup_gpu_kernel.h add ExtractImagePatches GPU 4 years ago
extract_image_patches_gpu_kernel.cc add ExtractImagePatches GPU 4 years ago
extract_image_patches_gpu_kernel.h fix the exception when occur error and replace magic number with const value. 4 years ago
gather_gpu_kernel.cc add dtypes for sponge 4 years ago
gather_gpu_kernel.h fix bug in gpu/array 4 years ago
gather_grad_gpu_kernel.cc add dtypes for sponge 4 years ago
gather_grad_gpu_kernel.h fix bug in gpu/array 4 years ago
gathernd_gpu_kernel.cc add dtypes for sponge 4 years ago
gathernd_gpu_kernel.h codecheck 4 years ago
gatherv2_gpu_kernel.cc add dtypes for sponge 4 years ago
gatherv2_gpu_kernel.h fix bug in gpu/array 4 years ago
in_top_k_gpu_kernel.cc InTopK gpu kernel initial commit 4 years ago
in_top_k_gpu_kernel.h initial commit: fix topk index 4 years ago
meshgrid_gpu_kernel.cc code check & pclint 4 years ago
meshgrid_gpu_kernel.h fix the exception when occur error and replace magic number with const value. 4 years ago
one_hot_gpu_kernel.cc initial commit 5 years ago
one_hot_gpu_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
oneslike_gpu_kernel.cc meshgrid gpu kernel initial commit 4 years ago
oneslike_gpu_kernel.h initial commit: fix 11 dts tickets 4 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 bug in gpu/array 4 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 bug in gpu/array 4 years ago
reverse_sequence_gpu_kernel.cc first commit - ReverseSequence - GPU 4 years ago
reverse_sequence_gpu_kernel.h first commit - ReverseSequence - GPU 4 years ago
reverse_v2_gpu_kernel.cc ReverseV2 op initial commit 4 years ago
reverse_v2_gpu_kernel.h ReverseV2 op initial commit 4 years ago
scatter_functor_gpu_kernel.cc Add scattersub op for gpu 4 years ago
scatter_functor_gpu_kernel.h Add scattersub op for gpu 4 years ago
scatter_nd_functor_gpu_kernel.cc add scatter_nd_add and scatter_nd_sub op for gpu 4 years ago
scatter_nd_functor_gpu_kernel.h add scatter_nd_add and scatter_nd_sub op for gpu 4 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
select_gpu_kernel.cc add bool support to gpu select 5 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 [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
slice_grad_gpu_kernel.cc Add float64 support to slice ops 5 years ago
slice_grad_gpu_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
sort_gpu_kernel.cc gpu sort op initial commit 4 years ago
sort_gpu_kernel.h fix the exception when occur error and replace magic number with const value. 4 years ago
spacetodepth_gpu_kernel.cc Add GPU ops:depthtospace and spacetodepth 4 years ago
spacetodepth_gpu_kernel.h fix the exception when occur error and replace magic number with const value. 4 years ago
split_gpu_kernel.cc initial commit 5 years ago
split_gpu_kernel.h fix bug in gpu/array 4 years ago
squeeze_gpu_kernel.cc implement the squeeze operator of gpu 4 years ago
squeeze_gpu_kernel.h implement the squeeze operator of gpu 4 years ago
strided_slice_gpu_common.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
strided_slice_gpu_kernel.cc Add float64 support to slice ops 5 years ago
strided_slice_gpu_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
strided_slice_grad_gpu_kernel.cc Add float64 support to slice ops 5 years ago
strided_slice_grad_gpu_kernel.h codecheck 4 years ago
tensor_copy_slices_gpu_kernel.cc Add TensorCopySlice Op 4 years ago
tensor_copy_slices_gpu_kernel.h TensorCopySlices Generalization 4 years ago
tensor_scatter_add_gpu_kernel.cc add gpu op cumprod, tensor_scatter_add 4 years ago
tensor_scatter_add_gpu_kernel.h add gpu op cumprod, tensor_scatter_add 4 years ago
tensor_scatter_max_gpu_kernel.cc initial commit: change of requirements, no more div/mul, need sub 4 years ago
tensor_scatter_max_gpu_kernel.h initial commit: change of requirements, no more div/mul, need sub 4 years ago
tensor_scatter_min_gpu_kernel.cc initial commit: change of requirements, no more div/mul, need sub 4 years ago
tensor_scatter_min_gpu_kernel.h initial commit: change of requirements, no more div/mul, need sub 4 years ago
tensor_scatter_sub_gpu_kernel.cc initial commit: change of requirements, no more div/mul, need sub 4 years ago
tensor_scatter_sub_gpu_kernel.h initial commit: change of requirements, no more div/mul, need sub 4 years ago
tensor_scatter_update_gpu_kernel.cc add dtypes for sponge 4 years ago
tensor_scatter_update_gpu_kernel.h change tensorscatterupdate so that the input is not updated 4 years ago
tile_gpu_kernel.cc add dtypes for sponge 4 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 intial commit: cast to float32 4 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 float16 support to UnsortedSegmentSum 4 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