Searched defs:scale_shape (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | scale_fp32_tests.cc | 64 …id TestScaleFp32::Prepare(const std::vector<int> &input_shape, const std::vector<int> &scale_shape, in Prepare()
100 std::vector<int> scale_shape{3}; in TEST_F() local
122 std::vector<int> scale_shape{3}; in TEST_F() local
143 std::vector<int> scale_shape{3}; in TEST_F() local
|
| /third_party/mindspore/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/lite/src/runtime/kernel/opencl/kernel/ |
| D | scale.cc | 46 auto scale_shape = scale_tensor->shape(); in CheckSpecs() local
123 GpuTensorInfo scale_shape = GpuTensorInfo(scale_tensor); in InitWeights() local
162 auto scale_shape = scale_tensor->shape(); in Prepare() local
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
| D | scale_fp32_coder.cc | 69 std::vector<int> scale_shape = scale_tensor->shape(); in CalculateParameter() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | scale_fp32.cc | 90 auto scale_shape = scale_tensor->shape(); in CalculateParameter() local
|
| /third_party/mindspore/mindspore/lite/src/delegate/npu/op/ |
| D | scale_npu.cc | 127 auto scale_shape = in_tensors[SCALE_INDEX].Shape(); in ConvertScaleToMul() local
|
| /third_party/mindspore/mindspore/lite/tools/converter/legacy_optimizer/fusion/ |
| D | mul_add_fusion_pass.cc | 106 bool ScaleInputShapeValid(const std::vector<int> &input_shape, const std::vector<int> &scale_shape, in ScaleInputShapeValid()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | scale_int8.cc | 105 auto scale_shape = scale_tensor->shape(); in InitParameter() local
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | _grad_ops.py | 187 …def infer_shape(self, y_backprop_shape, x_shape, scale_shape, save_mean_shape, save_variance_shape… argument
205 …def infer_shape(self, y_backprop_shape, x_shape, scale_shape, save_mean_shape, save_variance_shape… argument
|