| /third_party/mindspore/tests/st/ops/ascend/test_aicpu_ops/ |
| D | test_tensor_setitem.py | 28 def test_tensor_setitem_net(input_shape, update_shape, slice_tuple, dtype): argument
39 def test_tensor_setitem_net_many_dtype(input_shape, update_shape, slice_tuple, dtypes): argument
|
| D | test_tensor_copy_slices.py | 51 def test_tensor_copy_slices_net(input_shape, update_shape, begin, end, strides, dtype): argument
62 def test_tensor_copy_slices_net_many_dtype(input_shape, update_shape, begin, end, strides, dtypes): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | tensor_copy_slices_gpu_kernel.h | 75 auto update_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 1); in Init() local
111 auto update_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 1); in CheckAtrrAndShapeValid() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/ |
| D | tensor_copy_slices.cc | 102 auto update_shape = AnfAlgo::GetPrevNodeOutputInferShape(anf_node, 1); in GetInputOutputInfo() local
128 auto update_shape = AnfAlgo::GetInputDeviceShape(anf_node, 1); in GetInputOutputTotalCount() local
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | scatter_nd_update.cc | 29 …auto update_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->Buil… in InferShape() local
|
| D | scatter_nd.cc | 33 …auto update_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->Buil… in InferShape() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | tensor_copy_slices_cpu_kernel.cc | 35 auto update_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 1); in InitKernel() local
|
| D | update_cache_cpu_kernel.cc | 66 auto update_shape = AnfAlgo::GetPrevNodeOutputInferShape(node, 2); in LaunchKernel() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | scatter_nd_fp32.cc | 78 auto update_shape = update->shape(); in ReSize() local
|
| D | scatter_nd_update_fp32.cc | 71 auto update_shape = update->shape(); in ReSize() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | slice_copy_impl.cu | 78 void CopySlices(const std::vector<size_t> &update_shape, const std::vector<int64_t> &begin, in CopySlices()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | _embedding_cache_ops.py | 50 def check_shape(self, input_x_shape, indices_shape, update_shape, max_num_shape): argument
|
| D | inner_ops.py | 324 def infer_shape(self, w_norm_shape, g_norm_shape, lr_shape, update_shape, var_shape): argument
|
| D | _grad_ops.py | 1493 … dy_shape, dh_shape, update_shape, reset_shape, new_shape, hnew_shape, seq_shape, mask_shape): argument
|