| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | smooth_l1_loss_impl.cu | 21 __global__ void SmoothL1LossKernel(const int input_size, const float beta, const T *prediction, con… in SmoothL1LossKernel()
34 void SmoothL1Loss(const int &input_size, const float &beta, const T *prediction, const T *target, T… in SmoothL1Loss()
40 __global__ void SmoothL1LossGradKernel(const int input_size, const float beta, const T *prediction,… in SmoothL1LossGradKernel()
55 void SmoothL1LossGrad(const int &input_size, const float &beta, const T *prediction, const T *targe… in SmoothL1LossGrad()
|
| D | l2_loss.cu | 22 __global__ void L2LossKernel(const size_t input_size, const T *input , T *output) { in L2LossKernel()
37 void L2Loss(const size_t input_size, const T *input , T *output, cudaStream_t stream) { in L2Loss()
|
| D | reverse_v2_impl.cu | 21 const int64_t* axis, size_t input_size, size_t axis_size) { in ReverseV2()
37 size_t input_size, size_t axis_size, cudaStream_t cuda_stream) { in CalReverseV2()
|
| D | tile_impl.cu | 20 __global__ void Tile(const size_t output_size, const size_t input_size, const size_t shape_size, in Tile()
50 void CalTile(const size_t output_size, const size_t input_size, const size_t shape_size, const size… in CalTile()
|
| D | loss_with_reduction_impl.cu | 30 __global__ void Copy(T *loss, T *tmp_loss, int reduction, int input_size) { in Copy()
137 __global__ void KLDivLossKernel(const int input_size, const int reduction, const T *input_x, const … in KLDivLossKernel()
156 void KLDivLoss(const int &input_size, const int &reduction, const T *input_x, const T *input_y, T *… in KLDivLoss()
176 __global__ void KLDivLossGradKernel(const int input_size, const int reduction, const T *input_x, co… in KLDivLossGradKernel()
200 void KLDivLossGrad(const int &input_size, const int &reduction, const T *input_x, const T *input_y,… in KLDivLossGrad()
207 __global__ void BinaryCrossEntropyLossKernel(const int input_size, const int reduction, const T *in… in BinaryCrossEntropyLossKernel()
237 void BinaryCrossEntropyLoss(const int &input_size, const int &reduction, const T *input_x, const T … in BinaryCrossEntropyLoss()
257 __global__ void BinaryCrossEntropyLossGradKernel(const int input_size, const int reduction, const T… in BinaryCrossEntropyLossGradKernel()
297 void BinaryCrossEntropyLossGrad(const int &input_size, const int &reduction, const T *input_x, cons… in BinaryCrossEntropyLossGrad()
379 int input_size = n * c; in NLLLossGrad() local
|
| D | rcwm_small_impl.cu | 43 __global__ void L2Rcwm(int seedc, int input_size, const K *input, K *output_mask, S *output_index, … in L2Rcwm()
126 void RCWMScaleK(int seedc, int input_size, K *input, int k, S *output_index, K *output_mask, cudaSt… in RCWMScaleK()
146 void CalRandomChoiceWithMaskSmall(int input_size, int seedc, int count, K *input, S *output_index, … in CalRandomChoiceWithMaskSmall()
|
| D | scatter_nd.cu | 22 …ScatterNdKernel(S *indices, T *update, T *output, const size_t block_size, const size_t input_size, in ScatterNdKernel()
50 void ScatterNd(S *indices, T *update, T *output, const size_t &block_size, const size_t &input_size, in ScatterNd()
|
| D | tensor_scatter_min.cu | 23 … const size_t input_size, const size_t output_size, const size_t indices_dim_0, in TensorScatterMinKernel()
50 …Min(T *input, S *indices, T *update, T *output, const size_t &block_size, const size_t &input_size, in TensorScatterMin()
|
| D | tensor_scatter_add.cu | 23 … const size_t input_size, const size_t output_size, const size_t indices_dim_0, in TensorScatterAddKernel()
50 …Add(T *input, S *indices, T *update, T *output, const size_t &block_size, const size_t &input_size, in TensorScatterAdd()
|
| D | tensor_scatter_max.cu | 23 … const size_t input_size, const size_t output_size, const size_t indices_dim_0, in TensorScatterMaxKernel()
50 …Max(T *input, S *indices, T *update, T *output, const size_t &block_size, const size_t &input_size, in TensorScatterMax()
|
| D | tensor_scatter_sub.cu | 23 … const size_t input_size, const size_t output_size, const size_t indices_dim_0, in TensorScatterSubKernel()
50 …Sub(T *input, S *indices, T *update, T *output, const size_t &block_size, const size_t &input_size, in TensorScatterSub()
|
| D | tensor_scatter_update.cu | 23 … const size_t input_size, const size_t output_size, const size_t indices_dim_0, in TensorScatterUpdateKernel()
50 …ate(T *input, S *indices, T *update, T *output, const size_t &block_size, const size_t &input_size, in TensorScatterUpdate()
|
| D | resize_nearest_neighbor_grad_impl.cu | 32 __global__ void ResizeNearestNeighborGrad(const int input_size, const T *input, const int s1, const… in ResizeNearestNeighborGrad()
69 void CalResizeNearestNeighborGrad(const int input_size, const T *input, const int s1, const int s2,… in CalResizeNearestNeighborGrad()
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_lstm_op.py | 32 …def __init__(self, seq_len, batch_size, input_size, hidden_size, num_layers, has_bias, bidirection… argument
164 …def __init__(self, seq_len, batch_size, input_size, hidden_size, num_layers, has_bias, bidirection… argument
321 …def __init__(self, seq_len, batch_size, input_size, hidden_size, num_layers, has_bias, bidirection… argument
595 …def __init__(self, seq_len, batch_size, input_size, hidden_size, num_layers, has_bias, bidirection… argument
869 …def __init__(self, seq_len, batch_size, input_size, hidden_size, num_layers, has_bias, bidirection… argument
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_lstm_op.py | 37 input_size, argument
102 …def __init__(self, batch_size, input_size, hidden_size, num_layers, has_bias, bidirectional, dropo… argument
205 …def __init__(self, batch_size, input_size, hidden_size, num_layers, has_bias, bidirectional, dropo… argument
293 …def __init__(self, seq_len, batch_size, input_size, hidden_size, num_layers, has_bias, bidirection… argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/ |
| D | binary_cross_entropy.c | 20 static void BinaryCrossEntropyLossKernel(const int input_size, const int reduction, const float *in… in BinaryCrossEntropyLossKernel()
38 void BinaryCrossEntropy(const int input_size, const int reduction, const float *input_x, const floa… in BinaryCrossEntropy()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/ |
| D | activation_grad_fp32_tests.cc | 42 size_t input_size; in TEST_F() local
96 size_t input_size; in TEST_F() local
147 size_t input_size; in TEST_F() local
198 size_t input_size; in TEST_F() local
250 size_t input_size; in TEST_F() local
300 size_t input_size; in TEST_F() local
353 size_t input_size; in TEST_F() local
|
| D | pooling_grad_fp32_tests.cc | 74 size_t input_size; in TEST_F() local
129 size_t input_size; in TEST_F() local
202 size_t input_size; in TEST_F() local
275 size_t input_size; in TEST_F() local
339 size_t input_size; in TEST_F() local
404 size_t input_size; in TEST_F() local
466 size_t input_size; in TEST_F() local
542 size_t input_size; in TEST_F() local
619 size_t input_size; in TEST_F() local
|
| D | softmax_grad_fp32_tests.cc | 71 size_t input_size; in TEST_F() local
132 size_t input_size; in TEST_F() local
197 size_t input_size; in TEST_F() local
262 size_t input_size; in TEST_F() local
327 size_t input_size; in TEST_F() local
|
| /third_party/mbedtls/programs/psa/ |
| D | crypto_examples.c | 63 size_t input_size, in cipher_operation()
102 size_t input_size, in cipher_encrypt()
133 size_t input_size, in cipher_decrypt()
212 input_size = 100, in cipher_example_encrypt_decrypt_aes_cbc_pkcs7_multi() enumerator
260 input_size = 100, in cipher_example_encrypt_decrypt_aes_ctr_multi() enumerator
|
| /third_party/jerryscript/jerry-core/ecma/builtin-objects/ |
| D | ecma-builtin-global.c | 200 … lit_utf8_size_t input_size, /**< routine's first argument's in ecma_builtin_global_object_decode_uri_helper()
421 … lit_utf8_size_t input_size, /**< routine's first argument's in ecma_builtin_global_object_encode_uri_helper()
588 lit_utf8_size_t input_size) /**< routine's first argument's in ecma_builtin_global_object_escape()
694 lit_utf8_size_t input_size) /**< routine's first argument's in ecma_builtin_global_object_unescape()
847 lit_utf8_size_t input_size = ecma_string_get_size (str_p); in ecma_builtin_global_dispatch_routine() local
|
| /third_party/mindspore/tests/st/ops/ascend/ |
| D | test_gru_op.py | 40 …def __init__(self, input_size, hidden_size, num_layers, has_bias, batch_first, bidirectional, drop… argument
50 … def __init__(self, num_layers, has_bias, input_size, num_directions, hidden_size, bidirectional): argument
|
| D | test_rnn_op.py | 40 …def __init__(self, input_size, hidden_size, num_layers, has_bias, batch_first, bidirectional, drop… argument
50 … def __init__(self, num_layers, has_bias, input_size, num_directions, hidden_size, bidirectional): argument
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp16_grad/ |
| D | activation_grad_fp16_test.cc | 35 size_t input_size; in TEST_F() local
92 size_t input_size; in TEST_F() local
|
| /third_party/boost/libs/compute/include/boost/compute/algorithm/ |
| D | count_if.hpp | 43 size_t input_size = detail::iterator_range_size(first, last); in count_if() local
|