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.
 
 
 
 
 
 
z00512249 a1b8c2f065 while inplace op for cpu backend 4 years ago
..
argmax_gpu_kernel.cc fix cpu/gpu argmax op 5 years ago
argmax_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
argmaxandminwithvalue_gpu_kernel.cc add argminwithvalue oparetor arithmetric 4 years ago
argmaxandminwithvalue_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
array_reduce_gpu_kernel.cc Add support float64 as input type for ReduceProd GPU op. 4 years ago
array_reduce_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
batchtospace_gpu_kernel.cc modified commenting 4 years ago
batchtospace_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
broadcast_to_gpu_kernel.cc fix 4 years ago
broadcast_to_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
cast_gpu_kernel.cc add gpu complex ops 4 years ago
cast_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
concatv2_gpu_kernel.cc Add concat op dtype. 4 years ago
concatv2_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
crop_and_resize_gpu_kernel.cc firstCommit gpu CropAndResizeSupport 4 years ago
crop_and_resize_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
depthtospace_gpu_kernel.cc Add GPU ops:depthtospace and spacetodepth 5 years ago
depthtospace_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
dynamic_range_gpu_kernel.cc range past segfault addressed comments 5 years ago
dynamic_range_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
dynamic_shape_gpu_kernel.cc int64 support and typo fix 5 years ago
dynamic_shape_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
embedding_lookup_gpu_kernel.cc add EmbeddingLookup kernel 4 years ago
embedding_lookup_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
extract_image_patches_gpu_kernel.cc add ExtractImagePatches GPU 4 years ago
extract_image_patches_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
gather_gpu_kernel.cc add dtypes for sponge 4 years ago
gather_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
gather_grad_gpu_kernel.cc add dtypes for sponge 4 years ago
gather_grad_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
gathernd_gpu_kernel.cc initial commit: fixed python class 4 years ago
gathernd_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
gatherv2_gpu_kernel.cc add dtypes for sponge 4 years ago
gatherv2_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
in_top_k_gpu_kernel.cc InTopK gpu kernel initial commit 4 years ago
in_top_k_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
meshgrid_gpu_kernel.cc code check & pclint 4 years ago
meshgrid_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
one_hot_gpu_kernel.cc initial commit 5 years ago
one_hot_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
oneslike_gpu_kernel.cc meshgrid gpu kernel initial commit 4 years ago
oneslike_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
pack_gpu_kernel.cc modify pack to stack 5 years ago
pack_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
range_gpu_kernel.cc support categorical log_prob 5 years ago
range_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
resize_nearest_neighbor_gpu_kernel.cc add GPU support to ResizeNearestNeighbor 5 years ago
resize_nearest_neighbor_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 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 optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
reverse_sequence_gpu_kernel.cc first commit - ReverseSequence - GPU 4 years ago
reverse_sequence_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
reverse_v2_gpu_kernel.cc ReverseV2 op initial commit 4 years ago
reverse_v2_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
scatter_functor_gpu_kernel.cc Add scattersub op for gpu 4 years ago
scatter_functor_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 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 while inplace op for cpu backend 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 optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
select_gpu_kernel.cc add bool support to gpu select 5 years ago
select_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
slice_gpu_kernel.cc initial commit: variadic template slice kernel 4 years ago
slice_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
slice_grad_gpu_kernel.cc Add float64 support to slice ops 5 years ago
slice_grad_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
sort_gpu_kernel.cc gpu sort op initial commit 4 years ago
sort_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
spacetobatch_gpu_kernel.cc modified commenting 4 years ago
spacetobatch_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
spacetodepth_gpu_kernel.cc Add GPU ops:depthtospace and spacetodepth 5 years ago
spacetodepth_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
split_gpu_kernel.cc initial commit 5 years ago
split_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
squeeze_gpu_kernel.cc implement the squeeze operator of gpu 4 years ago
squeeze_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
strided_slice_gpu_common.h Clean up the first batch of static alarms of operator group 4 years ago
strided_slice_gpu_kernel.cc Add float64 support to slice ops 5 years ago
strided_slice_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
strided_slice_grad_gpu_kernel.cc Add float64 support to slice ops 5 years ago
strided_slice_grad_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
tensor_copy_slices_gpu_kernel.cc Add TensorCopySlice Op 4 years ago
tensor_copy_slices_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 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 optimizes the kernel error description of Split, Meshgrid, Select, etc. 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 optimizes the kernel error description of Split, Meshgrid, Select, etc. 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 optimizes the kernel error description of Split, Meshgrid, Select, etc. 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 optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
tensor_scatter_update_gpu_kernel.cc Support int64 as input type for GPU OP TensorScatterUpdate. 4 years ago
tensor_scatter_update_gpu_kernel.h optimizes the kernel error description of Split, Meshgrid, Select, etc. 4 years ago
tile_gpu_kernel.cc add dtypes for sponge 4 years ago
tile_gpu_kernel.h add vector size check, input shape check and divide by zero check for gpu operators. 4 years ago
topk_gpu_kernel.cc gpu sort op initial commit 4 years ago
topk_gpu_kernel.h Clean up the first batch of static alarms of operator group 4 years ago
transpose_gpu_kernel.cc add float64 support to gather grad and gatherd grad 5 years ago
transpose_gpu_kernel.h add validation of vector size and non-zero validation of denominator for nn gpu operators. 4 years ago
unique_gpu_kernel.cc unique op int64 support 5 years ago
unique_gpu_kernel.h add vector size check, nullptr check and clean code for gpu operators. 4 years ago
unpack_gpu_kernel.cc modify pack to stack 5 years ago
unpack_gpu_kernel.h Clean up the first batch of static alarms of operator group 4 years ago
unsorted_segment_max_gpu_kernel.cc UnsortedSegMax_Int64_support 5 years ago
unsorted_segment_max_gpu_kernel.h add vector size check, nullptr check and clean code for gpu operators. 4 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 add vector size check, nullptr check and clean code for gpu operators. 4 years ago
unsorted_segment_sum_gpu_kernel.cc add float16 support to UnsortedSegmentSum 4 years ago
unsorted_segment_sum_gpu_kernel.h add vector size check, nullptr check and clean code for gpu operators. 4 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 add validation of vector size and non-zero validation of denominator for nn gpu operators. 4 years ago