Searched refs:shift_idx (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | pad_and_shift_cpu_kernel.cc | 65 T shift_idx = *reinterpret_cast<T *>(inputs[2]->addr); in LaunchKernel() local
68 if (shift_idx >= static_cast<T>(cum_sum_size_)) { in LaunchKernel()
72 size_t shift_size = static_cast<size_t>(cum_sum_arr[shift_idx]); in LaunchKernel()
73 size_t valid_size = static_cast<size_t>(cum_sum_arr[shift_idx + 1] - shift_size); in LaunchKernel()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/mindrecord/meta/ |
| D | shard_column.cc | 364 uint64_t *num_bytes, uint64_t *shift_idx) { in GetColumnAddressInBlock() argument
366 RETURN_UNEXPECTED_IF_NULL(shift_idx); in GetColumnAddressInBlock()
369 *shift_idx = 0; in GetColumnAddressInBlock()
375 *shift_idx += kInt64Len + BytesBigToUInt64(columns_blob, *shift_idx, kInt64Type); in GetColumnAddressInBlock()
377 *num_bytes = BytesBigToUInt64(columns_blob, *shift_idx, kInt64Type); in GetColumnAddressInBlock()
379 (*shift_idx) += kInt64Len; in GetColumnAddressInBlock()
386 … const std::vector<uint8_t> &columns_blob, uint64_t *num_bytes, uint64_t shift_idx) { in UncompressInt() argument
389 auto num_elements = BytesBigToUInt64(columns_blob, shift_idx, kInt32Type); in UncompressInt()
393 …uint64_t i_source = shift_idx + kBytesOfColumnLen + (num_elements + kNumDataOfByte - 1) / kNumData… in UncompressInt()
397 uint8_t iBitMap = columns_blob[shift_idx + kBytesOfColumnLen + i / kNumDataOfByte]; in UncompressInt()
|
| /third_party/mindspore/tests/st/dynamic_shape/ |
| D | test_pad_and_shift_cpu.py | 30 self.shift_idx = 1
33 return self.pad_and_shift(x, y, self.shift_idx)
|
| /third_party/mindspore/mindspore/ccsrc/minddata/mindrecord/include/ |
| D | shard_column.h | 120 uint64_t *num_bytes, uint64_t *shift_idx);
131 … const std::vector<uint8_t> &columns_blob, uint64_t *num_bytes, uint64_t shift_idx);
|
| /third_party/mindspore/config/ |
| D | op_info.config | 13 …"name": "cum_sum_arr", "param_type": "required"}, {"index": 2, "name": "shift_idx", "param_type": …
|