| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | dynamic_quant_ext.cc | 31 ShapeVector scale_shape = {abstract::Shape::kShapeRankAny}; in InferShape() local
40 ShapeVector scale_shape(input_shape_vector.begin(), input_shape_vector.end() - 1); in InferShape() local
|
| D | batch_norm.cc | 56 … const ShapeVector &x_shape, const ShapeVector &scale_shape, const ShapeVector &bias_shape, in BatchNormShapeCheck()
111 auto scale_shape = input_args[kInputIndex1]->GetShape()->GetShapeVector(); in InferShape() local
|
| D | batch_norm_grad_grad.cc | 34 auto scale_shape = input_args[kInputIndex2]->GetShape()->Clone(); in InferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | scale_fp32_tests.cc | 66 …id TestScaleFp32::Prepare(const std::vector<int> &input_shape, const std::vector<int> &scale_shape, in Prepare()
105 std::vector<int> scale_shape{3}; in TEST_F() local
127 std::vector<int> scale_shape{3}; in TEST_F() local
148 std::vector<int> scale_shape{3}; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/ |
| D | scale_int8.cc | 65 …:Prepare(const std::vector<int> &in_shape, int8_t *input_data, const std::vector<int> &scale_shape, in Prepare()
113 std::vector<int> scale_shape = {2, 2, 1}; in TEST_F() local
139 std::vector<int> scale_shape = {2, 2, 1}; in TEST_F() local
165 std::vector<int> scale_shape = {2, 2, 1}; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/opencl/kernel/ |
| D | scale.cc | 53 auto scale_shape = scale_tensor->shape(); in CheckSpecs() local
131 GpuTensorInfo scale_shape = GpuTensorInfo(scale_tensor); in InitWeights() local
170 auto scale_shape = scale_tensor->shape(); in Prepare() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/ |
| D | scale_fp32_coder.cc | 47 std::vector<int> scale_shape = scale_tensor->shape(); in CalculateParameter() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | mul_add_fusion.cc | 139 auto scale_shape = scale_tensor_->shape_c(); in AdjustScaleBiasTensorShape() local
185 auto scale_shape = scale_tensor_->shape_c(); in ScaleInputShapeValid() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp16/ |
| D | scale_dynamic_fp16_coder.cc | 109 std::vector<std::string> scale_shape; in CalculateParameter() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | bn_infer.cc | 39 …auto scale_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->GetSh… in InferShape() local
|
| D | bn_training_update.cc | 65 const auto &scale_shape = shapes[kInputIndex3]; in BNTrainingUpdateCheckShapes() local
|
| D | sync_batch_norm.cc | 110 auto scale_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(scale_shape_ptr)[kShape]; in SyncBatchNormInferShape() local
|
| D | scale_and_translate.cc | 56 …auto scale_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->GetSh… in ScaleAndTranslateInferShape() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/ |
| D | test_ops_dynamic_quant_ext.cc | 34 ShapeVector scale_shape; member
|
| D | test_ops_batchnorm.cc | 40 ShapeVector scale_shape; member
|
| D | test_ops_batchnorm_grad.cc | 41 ShapeVector scale_shape; member
|
| D | test_ops_batch_norm_grad_grad.cc | 41 ShapeVector scale_shape; member
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/ |
| D | scale_and_translate_grad.cc | 51 …auto scale_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->GetSh… in ScaleAndTranslateGradInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | batch_norm_grad_grad_gpu_kernel.cc | 54 auto scale_shape = inputs[kIndex2]->GetShapeVector(); in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/npu/op/ |
| D | scale_npu.cc | 162 auto scale_shape = in_tensors.at(SCALE_INDEX).Shape(); in ConvertScaleToMul() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/ |
| D | nn_norm_ops_proto.cc | 313 ge::Shape scale_shape = op.GetInputDescByName("scale").GetShape(); in CUST_IMPLEMT_INFERFUNC() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | batch_norm_grad_grad_cpu_kernel.cc | 52 std::vector<int64_t> scale_shape = inputs.at(kIndex2)->GetShapeVector(); in Resize() local
|