| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/ |
| D | tbe_kernel_broadcast_selecter.cc | 34 input_shapes_.clear(); in GetShapeInfo()
48 input_shapes_.emplace_back(dynamic_input_shape0_); in GetShapeInfo()
55 input_shapes_.emplace_back(input_shape); in GetShapeInfo()
82 for (const auto &shape : input_shapes_) { in IsBroadCastSupport5HD()
96 for (const auto &shape : input_shapes_) { in IsBroadCastSupport5HD()
101 auto shape_tmp = input_shapes_[0]; in IsBroadCastSupport5HD()
103 input_shapes_.begin(), input_shapes_.end(), in IsBroadCastSupport5HD()
130 for (const auto &shape : input_shapes_) { in IsBroadCastSupportFracZ()
164 for (const auto &shape : input_shapes_) { in IsBroadCastSupportC1HWNCoC0()
178 for (const auto &shape : input_shapes_) { in IsBroadCastSupportC1HWNCoC0()
[all …]
|
| D | tbe_kernel_broadcast_selecter.h | 52 std::vector<std::vector<size_t>> input_shapes_; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | gather_gpu_kernel.h | 57 input_shapes_ = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0); in Init()
61 …CHECK_NULL_INPUT(input_shapes_) || CHECK_NULL_INPUT(index_shapes_) || CHECK_NULL_INPUT(output_shap… in Init()
67 …if (input_shapes_.size() != index_shapes_.size() || input_shapes_.size() != output_shapes_.size())… in Init()
71 int dims = SizeToInt(input_shapes_.size()); in Init()
88 size_t size = GetSize(input_shapes_, true); in InitSizeLists()
104 size_t dim_at_axis_input = input_shapes_[IntToSize(axis_)]; in Reshape()
125 std::vector<size_t> input_shapes_; variable
|
| D | gathernd_gpu_kernel.h | 81 input_shapes_ = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0); in Init()
85 …CHECK_NULL_INPUT(input_shapes_) || CHECK_NULL_INPUT(indices_shapes_) || CHECK_NULL_INPUT(output_sh… in Init()
98 batch_strides_[dim_indices_last - 1] = input_shapes_[dim_indices_last - 1]; in Init()
102 batch_strides_[i - 1] = input_shapes_[i - 1]; in Init()
103 batch_indices_[i - 1] = batch_indices_[i] * input_shapes_[i]; in Init()
126 size_t size = AnfAlgo::TensorSizeInByte<T>(input_shapes_); in InitSizeLists()
145 for (size_t i = dim_indices_last; i < input_shapes_.size(); i++) { in Reshape()
146 dim_after_indices *= input_shapes_[i]; in Reshape()
154 std::vector<size_t> input_shapes_; variable
|
| D | gatherv2_gpu_kernel.h | 58 auto input_dim1 = input_shapes_[IntToSize(axis_)]; in Launch()
78 input_shapes_ = AnfAlgo::GetInputRealDeviceShapeIfExist(kernel_node, 0); in Init()
82 …CHECK_NULL_INPUT(input_shapes_) || CHECK_NULL_INPUT(indices_shapes_) || CHECK_NULL_INPUT(output_sh… in Init()
89 int dims = SizeToInt(input_shapes_.size()); in Init()
102 input_shapes_.clear(); in ResetResource()
115 size_t size = AnfAlgo::TensorSizeInByte<T>(input_shapes_); in InitSizeLists()
129 axis_ = axis_ + SizeToInt(input_shapes_.size()); in Reshape()
149 std::vector<size_t> input_shapes_; variable
|
| D | embedding_lookup_gpu_kernel.h | 56 auto input_dim1 = input_shapes_[0]; in Launch()
72 input_shapes_ = AnfAlgo::GetInputRealDeviceShapeIfExist(kernel_node, 0); in Init()
76 …CHECK_NULL_INPUT(input_shapes_) || CHECK_NULL_INPUT(indices_shapes_) || CHECK_NULL_INPUT(output_sh… in Init()
82 if (input_shapes_.size() < 1) { in Init()
95 input_shapes_.clear(); in ResetResource()
107 size_t size = GetSize(input_shapes_); in InitSizeLists()
149 std::vector<size_t> input_shapes_; variable
|
| D | meshgrid_gpu_kernel.h | 59 broadcasted_input_shape[i] = input_shapes_[i]; in Launch()
91 input_shapes_.push_back(input_size); in Init()
129 input_shapes_.clear(); in ResetResource()
145 for (const size_t &input_shape : input_shapes_) { in InitSizeLists()
157 std::vector<size_t> input_shapes_;
|
| D | tensor_scatter_update_gpu_kernel.h | 111 input_shapes_ = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0); in Init()
114 CHECK_NULL_INPUT(input_shapes_) || CHECK_NULL_INPUT(output_shapes_); in Init()
120 std::vector<size_t> shape_me = input_shapes_; in Init()
156 for (size_t i = 0; i < input_shapes_.size(); i++) { in GetSize()
157 input_size_ *= input_shapes_[i]; in GetSize()
194 std::vector<size_t> input_shapes_; variable
|
| D | tensor_scatter_add_gpu_kernel.h | 111 input_shapes_ = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0); in Init()
114 CHECK_NULL_INPUT(input_shapes_) || CHECK_NULL_INPUT(output_shapes_); in Init()
121 std::vector<size_t> shape_me = input_shapes_; in Init()
161 for (size_t i = 0; i < input_shapes_.size(); i++) { in GetSize()
162 input_size_ *= input_shapes_[i]; in GetSize()
208 std::vector<size_t> input_shapes_; variable
|
| D | tensor_scatter_max_gpu_kernel.h | 108 input_shapes_ = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0); in Init()
111 std::vector<size_t> shape_me = input_shapes_; in Init()
147 for (size_t i = 0; i < input_shapes_.size(); i++) { in GetSize()
148 input_size_ *= input_shapes_[i]; in GetSize()
185 std::vector<size_t> input_shapes_; variable
|
| D | tensor_scatter_sub_gpu_kernel.h | 108 input_shapes_ = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0); in Init()
111 std::vector<size_t> shape_me = input_shapes_; in Init()
147 for (size_t i = 0; i < input_shapes_.size(); i++) { in GetSize()
148 input_size_ *= input_shapes_[i]; in GetSize()
185 std::vector<size_t> input_shapes_; variable
|
| D | tensor_scatter_min_gpu_kernel.h | 107 input_shapes_ = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0); in Init()
110 std::vector<size_t> shape_me = input_shapes_; in Init()
146 for (size_t i = 0; i < input_shapes_.size(); i++) { in GetSize()
147 input_size_ *= input_shapes_[i]; in GetSize()
184 std::vector<size_t> input_shapes_; variable
|
| D | scatter_nd_gpu_kernel.h | 106 input_shapes_ = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 1); in Init()
110 …CHECK_NULL_INPUT(input_shapes_) || CHECK_NULL_INPUT(indices_shapes_) || CHECK_NULL_INPUT(output_sh… in Init()
155 for (size_t i = 0; i < input_shapes_.size(); i++) { in GetSize()
156 input_size_ *= input_shapes_[i]; in GetSize()
182 std::vector<size_t> input_shapes_;
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | gathernd_cpu_kernel.cc | 33 input_shapes_ = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0); in InitKernel()
50 for (size_t i = dim_indices_last; i < input_shapes_.size(); i++) { in InitKernel()
51 dim_after_indices *= input_shapes_[i]; in InitKernel()
62 batch_strides_[dim_indices_last - 1] = input_shapes_[dim_indices_last - 1]; in InitKernel()
67 batch_strides_[i - 1] = input_shapes_[i - 1]; in InitKernel()
68 batch_indices_[i - 1] = batch_indices_[i] * SizeToInt(input_shapes_[i]); in InitKernel()
|
| D | print_cpu_kernel.cc | 32 (void)input_shapes_.emplace_back(input_shape); in InitKernel()
55 … (void)std::transform(input_shapes_[i].begin(), input_shapes_[i].end(), std::back_inserter(shape), in Launch()
|
| D | gathernd_cpu_kernel.h | 39 std::vector<size_t> input_shapes_;
|
| D | print_cpu_kernel.h | 43 std::vector<std::vector<size_t>> input_shapes_;
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/ |
| D | concat_int8_coder.cc | 39 concat_param_->input_shapes_ = nullptr; in Prepare()
60 concat_param_->input_shapes_ = reinterpret_cast<int **>(malloc(sizeof(int *) * input_num)); in Prepare()
61 MS_CHECK_PTR(concat_param_->input_shapes_); in Prepare()
64 … concat_param_->input_shapes_[i] = reinterpret_cast<int *>(malloc(in_shape.size() * sizeof(int))); in Prepare()
65 MS_CHECK_PTR(concat_param_->input_shapes_[i]); in Prepare()
66 …MS_CHECK_RET_CODE(memcpy_s(reinterpret_cast<void *>(concat_param_->input_shapes_[i]), sizeof(int) … in Prepare()
|
| D | concat_int8_coder.h | 39 if (concat_param_->input_shapes_ != nullptr) { in ~ConcatInt8Coder()
40 free(concat_param_->input_shapes_); in ~ConcatInt8Coder()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | concat_int8.cc | 33 concat_param_->input_shapes_ = nullptr; in Init()
78 concat_param_->input_shapes_ = reinterpret_cast<int **>(malloc(sizeof(int *) * input_num)); in ReSize()
79 if (concat_param_->input_shapes_ == nullptr) { in ReSize()
86 … concat_param_->input_shapes_[i] = reinterpret_cast<int *>(malloc(in_shape.size() * sizeof(int))); in ReSize()
87 if (concat_param_->input_shapes_[i] == nullptr) { in ReSize()
92 …memcpy(reinterpret_cast<void *>(concat_param_->input_shapes_[i]), in_shape.data(), sizeof(int) * i… in ReSize()
|
| D | concat_int8.h | 43 if (concat_param_->input_shapes_ != nullptr) { in ~ConcatInt8CPUKernel()
45 int *input_shape = concat_param_->input_shapes_[i]; in ~ConcatInt8CPUKernel()
50 free(concat_param_->input_shapes_); in ~ConcatInt8CPUKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
| D | concat_parameter.h | 29 int **input_shapes_; member
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | concat_int8.c | 40 const int *input_shape = para->input_shapes_[i]; in Int8Concat()
|
| D | quantize.h | 56 int **input_shapes_; member
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ |
| D | akg_kernel_json_decoder.cc | 191 input_shapes_.push_back(input_desc[kJsonKeyShape]); in DecodeInputDesc()
326 std::vector<ShapeVector> input_shapes_; member in mindspore::kernel::__anonc55854f90111::CNodeDecoder
|