Searched refs:begin_tensor (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/ |
| D | slice_int8_tests.cc | 39 lite::Tensor begin_tensor(kNumberTypeInt32, {4}); in TEST_F() local
41 begin_tensor.set_data(begin_data); in TEST_F()
51 std::vector<lite::Tensor *> inputs = {&in_tensor, &begin_tensor, &size_tensor}; in TEST_F()
76 begin_tensor.set_data(nullptr); in TEST_F()
91 lite::Tensor begin_tensor(kNumberTypeInt32, {5}); in TEST_F() local
93 begin_tensor.set_data(begin_data); in TEST_F()
103 std::vector<lite::Tensor *> inputs = {&in_tensor, &begin_tensor, &size_tensor}; in TEST_F()
129 begin_tensor.set_data(nullptr); in TEST_F()
143 lite::Tensor begin_tensor(kNumberTypeInt32, {6}); in TEST_F() local
145 begin_tensor.set_data(begin_data); in TEST_F()
[all …]
|
| /third_party/mindspore/mindspore/lite/src/ops/compat/v0/ |
| D | slice_compat_v0.cc | 48 …auto begin_tensor = AttrToTensor(begin_shape.data(), begin_shape.size(), true, kNumberTypeInt32, t… in TransferSliceAttr() local
49 if (begin_tensor == nullptr) { in TransferSliceAttr()
53 dst_tensors->push_back(begin_tensor); in TransferSliceAttr()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | strided_slice_infer.c | 54 int HandleAxesCheckNull(const TensorC *input_tensor, const TensorC *begin_tensor, int *begin_data, in HandleAxesCheckNull() argument
56 …if (input_tensor == NULL || begin_tensor == NULL || end_tensor == NULL || begin_data == NULL || en… in HandleAxesCheckNull()
63 const TensorC *begin_tensor = inputs[1]; in HandleAxesInputNotExist() local
64 int *begin_data = (int *)(begin_tensor->data_); in HandleAxesInputNotExist()
72 transfer_buffer->ndim_ = GetElementNum(begin_tensor); in HandleAxesInputNotExist()
111 const TensorC *begin_tensor = inputs[1]; in HandleAxesInputExist() local
112 int *begin_data = (int *)(begin_tensor->data_); in HandleAxesInputExist()
116 …int handle_check_ret = HandleAxesCheckNull(input_tensor, begin_tensor, begin_data, end_tensor, end… in HandleAxesInputExist()
123 int begin_ndim = GetElementNum(begin_tensor); in HandleAxesInputExist()
|
| D | strided_slice_grad_infer.c | 83 const TensorC *begin_tensor = inputs[2]; in StridedSliceGradInferShape() local
84 int *begin_data = (int *)(begin_tensor->data_); in StridedSliceGradInferShape()
88 size_t ndim_ = (size_t)GetElementNum(begin_tensor); in StridedSliceGradInferShape()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | slice_base.cc | 45 auto begin_tensor = in_tensors_[kNumInput1]; in ReSize() local
47 MS_ASSERT(in_tensor->shape().size() == static_cast<size_t>(begin_tensor->ElementsNum())); in ReSize()
50 auto begin = reinterpret_cast<int32_t *>(begin_tensor->data()); in ReSize()
|
| D | strided_slice.cc | 74 auto begin_tensor = in_tensors_.at(1); in ReSize() local
75 if (input_tensor->shape().size() > DIMENSION_8D || begin_tensor->shape().size() > DIMENSION_8D) { in ReSize()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/common/ |
| D | strided_slice_tests.cc | 30 const lite::Tensor *begin_tensor, const lite::Tensor *end_tensor, in InitStridedSliceParam() argument
32 int dim = begin_tensor->ElementsNum(); in InitStridedSliceParam()
34 int *begin = reinterpret_cast<int *>(begin_tensor->data()); in InitStridedSliceParam()
|
| /third_party/mindspore/mindspore/lite/src/delegate/npu/pass/ |
| D | npu_fusion_pass.cc | 345 auto begin_tensor = cur_op->inputs().at(BEGIN_INDEX); in StridedSliceFusion() local
346 int *begin = reinterpret_cast<int *>(begin_tensor.MutableData()); in StridedSliceFusion()
|