kNumberTypeFloat16 (reference) in projects: third_party

Project(s)

Full Search
Definition
Symbol
File Path
History
Type

Searched refs:kNumberTypeFloat16 (Results 1 – 25 of 526) sorted by relevance

12 3 4 5 6 7 8 9 10 >>...22

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/
D	sync_batch_norm_gpu_kernel.cc	`36 .AddInputAttr(kNumberTypeFloat16) 41 .AddOutputAttr(kNumberTypeFloat16) 44 .AddOutputAttr(kNumberTypeFloat16) 45 .AddOutputAttr(kNumberTypeFloat16), 50 .AddInputAttr(kNumberTypeFloat16) 51 .AddInputAttr(kNumberTypeFloat16) 55 .AddOutputAttr(kNumberTypeFloat16) 56 .AddOutputAttr(kNumberTypeFloat16) 62 .AddInputAttr(kNumberTypeFloat16) 63 .AddInputAttr(kNumberTypeFloat16) [all …]`
D	sync_batch_norm_grad_gpu_kernel.cc	`34 .AddInputAttr(kNumberTypeFloat16) 35 .AddInputAttr(kNumberTypeFloat16) 39 .AddOutputAttr(kNumberTypeFloat16) 47 .AddInputAttr(kNumberTypeFloat16) 51 .AddOutputAttr(kNumberTypeFloat16) 52 .AddOutputAttr(kNumberTypeFloat16), 56 .AddInputAttr(kNumberTypeFloat16) 57 .AddInputAttr(kNumberTypeFloat16) 58 .AddInputAttr(kNumberTypeFloat16) 61 .AddOutputAttr(kNumberTypeFloat16) [all …]`
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/
D	unary_op_gpu_kernel.cc	23 …REG_GPU_KERNEL_ONE(Exp, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumberTypeFlo… 27 …G_GPU_KERNEL_ONE(Expm1, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumberTypeFlo… 31 …REG_GPU_KERNEL_ONE(Log, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumberTypeFlo… 35 …G_GPU_KERNEL_ONE(Log1p, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumberTypeFlo… 39 …REG_GPU_KERNEL_ONE(Erf, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumberTypeFlo… 43 …EG_GPU_KERNEL_ONE(Erfc, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumberTypeFlo… 49 …REG_GPU_KERNEL_ONE(Neg, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumberTypeFlo… 59 …_KERNEL_ONE(Reciprocal, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumberTypeFlo… 63 …_GPU_KERNEL_ONE(Square, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumberTypeFlo… 69 …EG_GPU_KERNEL_ONE(Sqrt, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumberTypeFlo… [all …]
D	unary_op_grad_gpu_kernel.cc	31 …KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumb… 39 …KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumb… 47 …KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumb… 55 …KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumb… 63 …KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumb… 71 …KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumb… 79 …KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumb… 87 …KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumb…
D	broadcast_gpu_kernel.cc	164 …KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumb… 167 …Less, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAttr… 170 …Equal, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAtt… 174 …KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumb… 178 …KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumb… 181 …Pow, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAttr(… 185 …KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumb… 188 …Mul, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAttr(… 191 …Sub, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAttr(… 194 …Add, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAttr(… [all …]
D	matmul_gpu_kernel.cc	`31 …KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumb… 43 …KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumb… 63 .AddInputAttr(kNumberTypeFloat16) 64 .AddInputAttr(kNumberTypeFloat16) 65 .AddInputAttr(kNumberTypeFloat16) 66 .AddOutputAttr(kNumberTypeFloat16),`
D	broadcast_grad_gpu_kernel.cc	`47 .AddInputAttr(kNumberTypeFloat16) 48 .AddInputAttr(kNumberTypeFloat16) 49 .AddInputAttr(kNumberTypeFloat16) 50 .AddOutputAttr(kNumberTypeFloat16) 51 .AddOutputAttr(kNumberTypeFloat16), 55 .AddInputAttr(kNumberTypeFloat16) 56 .AddInputAttr(kNumberTypeFloat16) 57 .AddInputAttr(kNumberTypeFloat16) 58 .AddOutputAttr(kNumberTypeFloat16) 59 .AddOutputAttr(kNumberTypeFloat16),`
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
D	adagrad_gpu_kernel.cc	`32 .AddInputAttr(kNumberTypeFloat16) 33 .AddInputAttr(kNumberTypeFloat16) 34 .AddInputAttr(kNumberTypeFloat16) 35 .AddInputAttr(kNumberTypeFloat16) 36 .AddOutputAttr(kNumberTypeFloat16) 37 .AddOutputAttr(kNumberTypeFloat16), 41 .AddInputAttr(kNumberTypeFloat16) 42 .AddInputAttr(kNumberTypeFloat16) 44 .AddInputAttr(kNumberTypeFloat16) 45 .AddOutputAttr(kNumberTypeFloat16) [all …]`
D	adam_gpu_kernel.cc	`39 .AddInputAttr(kNumberTypeFloat16) 40 .AddInputAttr(kNumberTypeFloat16) 41 .AddInputAttr(kNumberTypeFloat16) 42 .AddInputAttr(kNumberTypeFloat16) 43 .AddInputAttr(kNumberTypeFloat16) 44 .AddInputAttr(kNumberTypeFloat16) 45 .AddInputAttr(kNumberTypeFloat16) 46 .AddInputAttr(kNumberTypeFloat16) 47 .AddInputAttr(kNumberTypeFloat16) 48 .AddInputAttr(kNumberTypeFloat16) [all …]`
D	sparse_ftrl_gpu_kernel.cc	`45 .AddInputAttr(kNumberTypeFloat16) 46 .AddInputAttr(kNumberTypeFloat16) 47 .AddInputAttr(kNumberTypeFloat16) 48 .AddInputAttr(kNumberTypeFloat16) 50 .AddOutputAttr(kNumberTypeFloat16) 51 .AddOutputAttr(kNumberTypeFloat16) 52 .AddOutputAttr(kNumberTypeFloat16), 56 .AddInputAttr(kNumberTypeFloat16) 57 .AddInputAttr(kNumberTypeFloat16) 58 .AddInputAttr(kNumberTypeFloat16) [all …]`
D	lstm_grad_data_gpu_kernel.cc	`38 .AddInputAttr(kNumberTypeFloat16) 39 .AddInputAttr(kNumberTypeFloat16) 40 .AddInputAttr(kNumberTypeFloat16) 41 .AddInputAttr(kNumberTypeFloat16) 42 .AddInputAttr(kNumberTypeFloat16) 43 .AddInputAttr(kNumberTypeFloat16) 44 .AddInputAttr(kNumberTypeFloat16) 45 .AddInputAttr(kNumberTypeFloat16) 46 .AddInputAttr(kNumberTypeFloat16) 47 .AddOutputAttr(kNumberTypeFloat16) [all …]`
D	pooling_grad_gpu_kernel.cc	`30 .AddInputAttr(kNumberTypeFloat16) 31 .AddInputAttr(kNumberTypeFloat16) 32 .AddInputAttr(kNumberTypeFloat16) 33 .AddOutputAttr(kNumberTypeFloat16), 44 .AddInputAttr(kNumberTypeFloat16) 45 .AddInputAttr(kNumberTypeFloat16) 46 .AddInputAttr(kNumberTypeFloat16) 47 .AddOutputAttr(kNumberTypeFloat16), 58 .AddInputAttr(kNumberTypeFloat16) 59 .AddInputAttr(kNumberTypeFloat16) [all …]`
D	layer_norm_grad_grad_gpu_kernel.cc	`37 .AddInputAttr(kNumberTypeFloat16) 38 .AddInputAttr(kNumberTypeFloat16) 39 .AddInputAttr(kNumberTypeFloat16) 40 .AddInputAttr(kNumberTypeFloat16) 41 .AddInputAttr(kNumberTypeFloat16) 42 .AddInputAttr(kNumberTypeFloat16) 43 .AddInputAttr(kNumberTypeFloat16) 44 .AddInputAttr(kNumberTypeFloat16) 45 .AddOutputAttr(kNumberTypeFloat16) 46 .AddOutputAttr(kNumberTypeFloat16) [all …]`
D	activation_grad_kernel.cc	`27 …KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumb… 36 …KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumb… 45 …KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumb… 54 …KernelAttr().AddInputAttr(kNumberTypeFloat16).AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumb…`
D	momentum_gpu_kernel.cc	`32 .AddInputAttr(kNumberTypeFloat16) 33 .AddInputAttr(kNumberTypeFloat16) 34 .AddInputAttr(kNumberTypeFloat16) 35 .AddInputAttr(kNumberTypeFloat16) 36 .AddInputAttr(kNumberTypeFloat16) 37 .AddOutputAttr(kNumberTypeFloat16), 41 .AddInputAttr(kNumberTypeFloat16) 42 .AddInputAttr(kNumberTypeFloat16) 44 .AddInputAttr(kNumberTypeFloat16) 46 .AddOutputAttr(kNumberTypeFloat16), [all …]`
D	nll_loss_grad_gpu_kernel.cc	`35 .AddInputAttr(kNumberTypeFloat16) 36 .AddInputAttr(kNumberTypeFloat16) 41 .AddInputAttr(kNumberTypeFloat16) 42 .AddInputAttr(kNumberTypeFloat16) 46 .AddOutputAttr(kNumberTypeFloat16), 50 .AddInputAttr(kNumberTypeFloat16) 51 .AddInputAttr(kNumberTypeFloat16) 53 .AddInputAttr(kNumberTypeFloat16) 54 .AddInputAttr(kNumberTypeFloat16) 55 .AddOutputAttr(kNumberTypeFloat16),`
D	ftrl_gpu_kernel.cc	`35 .AddInputAttr(kNumberTypeFloat16) 36 .AddInputAttr(kNumberTypeFloat16) 37 .AddInputAttr(kNumberTypeFloat16) 38 .AddInputAttr(kNumberTypeFloat16) 39 .AddInputAttr(kNumberTypeFloat16) 40 .AddInputAttr(kNumberTypeFloat16) 41 .AddInputAttr(kNumberTypeFloat16) 42 .AddInputAttr(kNumberTypeFloat16) 43 .AddOutputAttr(kNumberTypeFloat16),`
D	lstm_gpu_kernel.cc	`35 .AddInputAttr(kNumberTypeFloat16) 36 .AddInputAttr(kNumberTypeFloat16) 37 .AddInputAttr(kNumberTypeFloat16) 38 .AddInputAttr(kNumberTypeFloat16) 39 .AddOutputAttr(kNumberTypeFloat16) 40 .AddOutputAttr(kNumberTypeFloat16) 41 .AddOutputAttr(kNumberTypeFloat16) 42 .AddOutputAttr(kNumberTypeFloat16) 43 .AddOutputAttr(kNumberTypeFloat16),`
D	batch_norm_grad_gpu_kernel.cc	`35 .AddInputAttr(kNumberTypeFloat16) // dy 36 .AddInputAttr(kNumberTypeFloat16) // x 41 .AddOutputAttr(kNumberTypeFloat16) // dx 62 .AddInputAttr(kNumberTypeFloat16) // dy 63 .AddInputAttr(kNumberTypeFloat16) // x 69 .AddInputAttr(kNumberTypeFloat16) // y 70 .AddOutputAttr(kNumberTypeFloat16) // dx 92 .AddInputAttr(kNumberTypeFloat16) // dy 93 .AddInputAttr(kNumberTypeFloat16) // x 99 .AddInputAttr(kNumberTypeFloat16) // y [all …]`
D	binary_cross_entropy_grad_kernel.cc	`31 .AddInputAttr(kNumberTypeFloat16) 32 .AddInputAttr(kNumberTypeFloat16) 33 .AddInputAttr(kNumberTypeFloat16) 34 .AddInputAttr(kNumberTypeFloat16) 35 .AddOutputAttr(kNumberTypeFloat16), 46 .AddInputAttr(kNumberTypeFloat16) 47 .AddInputAttr(kNumberTypeFloat16) 48 .AddInputAttr(kNumberTypeFloat16) 49 .AddOutputAttr(kNumberTypeFloat16),`
D	layer_norm_grad_gpu_kernel.cc	`34 .AddInputAttr(kNumberTypeFloat16) 35 .AddInputAttr(kNumberTypeFloat16) 36 .AddInputAttr(kNumberTypeFloat16) 37 .AddInputAttr(kNumberTypeFloat16) 38 .AddInputAttr(kNumberTypeFloat16) 39 .AddOutputAttr(kNumberTypeFloat16) 40 .AddOutputAttr(kNumberTypeFloat16) 41 .AddOutputAttr(kNumberTypeFloat16),`
D	sparse_apply_proximal_adagrad_kernel.cc	`35 .AddInputAttr(kNumberTypeFloat16) 36 .AddInputAttr(kNumberTypeFloat16) 37 .AddInputAttr(kNumberTypeFloat16) 38 .AddInputAttr(kNumberTypeFloat16) 39 .AddInputAttr(kNumberTypeFloat16) 40 .AddInputAttr(kNumberTypeFloat16) 42 .AddOutputAttr(kNumberTypeFloat16) 43 .AddOutputAttr(kNumberTypeFloat16),`
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fusion/
D	layer_norm_beta_gamma_backprop_fusion_test.cc	49 …builder.SetInputsDeviceType({kNumberTypeFloat16, kNumberTypeFloat16, kNumberTypeFloat16, kNumberTy… in Query() 79 builder1.SetInputsDeviceType({kNumberTypeFloat16}); in TEST_F() 98 …builder2.SetInputsDeviceType({kNumberTypeFloat16, kNumberTypeFloat16, kNumberTypeFloat16, kNumberT… in TEST_F() 99 builder2.SetOutputsDeviceType({kNumberTypeFloat16, kNumberTypeFloat16}); in TEST_F() 140 builder1.SetInputsDeviceType({kNumberTypeFloat16}); in TEST_F() 157 …builder2.SetInputsDeviceType({kNumberTypeFloat16, kNumberTypeFloat16, kNumberTypeFloat16, kNumberT… in TEST_F() 158 builder2.SetOutputsDeviceType({kNumberTypeFloat16, kNumberTypeFloat16}); in TEST_F() 199 builder1.SetInputsDeviceType({kNumberTypeFloat16}); in TEST_F() 216 …builder2.SetInputsDeviceType({kNumberTypeFloat16, kNumberTypeFloat16, kNumberTypeFloat16, kNumberT… in TEST_F() 217 builder2.SetOutputsDeviceType({kNumberTypeFloat16, kNumberTypeFloat16}); in TEST_F() [all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
D	scatter_functor_gpu_kernel.cc	`38 .AddInputAttr(kNumberTypeFloat16) 40 .AddInputAttr(kNumberTypeFloat16) 41 .AddOutputAttr(kNumberTypeFloat16), 45 .AddInputAttr(kNumberTypeFloat16) 47 .AddInputAttr(kNumberTypeFloat16) 48 .AddOutputAttr(kNumberTypeFloat16), 110 .AddInputAttr(kNumberTypeFloat16) 112 .AddInputAttr(kNumberTypeFloat16) 113 .AddOutputAttr(kNumberTypeFloat16), 117 .AddInputAttr(kNumberTypeFloat16) [all …]`
D	array_reduce_gpu_kernel.cc	25 …U_KERNEL_ONE(ReduceMax, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumberTypeFlo… 29 …_KERNEL_ONE(ReduceMean, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumberTypeFlo… 35 …U_KERNEL_ONE(ReduceSum, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumberTypeFlo… 39 …U_KERNEL_ONE(ReduceMin, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumberTypeFlo… 47 …_KERNEL_ONE(ReduceProd, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumberTypeFlo…

12 3 4 5 6 7 8 9 10 >>...22