Home
last modified time | relevance | path

Searched defs:UpdateOutputShapeAndSize (Results 1 – 25 of 58) sorted by relevance

123

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/acl/acl_kernel/
Dgetnext_kernel_mod.h35 void UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &inputs, in UpdateOutputShapeAndSize() function
Dcustom_op_kernel_mod.h41 void UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &inputs, in UpdateOutputShapeAndSize() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Ddynamic_broadcast_grad_args_cpu_kernel.h51 void UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &inputs, in UpdateOutputShapeAndSize() function
Dnormalize_tuple_index_cpu_kernel.h47 void UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &inputs, in UpdateOutputShapeAndSize() function
Dpad_and_shift_cpu_kernel.cc80 void PadAndShiftCpuKernelMod::UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &inputs, in UpdateOutputShapeAndSize() function in mindspore::kernel::PadAndShiftCpuKernelMod
Ddynamic_assign_cpu_kernel.cc96 void DynamicAssignCpuKernelMod::UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &inputs, in UpdateOutputShapeAndSize() function in mindspore::kernel::DynamicAssignCpuKernelMod
Dsub_and_filter_cpu_kernel.cc92 void SubAndFilterCpuKernelMod::UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &inputs, in UpdateOutputShapeAndSize() function in mindspore::kernel::SubAndFilterCpuKernelMod
Dmasked_select_cpu_kernel.cc101 void MaskedSelectCpuKernelMod::UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &inputs, in UpdateOutputShapeAndSize() function in mindspore::kernel::MaskedSelectCpuKernelMod
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/opapi/aclnn/
Dnon_zero_aclnn_kernel.cc49 void NonZeroAscend::UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &, in UpdateOutputShapeAndSize() function in mindspore::kernel::NonZeroAscend
Dmasked_select_aclnn_kernel.cc49 void MaskedSelectAclnnKernelMod::UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &, in UpdateOutputShapeAndSize() function in mindspore::kernel::MaskedSelectAclnnKernelMod
Dunique2_aclnn_kernel.cc60 void Unique2Ascend::UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &inputs, in UpdateOutputShapeAndSize() function in mindspore::kernel::Unique2Ascend
Dunique_dim_aclnn_kernel.cc60 void UniqueDimAscend::UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &inputs, in UpdateOutputShapeAndSize() function in mindspore::kernel::UniqueDimAscend
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/other/
Ddynamic_reshape_gpu_kernel.h54 void UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &inputs, in UpdateOutputShapeAndSize() function
Ddynamic_stitch_gpu_kernel.cc74 void DynamicStitchKernelMod::UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &inputs, in UpdateOutputShapeAndSize() function in mindspore::kernel::DynamicStitchKernelMod
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/
Dlist_diff_gpu_kernel.cc178 void ListDiffGpuKernelMod::UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &inputs, in UpdateOutputShapeAndSize() function in mindspore::kernel::ListDiffGpuKernelMod
Dunique_gpu_kernel.cc98 void UniqueGpuKernelMod::UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &inputs, in UpdateOutputShapeAndSize() function in mindspore::kernel::UniqueGpuKernelMod
Drange_gpu_kernel.h132 void UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &inputs, in UpdateOutputShapeAndSize() function
Dnon_zero_gpu_kernel.cc99 void NonZeroGpuKernelMod::UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &inputs, in UpdateOutputShapeAndSize() function in mindspore::kernel::NonZeroGpuKernelMod
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/rl/
Dtensor_array_stack_kernel.cc65 void TensorArrayStackKernelMod::UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &inputs, in UpdateOutputShapeAndSize() function in mindspore::kernel::TensorArrayStackKernelMod
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/rl/
Dtensor_array_stack_kernel.cc61 void TensorArrayStackCpuKernelMod::UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &inpu… in UpdateOutputShapeAndSize() function in mindspore::kernel::TensorArrayStackCpuKernelMod
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/map_tensor/
Dmap_tensor_get_grad_cpu_kernel.cc102 void MapTensorGetGradCpuKernelMod::UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &inpu… in UpdateOutputShapeAndSize() function in mindspore::kernel::MapTensorGetGradCpuKernelMod
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/map_tensor/
Dmap_tensor_get_grad_gpu_kernel.cc102 void MapTensorGetGradGpuKernelMod::UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &inpu… in UpdateOutputShapeAndSize() function in mindspore::kernel::MapTensorGetGradGpuKernelMod
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/
Dhcom_receive.cc189 void HcomReceiveKernel::UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &inputs, in UpdateOutputShapeAndSize() function in mindspore::kernel::HcomReceiveKernel
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/sparse/
Dsparse_slice_gpu_kernel.cc100 void SparseSliceGpuKernelMod::UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &inputs, in UpdateOutputShapeAndSize() function in mindspore::kernel::SparseSliceGpuKernelMod
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/
Dnonmaxsuppressionv3_gpu_kernel.cc150 void NonMaxSuppressionV3GpuKernelMod::UpdateOutputShapeAndSize(const std::vector<KernelTensor *> &i… in UpdateOutputShapeAndSize() function in mindspore::kernel::NonMaxSuppressionV3GpuKernelMod

123