Home
last modified time | relevance | path

Searched refs:CastShapeSizeToLong (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dtensor_copy_slices_cpu_kernel.cc38 CastShapeSizeToLong(input_shape, &input_shape_); in InitKernel()
39 CastShapeSizeToLong(update_shape, &update_shape_); in InitKernel()
40 CastShapeSizeToLong(output_shape, &output_shape_); in InitKernel()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/
Dtensor_copy_slices.cc104 CastShapeSizeToLong(input_shape, &input_shape_); in GetInputOutputInfo()
105 CastShapeSizeToLong(update_shape, &update_shape_); in GetInputOutputInfo()
106 CastShapeSizeToLong(output_shape, &output_shape_); in GetInputOutputInfo()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dcommon_utils.h138 void CastShapeSizeToLong(const std::vector<size_t> &shape, std::vector<int64_t> *long_shape);
Dcommon_utils.cc923 void CastShapeSizeToLong(const std::vector<size_t> &shape, std::vector<int64_t> *long_shape) { in CastShapeSizeToLong() function