Searched defs:output_dim (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | affine.cc | 22 void Affine::Init(const std::vector<int64_t> &contexts, int64_t output_dim, bool transpose_a, bool … in Init()
33 void Affine::set_output_dim(int64_t output_dim) { (void)this->AddAttr(kAffineOutputDim, MakeValue(o… in set_output_dim()
|
| D | splice.cc | 37 void Splice::set_output_dim(int64_t output_dim) { (void)this->AddAttr(kSpliceOutputDims, MakeValue(… in set_output_dim()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | leaky_relu_int8.c | 27 int output_dim = quant_prelu_parm->input_dim_; in DoLeakReluInt8() local
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_splice_parser.cc | 32 int64_t output_dim = 0; in Parse() local
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/ |
| D | concat_int8_coder.cc | 77 int output_dim = static_cast<int>(output_tensor_->shape().size()); in Prepare() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | leaky_relu_int8.cc | 98 size_t output_dim = output_shape.size(); in ReSize() local
|
| D | concat_int8.cc | 103 size_t output_dim = out_shape.size(); in ReSize() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | sparse_to_dense_fp32.cc | 58 int output_dim = static_cast<int>(output0->shape().size()); in ReSize() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/ |
| D | dynamic_broadcast_gradient_args_kernel.cc | 55 int output_dim = -1; in GetGradientIndices() local
|
| /third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/python_file_for_ci/ |
| D | wide_and_deep.py | 81 def __init__(self, input_dim, output_dim, weight_bias_init, act_str, argument
|
| /third_party/mindspore/tests/st/model_zoo_tests/DeepFM/src/ |
| D | deepfm.py | 131 …def __init__(self, input_dim, output_dim, weight_bias_init, act_str, scale_coef=1.0, convert_dtype… argument
|
| /third_party/mindspore/third_party/proto/caffe/ |
| D | caffe.proto | 1606 required int32 output_dim = 2; // output channel number field
|