MS_REG_GPU_KERNEL_TWO (reference) in projects: third_party

Project(s)

Full Search
Definition
Symbol
File Path
History
Type

Searched refs:MS_REG_GPU_KERNEL_TWO (Results 1 – 25 of 92) sorted by relevance

12 3 4

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
D	cast_gpu_kernel.cc	21 MS_REG_GPU_KERNEL_TWO(Cast, KernelAttr().AddInputAttr(kNumberTypeInt8).AddOutputAttr(kNumberTypeInt… 23 MS_REG_GPU_KERNEL_TWO(Cast, KernelAttr().AddInputAttr(kNumberTypeInt8).AddOutputAttr(kNumberTypeInt… 25 MS_REG_GPU_KERNEL_TWO(Cast, KernelAttr().AddInputAttr(kNumberTypeInt8).AddOutputAttr(kNumberTypeInt… 27 MS_REG_GPU_KERNEL_TWO(Cast, KernelAttr().AddInputAttr(kNumberTypeInt8).AddOutputAttr(kNumberTypeInt… 29 MS_REG_GPU_KERNEL_TWO(Cast, KernelAttr().AddInputAttr(kNumberTypeInt8).AddOutputAttr(kNumberTypeUIn… 31 MS_REG_GPU_KERNEL_TWO(Cast, KernelAttr().AddInputAttr(kNumberTypeInt8).AddOutputAttr(kNumberTypeUIn… 33 MS_REG_GPU_KERNEL_TWO(Cast, KernelAttr().AddInputAttr(kNumberTypeInt8).AddOutputAttr(kNumberTypeUIn… 35 MS_REG_GPU_KERNEL_TWO(Cast, KernelAttr().AddInputAttr(kNumberTypeInt8).AddOutputAttr(kNumberTypeUIn… 37 MS_REG_GPU_KERNEL_TWO(Cast, KernelAttr().AddInputAttr(kNumberTypeInt8).AddOutputAttr(kNumberTypeFlo… 39 MS_REG_GPU_KERNEL_TWO(Cast, KernelAttr().AddInputAttr(kNumberTypeInt8).AddOutputAttr(kNumberTypeFlo… [all …]
D	scatter_nd_functor_gpu_kernel.cc	`22 MS_REG_GPU_KERNEL_TWO(ScatterNdUpdate, 29 MS_REG_GPU_KERNEL_TWO(ScatterNdUpdate, 36 MS_REG_GPU_KERNEL_TWO(ScatterNdUpdate, 43 MS_REG_GPU_KERNEL_TWO(ScatterNdUpdate, 50 MS_REG_GPU_KERNEL_TWO(ScatterNdUpdate, 57 MS_REG_GPU_KERNEL_TWO(ScatterNdUpdate, 64 MS_REG_GPU_KERNEL_TWO(ScatterNdUpdate, 71 MS_REG_GPU_KERNEL_TWO(ScatterNdUpdate, 78 MS_REG_GPU_KERNEL_TWO(ScatterNdUpdate, 85 MS_REG_GPU_KERNEL_TWO(ScatterNdUpdate, [all …]`
D	gatherv2_gpu_kernel.cc	`21 MS_REG_GPU_KERNEL_TWO( 25 MS_REG_GPU_KERNEL_TWO( 29 MS_REG_GPU_KERNEL_TWO( 33 MS_REG_GPU_KERNEL_TWO( 37 MS_REG_GPU_KERNEL_TWO( 41 MS_REG_GPU_KERNEL_TWO( 45 MS_REG_GPU_KERNEL_TWO( 48 MS_REG_GPU_KERNEL_TWO( 51 MS_REG_GPU_KERNEL_TWO( 54 MS_REG_GPU_KERNEL_TWO( [all …]`
D	gather_gpu_kernel.cc	`21 MS_REG_GPU_KERNEL_TWO( 25 MS_REG_GPU_KERNEL_TWO( 29 MS_REG_GPU_KERNEL_TWO( 33 MS_REG_GPU_KERNEL_TWO( 37 MS_REG_GPU_KERNEL_TWO( 41 MS_REG_GPU_KERNEL_TWO( 45 MS_REG_GPU_KERNEL_TWO( 48 MS_REG_GPU_KERNEL_TWO( 51 MS_REG_GPU_KERNEL_TWO( 54 MS_REG_GPU_KERNEL_TWO( [all …]`
D	scatter_functor_gpu_kernel.cc	`22 MS_REG_GPU_KERNEL_TWO(ScatterUpdate, 29 MS_REG_GPU_KERNEL_TWO(ScatterUpdate, 36 MS_REG_GPU_KERNEL_TWO(ScatterUpdate, 43 MS_REG_GPU_KERNEL_TWO(ScatterUpdate, 50 MS_REG_GPU_KERNEL_TWO(ScatterUpdate, 57 MS_REG_GPU_KERNEL_TWO(ScatterUpdate, 64 MS_REG_GPU_KERNEL_TWO(ScatterUpdate, 71 MS_REG_GPU_KERNEL_TWO(ScatterUpdate, 78 MS_REG_GPU_KERNEL_TWO(ScatterUpdate, 85 MS_REG_GPU_KERNEL_TWO(ScatterUpdate, [all …]`
D	gather_grad_gpu_kernel.cc	`21 MS_REG_GPU_KERNEL_TWO( 25 MS_REG_GPU_KERNEL_TWO( 29 MS_REG_GPU_KERNEL_TWO( 33 MS_REG_GPU_KERNEL_TWO( 37 MS_REG_GPU_KERNEL_TWO( 41 MS_REG_GPU_KERNEL_TWO( 45 MS_REG_GPU_KERNEL_TWO( 49 MS_REG_GPU_KERNEL_TWO( 53 MS_REG_GPU_KERNEL_TWO( 56 MS_REG_GPU_KERNEL_TWO( [all …]`
D	unsorted_segment_sum_gpu_kernel.cc	`21 MS_REG_GPU_KERNEL_TWO( 26 MS_REG_GPU_KERNEL_TWO( 31 MS_REG_GPU_KERNEL_TWO( 36 MS_REG_GPU_KERNEL_TWO( 41 MS_REG_GPU_KERNEL_TWO( 46 MS_REG_GPU_KERNEL_TWO( 51 MS_REG_GPU_KERNEL_TWO( 56 MS_REG_GPU_KERNEL_TWO( 62 MS_REG_GPU_KERNEL_TWO(UnsortedSegmentSum, 69 MS_REG_GPU_KERNEL_TWO(UnsortedSegmentSum, [all …]`
D	embedding_lookup_gpu_kernel.cc	`21 MS_REG_GPU_KERNEL_TWO( 25 MS_REG_GPU_KERNEL_TWO( 29 MS_REG_GPU_KERNEL_TWO( 33 MS_REG_GPU_KERNEL_TWO( 37 MS_REG_GPU_KERNEL_TWO( 41 MS_REG_GPU_KERNEL_TWO( 45 MS_REG_GPU_KERNEL_TWO( 49 MS_REG_GPU_KERNEL_TWO( 53 MS_REG_GPU_KERNEL_TWO( 57 MS_REG_GPU_KERNEL_TWO( [all …]`
D	gathernd_gpu_kernel.cc	`21 MS_REG_GPU_KERNEL_TWO( 25 MS_REG_GPU_KERNEL_TWO( 29 MS_REG_GPU_KERNEL_TWO( 33 MS_REG_GPU_KERNEL_TWO( 36 MS_REG_GPU_KERNEL_TWO( 39 MS_REG_GPU_KERNEL_TWO( 43 MS_REG_GPU_KERNEL_TWO( 46 MS_REG_GPU_KERNEL_TWO( 49 MS_REG_GPU_KERNEL_TWO( 52 MS_REG_GPU_KERNEL_TWO( [all …]`
D	unsorted_segment_max_gpu_kernel.cc	`21 MS_REG_GPU_KERNEL_TWO( 26 MS_REG_GPU_KERNEL_TWO( 31 MS_REG_GPU_KERNEL_TWO( 36 MS_REG_GPU_KERNEL_TWO( 41 MS_REG_GPU_KERNEL_TWO( 46 MS_REG_GPU_KERNEL_TWO( 52 MS_REG_GPU_KERNEL_TWO(UnsortedSegmentMax, 60 MS_REG_GPU_KERNEL_TWO(UnsortedSegmentMax, 68 MS_REG_GPU_KERNEL_TWO(UnsortedSegmentMax, 76 MS_REG_GPU_KERNEL_TWO(UnsortedSegmentMax, [all …]`
D	reverse_sequence_gpu_kernel.cc	`22 MS_REG_GPU_KERNEL_TWO( 26 MS_REG_GPU_KERNEL_TWO( 31 MS_REG_GPU_KERNEL_TWO( 35 MS_REG_GPU_KERNEL_TWO( 40 MS_REG_GPU_KERNEL_TWO( 44 MS_REG_GPU_KERNEL_TWO( 49 MS_REG_GPU_KERNEL_TWO( 53 MS_REG_GPU_KERNEL_TWO( 58 MS_REG_GPU_KERNEL_TWO( 62 MS_REG_GPU_KERNEL_TWO( [all …]`
D	scatter_nd_gpu_kernel.cc	`21 MS_REG_GPU_KERNEL_TWO( 25 MS_REG_GPU_KERNEL_TWO( 29 MS_REG_GPU_KERNEL_TWO( 33 MS_REG_GPU_KERNEL_TWO( 37 MS_REG_GPU_KERNEL_TWO( 41 MS_REG_GPU_KERNEL_TWO( 45 MS_REG_GPU_KERNEL_TWO( 48 MS_REG_GPU_KERNEL_TWO( 51 MS_REG_GPU_KERNEL_TWO( 54 MS_REG_GPU_KERNEL_TWO( [all …]`
D	tensor_scatter_max_gpu_kernel.cc	`22 MS_REG_GPU_KERNEL_TWO(TensorScatterMax, 30 MS_REG_GPU_KERNEL_TWO(TensorScatterMax, 38 MS_REG_GPU_KERNEL_TWO(TensorScatterMax, 46 MS_REG_GPU_KERNEL_TWO(TensorScatterMax, 54 MS_REG_GPU_KERNEL_TWO(TensorScatterMax, 63 MS_REG_GPU_KERNEL_TWO(TensorScatterMax, 71 MS_REG_GPU_KERNEL_TWO(TensorScatterMax, 79 MS_REG_GPU_KERNEL_TWO(TensorScatterMax, 87 MS_REG_GPU_KERNEL_TWO(TensorScatterMax, 95 MS_REG_GPU_KERNEL_TWO(TensorScatterMax,`
D	tensor_scatter_sub_gpu_kernel.cc	`22 MS_REG_GPU_KERNEL_TWO(TensorScatterSub, 30 MS_REG_GPU_KERNEL_TWO(TensorScatterSub, 38 MS_REG_GPU_KERNEL_TWO(TensorScatterSub, 46 MS_REG_GPU_KERNEL_TWO(TensorScatterSub, 54 MS_REG_GPU_KERNEL_TWO(TensorScatterSub, 63 MS_REG_GPU_KERNEL_TWO(TensorScatterSub, 71 MS_REG_GPU_KERNEL_TWO(TensorScatterSub, 79 MS_REG_GPU_KERNEL_TWO(TensorScatterSub, 87 MS_REG_GPU_KERNEL_TWO(TensorScatterSub, 95 MS_REG_GPU_KERNEL_TWO(TensorScatterSub,`
D	tensor_scatter_min_gpu_kernel.cc	`22 MS_REG_GPU_KERNEL_TWO(TensorScatterMin, 30 MS_REG_GPU_KERNEL_TWO(TensorScatterMin, 38 MS_REG_GPU_KERNEL_TWO(TensorScatterMin, 46 MS_REG_GPU_KERNEL_TWO(TensorScatterMin, 54 MS_REG_GPU_KERNEL_TWO(TensorScatterMin, 63 MS_REG_GPU_KERNEL_TWO(TensorScatterMin, 71 MS_REG_GPU_KERNEL_TWO(TensorScatterMin, 79 MS_REG_GPU_KERNEL_TWO(TensorScatterMin, 87 MS_REG_GPU_KERNEL_TWO(TensorScatterMin, 95 MS_REG_GPU_KERNEL_TWO(TensorScatterMin,`
D	tensor_scatter_update_gpu_kernel.cc	`21 MS_REG_GPU_KERNEL_TWO(TensorScatterUpdate, 28 MS_REG_GPU_KERNEL_TWO(TensorScatterUpdate, 35 MS_REG_GPU_KERNEL_TWO(TensorScatterUpdate, 42 MS_REG_GPU_KERNEL_TWO(TensorScatterUpdate, 49 MS_REG_GPU_KERNEL_TWO(TensorScatterUpdate, 56 MS_REG_GPU_KERNEL_TWO(TensorScatterUpdate, 63 MS_REG_GPU_KERNEL_TWO(TensorScatterUpdate, 70 MS_REG_GPU_KERNEL_TWO(TensorScatterUpdate, 77 MS_REG_GPU_KERNEL_TWO(TensorScatterUpdate,`
D	dynamic_shape_gpu_kernel.cc	22 MS_REG_GPU_KERNEL_TWO(DynamicShape, KernelAttr().AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumb… 25 MS_REG_GPU_KERNEL_TWO(DynamicShape, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNu… 28 MS_REG_GPU_KERNEL_TWO(DynamicShape, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNu… 31 MS_REG_GPU_KERNEL_TWO(DynamicShape, KernelAttr().AddInputAttr(kNumberTypeBool).AddOutputAttr(kNumbe… 34 MS_REG_GPU_KERNEL_TWO(DynamicShape, KernelAttr().AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumb… 37 MS_REG_GPU_KERNEL_TWO(DynamicShape, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNu… 40 MS_REG_GPU_KERNEL_TWO(DynamicShape, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNu… 43 MS_REG_GPU_KERNEL_TWO(DynamicShape, KernelAttr().AddInputAttr(kNumberTypeBool).AddOutputAttr(kNumbe…
D	tensor_scatter_add_gpu_kernel.cc	`21 MS_REG_GPU_KERNEL_TWO(TensorScatterAdd, 28 MS_REG_GPU_KERNEL_TWO(TensorScatterAdd, 35 MS_REG_GPU_KERNEL_TWO(TensorScatterAdd, 42 MS_REG_GPU_KERNEL_TWO(TensorScatterAdd, 49 MS_REG_GPU_KERNEL_TWO(TensorScatterAdd, 56 MS_REG_GPU_KERNEL_TWO(TensorScatterAdd, 63 MS_REG_GPU_KERNEL_TWO(TensorScatterAdd,`
D	argmaxandminwithvalue_gpu_kernel.cc	`21 MS_REG_GPU_KERNEL_TWO( 25 MS_REG_GPU_KERNEL_TWO( 29 MS_REG_GPU_KERNEL_TWO( 34 MS_REG_GPU_KERNEL_TWO( 38 MS_REG_GPU_KERNEL_TWO( 42 MS_REG_GPU_KERNEL_TWO(`
D	unique_gpu_kernel.cc	`21 MS_REG_GPU_KERNEL_TWO( 25 MS_REG_GPU_KERNEL_TWO( 29 MS_REG_GPU_KERNEL_TWO( 32 MS_REG_GPU_KERNEL_TWO(`
D	one_hot_gpu_kernel.cc	`22 MS_REG_GPU_KERNEL_TWO(OneHot, 29 MS_REG_GPU_KERNEL_TWO(OneHot, 36 MS_REG_GPU_KERNEL_TWO(OneHot, 43 MS_REG_GPU_KERNEL_TWO(OneHot,`
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/
D	matmul_gpu_kernel.cc	`21 MS_REG_GPU_KERNEL_TWO( 25 MS_REG_GPU_KERNEL_TWO( 29 MS_REG_GPU_KERNEL_TWO( 33 MS_REG_GPU_KERNEL_TWO( 37 MS_REG_GPU_KERNEL_TWO( 41 MS_REG_GPU_KERNEL_TWO( 47 MS_REG_GPU_KERNEL_TWO(FusedMatMulBiasAdd, 54 MS_REG_GPU_KERNEL_TWO(FusedMatMulBiasAdd, 61 MS_REG_GPU_KERNEL_TWO(FusedMatMulBiasAdd,`
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/other/
D	check_valid_gpu_kernel.cc	`21 MS_REG_GPU_KERNEL_TWO( 25 MS_REG_GPU_KERNEL_TWO( 29 MS_REG_GPU_KERNEL_TWO( 32 MS_REG_GPU_KERNEL_TWO(`
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
D	nll_loss_gpu_kernel.cc	`21 MS_REG_GPU_KERNEL_TWO(NLLLoss, 29 MS_REG_GPU_KERNEL_TWO(NLLLoss, 37 MS_REG_GPU_KERNEL_TWO(NLLLoss, 45 MS_REG_GPU_KERNEL_TWO(NLLLoss,`
D	nll_loss_grad_gpu_kernel.cc	`21 MS_REG_GPU_KERNEL_TWO(NLLLossGrad, 30 MS_REG_GPU_KERNEL_TWO(NLLLossGrad, 39 MS_REG_GPU_KERNEL_TWO(NLLLossGrad, 48 MS_REG_GPU_KERNEL_TWO(NLLLossGrad,`

12 3 4