| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/image/ |
| D | horizontal_flip_op.cc | 33 dsize_t num_batch = input->Size() / (input_shape[-3] * input_shape[-2] * input_shape[-1]); in Compute() local
|
| D | vertical_flip_op.cc | 34 dsize_t num_batch = input->Size() / (input_shape[-3] * input_shape[-2] * input_shape[-1]); in Compute() local
|
| D | random_horizontal_flip_op.cc | 44 … dsize_t num_batch = input[i]->Size() / (input_shape[-3] * input_shape[-2] * input_shape[-1]); in Compute() local
|
| D | normalize_op.cc | 52 dsize_t num_batch = input->Size() / (input_shape[-3] * input_shape[-2] * input_shape[-1]); in Compute() local
|
| D | center_crop_op.cc | 94 dsize_t num_batch = input->Size() / (input_shape[-3] * input_shape[-2] * input_shape[-1]); in Compute() local
|
| D | resize_op.cc | 68 …dsize_t num_batch = input->Size() / (original_shape[-3] * original_shape[-2] * original_shape[-1]); in Compute() local
|
| D | rotate_op.cc | 71 …dsize_t num_batch = input->Size() / (original_shape[-3] * original_shape[-2] * original_shape[-1]); in Compute() local
|
| D | random_crop_and_resize_op.cc | 77 dsize_t num_batch = input[i]->Size() / (input_shape[-3] * input_shape[-2] * input_shape[-1]); in Compute() local
|
| D | random_crop_op.cc | 178 dsize_t num_batch = input[i]->Size() / (input_shape[-3] * input_shape[-2] * input_shape[-1]); in Compute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | max_unpool2d_cpu_kernel.cc | 85 size_t num_batch = LongToSize(input_shape_[kInputIndex0]); in LaunchKernel() local
118 size_t num_batch = LongToSize(input_shape_[kInputIndex0]); in LaunchKernel() local
|
| D | max_unpool2d_grad_cpu_kernel.cc | 86 size_t num_batch = LongToSize(grads_shape_[kInputIndex0]); in LaunchKernel() local
119 size_t num_batch = LongToSize(grads_shape_[kInputIndex0]); in LaunchKernel() local
|
| D | sparse_matrix_transpose_cpu_kernel.cc | 201 size_t num_batch = batch_pointers - static_cast<size_t>(1); in LaunchKernel() local
287 size_t num_batch = batch_pointers - static_cast<size_t>(1); in LaunchcomplexKernel() local
|
| D | dilation2d_cpu_kernel.cc | 80 int64_t num_batch = input_shape_[kFormatNCHWIndexN]; in LaunchKernel() local
|
| D | dilation2d_backprop_input_cpu_kernel.cc | 82 size_t num_batch = LongToSize(i_input_shape_[kFormatNCHWIndexN]); in LaunchKernel() local
|
| D | dilation2d_backprop_filter_cpu_kernel.cc | 82 size_t num_batch = LongToSize(f_input_shape_[kFormatNCHWIndexN]); in LaunchKernel() local
|
| D | max_unpool3d_cpu_kernel.cc | 87 size_t num_batch = static_cast<size_t>(input_shape_[kInputIndex0]); in LaunchKernel() local
|
| D | max_unpool3d_grad_cpu_kernel.cc | 89 auto num_batch = LongToSize(grads_shape_[kInputIndex0]); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | sparse_matrix_transpose.cc | 178 auto num_batch = ctx.Input(1)->NumElements() - 1; in SparseMatrixTransposeCompute() local
266 auto num_batch = ctx.Input(1)->NumElements() - 1; in SparseMatrixTransposeComputecomplex() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | sparse_matrix_ordering_amd.cc | 44 const int64_t num_batch = b_ptrs_shape[kZero] - 1; in SparseMatrixOrderingAMDInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/ |
| D | sparse_matrix_ordering_amd_cpu_kernel.cc | 97 const int64_t num_batch = x_batch_pointers_shape_[0] - 1; in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/dataset/engine/ |
| D | datasets.py | 707 def sync_wait(self, condition_name, num_batch=1, callback=None): argument
1896 def sync_update(self, condition_name, num_batch=None, data=None): argument
2949 def __init__(self, input_dataset, condition_name, num_batch, callback=None): argument
|