Home
last modified time | relevance | path

Searched defs:element_num (Results 1 – 23 of 23) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dmomentum_impl.cu74 __global__ void FusedMomentumWeightDecayScaleKernel(const size_t element_num, T *weight_decay, T *s… in FusedMomentumWeightDecayScaleKernel()
85 void FusedWeightDecayScaleMomentum(const size_t element_num, T *weight_decay, T *scale, T *variable… in FusedWeightDecayScaleMomentum()
95 __global__ void FusedMomentumScaleKernel(const size_t element_num, T *scale, T *variable, T *accumu… in FusedMomentumScaleKernel()
104 void FusedScaleMomentum(const size_t element_num, T *scale, T *variable, T *accumulation, const T *… in FusedScaleMomentum()
113 __global__ void FusedWeightDecayMomentumKernel(const size_t element_num, T *weight_decay, T *variab… in FusedWeightDecayMomentumKernel()
123 void FusedWeightDecayMomentum(const size_t element_num, T *weight_decay, T *variable, T *accumulati… in FusedWeightDecayMomentum()
133 __global__ void CombineFusedMomentumScaleKernel(const size_t num, const size_t *element_num, T **sc… in CombineFusedMomentumScaleKernel()
156 …obal__ void CombineFusedMomentumWeightDecayScaleKernel(const size_t num, const size_t *element_num, in CombineFusedMomentumWeightDecayScaleKernel()
169 …CombineFusedWeightDecayScaleMomentum(const size_t max, const size_t num, const size_t *element_num, in CombineFusedWeightDecayScaleMomentum()
Dfill_impl.cu22 size_t element_num = m * n; in FillKernel() local
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Ddiv_int8_coder.cc54 int element_num = output_tensor_->ElementsNum(); in DoCode() local
Dsub_int8_coder.cc78 auto element_num = output_tensor_->ElementsNum(); in DoCode() local
/third_party/mindspore/mindspore/lite/test/ut/src/api/
Dmodel_c_test.cc177 int64_t element_num = OH_AI_TensorGetElementNum(tensor); in TEST_F() local
190 int64_t element_num = OH_AI_TensorGetElementNum(tensor); in TEST_F() local
/third_party/mindspore/mindspore/lite/test/ohos_test/src/
Dmodel_test.cc208 int64_t element_num = OH_AI_TensorGetElementNum(tensor); variable
220 int64_t element_num = OH_AI_TensorGetElementNum(tensor); variable
/third_party/jerryscript/jerry-core/ecma/builtin-objects/typedarray/
Decma-builtin-typedarray-prototype.c225 ecma_number_t element_num = typedarray_getter_cb (info.buffer_p + byte_pos); in ecma_builtin_typedarray_prototype_exec_routine() local
474 ecma_number_t element_num = src_typedarray_getter_cb (src_info.buffer_p + src_byte_pos); in ecma_builtin_typedarray_prototype_map() local
1526 ecma_number_t element_num = typedarray_getter_cb (info.buffer_p + byte_index); in ecma_builtin_typedarray_prototype_sort() local
1555 ecma_number_t element_num = ecma_get_number_from_value (element_value); in ecma_builtin_typedarray_prototype_sort() local
1622 ecma_number_t element_num = typedarray_getter_cb (info.buffer_p + byte_index); in ecma_builtin_typedarray_prototype_find_helper() local
1759 ecma_number_t element_num = getter_cb (info.buffer_p + position); in ecma_builtin_typedarray_prototype_index_of() local
1840 ecma_number_t element_num = getter_cb (info.buffer_p + position); in ecma_builtin_typedarray_prototype_last_index_of() local
2017 ecma_number_t element_num = ecma_get_typedarray_element (typedarray_buffer_p + index, class_id); in ecma_builtin_typedarray_prototype_to_locale_string_helper() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Darithmetic_compare_fp32.cc66 auto element_num = out_tensors_[0]->ElementsNum(); in DoArithmetic() local
Darithmetic_fp32.cc376 auto element_num = out_tensors_[0]->ElementsNum(); in DoArithmetic() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dsigmoid_parameter.h38 int element_num; member
/third_party/mindspore/mindspore/lite/examples/quick_start_c/
Dmain.c105 int64_t element_num = OH_AI_TensorGetElementNum(tensor); in QuickStart() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Ddiv_int8.cc78 auto element_num = out_tensors_[0]->ElementsNum(); in DoExecute() local
Dsub_int8.cc99 auto element_num = out_tensors_[0]->ElementsNum(); in DoExecute() local
Darithmetic_int8.cc115 auto element_num = out_tensors_[0]->ElementsNum(); in DoArithmetic() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Drelu_v2_pass.cc82 auto element_num = in CreateReluV2() local
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dconv_biasadd_fusion.cc147 auto element_num = std::accumulate(shape.begin(), shape.end(), 1, std::multiplies<int>()); in CheckCanFusion() local
/third_party/mindspore/mindspore/lite/src/
Dtensor.cc204 …auto element_num = (format_ == mindspore::NC4HW4 || format_ == mindspore::NHWC4) ? ElementsC4Num()… in Size() local
/third_party/mindspore/mindspore/lite/tools/converter/quantizer/
Dmixed_bit_weight_quantizer.cc101 int element_num = 1; in BinarySearchForQuantizationScale() local
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/
Darithmetic_fp32_coder.cc425 int element_num = output_tensor_->ElementsNum(); in DoCode() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dcpu_kernel.cc84 …lUtils::GetElementNumEveryDim(const std::vector<size_t> &shape, std::vector<size_t> *element_num) { in GetElementNumEveryDim()
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dtranspose_strategy.cc277 int element_num = shape.front(); in ChangeOpSlice() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dquantize.h165 int element_num; member
/third_party/openh264/codec/common/inc/
Dloongson_intrinsics.h1838 #define VECT_PRINT(RTYPE, element_num, in0, enter) \ argument