Home
last modified time | relevance | path

Searched refs:output_dim (Results 1 – 21 of 21) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Daffine.cc22 void Affine::Init(const std::vector<int64_t> &contexts, int64_t output_dim, bool transpose_a, bool … in Init() argument
24 this->set_output_dim(output_dim); in Init()
33 void Affine::set_output_dim(int64_t output_dim) { (void)this->AddAttr(kAffineOutputDim, MakeValue(o… in set_output_dim() argument
Daffine.h41 void Init(const std::vector<int64_t> &contexts, int64_t output_dim, bool transpose_a = false,
50 void set_output_dim(int64_t output_dim);
Dsplice.cc37 void Splice::set_output_dim(int64_t output_dim) { (void)this->AddAttr(kSpliceOutputDims, MakeValue(… in set_output_dim() argument
Dsplice.h58 void set_output_dim(int64_t output_dim);
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dconcat_int8_coder.cc77 int output_dim = static_cast<int>(output_tensor_->shape().size()); in Prepare() local
78 concat_param_->output_shapes_ = reinterpret_cast<int *>(malloc(output_dim * sizeof(int))); in Prepare()
80 …MS_CHECK_RET_CODE(memcpy_s(reinterpret_cast<void *>(concat_param_->output_shapes_), output_dim * s… in Prepare()
81 output_tensor_->shape().data(), sizeof(int) * output_dim), in Prepare()
83 for (int i = axis_ + 1; i < output_dim; i++) { in Prepare()
/third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/
Donnx_splice_parser.cc32 int64_t output_dim = 0; in Parse() local
48 output_dim = static_cast<int>(onnx_node_attr.i()); in Parse()
54 primitive->Init(context, forward_indexes, output_dim); in Parse()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dconcat_int8.cc103 size_t output_dim = out_shape.size(); in ReSize() local
104 MS_CHECK_FALSE_MSG(SIZE_MUL_OVERFLOW(output_dim, sizeof(int)), RET_ERROR, "mul overflow"); in ReSize()
105 concat_param_->output_shapes_ = reinterpret_cast<int *>(malloc(output_dim * sizeof(int))); in ReSize()
111 sizeof(int) * output_dim); in ReSize()
113 for (size_t i = static_cast<size_t>(concat_param_->axis_ + 1); i < output_dim; i++) { in ReSize()
Dleaky_relu_int8.cc98 size_t output_dim = output_shape.size(); in ReSize() local
103 quant_prelu_parm_.out_shape_ = reinterpret_cast<int *>(malloc(output_dim * sizeof(int))); in ReSize()
109 sizeof(int) * output_dim); in ReSize()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dsparse_to_dense_fp32.cc58 int output_dim = static_cast<int>(output0->shape().size()); in ReSize() local
59 MS_CHECK_TRUE_MSG(output_dim <= DIMENSION_4D, RET_ERROR, "output_dim should <= 4"); in ReSize()
60 for (int i = 0; i < DIMENSION_4D - output_dim; i++) { in ReSize()
63 for (int i = 0; i < output_dim; i++) { in ReSize()
64 output_shape[i + DIMENSION_4D - output_dim] = output_shape_tmp[i]; in ReSize()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dleaky_relu_int8.c27 int output_dim = quant_prelu_parm->input_dim_; in DoLeakReluInt8() local
31 for (int i = 0; i < output_dim; i++) { in DoLeakReluInt8()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/
Ddynamic_broadcast_gradient_args_kernel.cc55 int output_dim = -1; in GetGradientIndices() local
65 if (!output_dim_set || reverse_shape[i][j] == static_cast<int64_t>(output_dim)) { in GetGradientIndices()
66 output_dim = LongToInt(reverse_shape[i][j]); in GetGradientIndices()
/third_party/mindspore/mindspore/explainer/benchmark/_attribution/
Dmetric.py217 output_dim = explainer.network(inputs).shape[1]
218 if output_dim != self._num_labels:
/third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/python_file_for_ci/
Dwide_and_deep.py81 def __init__(self, input_dim, output_dim, weight_bias_init, act_str, argument
86 weight_init, [input_dim, output_dim], name="weight")
87 self.bias = init_method(bias_init, [output_dim], name="bias")
/third_party/mindspore/mindspore/lite/src/ops/populate/
Daffine_populate.cc83 affine_param->output_dim_ = value->output_dim(); in PopulateAffineParameter()
Dsplice_populate.cc96 param->output_dim_ = value->output_dim(); in PopulateSpliceParameter()
/third_party/mindspore/tests/st/model_zoo_tests/DeepFM/src/
Ddeepfm.py131 …def __init__(self, input_dim, output_dim, weight_bias_init, act_str, scale_coef=1.0, convert_dtype… argument
135 self.weight = init_method(weight_init, [input_dim, output_dim], name="weight")
136 self.bias = init_method(bias_init, [output_dim], name="bias")
/third_party/mindspore/mindspore/lite/schema/
Dops.fbs1112 output_dim: long;
1197 output_dim: long;
Dops_generated.h11526 int64_t output_dim() const {
11550 void add_output_dim(int64_t output_dim) {
11551 fbb_.AddElement<int64_t>(Splice::VT_OUTPUT_DIM, output_dim, 0);
11568 int64_t output_dim = 0) {
11570 builder_.add_output_dim(output_dim);
11580 int64_t output_dim = 0) {
11587 output_dim);
12540 int64_t output_dim() const {
12571 void add_output_dim(int64_t output_dim) {
12572 fbb_.AddElement<int64_t>(Affine::VT_OUTPUT_DIM, output_dim, 0);
[all …]
/third_party/mindspore/mindspore/lite/src/ops/
Dops_def.cc1112 OP_ATTR(output_dim, long)
1197 OP_ATTR(output_dim, long)
/third_party/mindspore/mindspore/lite/schema/inner/
Dops_generated.h17481 int64_t output_dim = 0;
17507 int64_t output_dim() const {
17537 void add_output_dim(int64_t output_dim) {
17538 fbb_.AddElement<int64_t>(Splice::VT_OUTPUT_DIM, output_dim, 0);
17555 int64_t output_dim = 0) {
17557 builder_.add_output_dim(output_dim);
17567 int64_t output_dim = 0) {
17574 output_dim);
18851 int64_t output_dim = 0;
18876 int64_t output_dim() const {
[all …]
/third_party/mindspore/third_party/proto/caffe/
Dcaffe.proto1606 required int32 output_dim = 2; // output channel number field