Searched refs:offset_shape (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | scale_fp32_tests.cc | 35 … const std::vector<int> &offset_shape, const std::vector<int> &output_shape, float *input_data,
65 … const std::vector<int> &offset_shape, const std::vector<int> &output_shape, in Prepare() argument
76 offset_tensor_.set_shape(offset_shape); in Prepare()
101 std::vector<int> offset_shape{3}; in TEST_F() local
109 …Prepare(input_shape, scale_shape, offset_shape, output_shape, in_data, scale_data, offset_data, ou… in TEST_F()
123 std::vector<int> offset_shape{3}; in TEST_F() local
131 …Prepare(input_shape, scale_shape, offset_shape, output_shape, in_data, scale_data, offset_data, ou… in TEST_F()
144 std::vector<int> offset_shape{3}; in TEST_F() local
152 …Prepare(input_shape, scale_shape, offset_shape, output_shape, in_data, scale_data, offset_data, ou… in TEST_F()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/ |
| D | scale.cc | 134 GpuTensorInfo offset_shape = GpuTensorInfo(offset_tensor); in InitWeights() local
135 std::vector<char> offset_data(offset_shape.ElementsC4Num * FLT_size, 0); in InitWeights()
137 …WCToNHWC4(offset_tensor->data(), offset_data.data(), offset_dtype_flag, fp16_enable, offset_shape); in InitWeights()
138 ImageSize offset_img_size{offset_shape.width, offset_shape.height, FLT_type}; in InitWeights()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ascend/ |
| D | ascend_ps_cache.cc | 223 std::vector<size_t> offset_shape = {1}; in HashSwapOut() local
224 …::vector<std::vector<size_t>> input_shape = {hash_table_shape, swap_out_index_shape, offset_shape}; in HashSwapOut()
264 std::vector<size_t> offset_shape = {1}; in HashSwapIn() local
266 offset_shape}; in HashSwapIn()
267 std::vector<std::vector<size_t>> output_shape = {offset_shape}; in HashSwapIn()
|
| /third_party/mindspore/mindspore/lite/tools/converter/legacy_optimizer/fusion/ |
| D | mul_add_fusion_pass.cc | 107 const std::vector<int> &offset_shape) { in ScaleInputShapeValid() argument
117 if (scale_shape != offset_shape) { in ScaleInputShapeValid()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | _embedding_cache_ops.py | 95 def check_shape(self, input_x_shape, max_num_shape, offset_shape): argument
|