| /third_party/mindspore/mindspore-src/source/tests/st/numpy_native/ |
| D | test_array_ops.py | 114 def mnp_transpose(input_tensor): argument
145 def mnp_expand_dims(input_tensor): argument
176 def mnp_squeeze(input_tensor): argument
217 def mnp_rollaxis(input_tensor): argument
252 def mnp_swapaxes(input_tensor): argument
287 def mnp_reshape(input_tensor): argument
330 def mnp_ravel(input_tensor): argument
355 def mnp_concatenate(input_tensor): argument
447 def mnp_column_stack(input_tensor): argument
472 def mnp_hstack(input_tensor): argument
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/pyboost/customize/ |
| D | max_min.cc | 26 void MinOrMaxCPUCall(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &input_tensor, in MinOrMaxCPUCall()
59 void MinCPUCustomize(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &input_tensor) { in MinCPUCustomize()
63 void MaxCPUCustomize(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &input_tensor) { in MaxCPUCustomize()
|
| D | mean_ext.cc | 28 void MeanExtCPUCall(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &input_tensor, const V… in MeanExtCPUCall()
55 void MeanExtCPUCustomize(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &input_tensor, in MeanExtCPUCustomize()
|
| D | prod_ext.cc | 28 void ProdExtCPUCall(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &input_tensor, const V… in ProdExtCPUCall()
55 void ProdExtCPUCustomize(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &input_tensor, in ProdExtCPUCustomize()
|
| /third_party/mindspore/mindspore-src/source/tests/st/mint/ |
| D | test_repeat_interleave.py | 28 def generate_expect_forward_output(input_tensor, repeats, dim=None): argument
31 def generate_expect_backward_output(input_tensor, repeats, dim): argument
47 def repeat_interleave_forward(input_tensor, repeats, dim, output_size=None): argument
51 def repeat_interleave_backward(input_tensor, repeats, dim, output_size=None): argument
|
| D | test_zeros_like.py | 24 def zeros_like_forward_func(input_tensor, dtype=None): argument
29 def zeros_like_backward_func(input_tensor, dtype=None): argument
|
| D | test_ones_like.py | 24 def ones_like_forward_func(input_tensor, dtype=None): argument
29 def ones_like_backward_func(input_tensor, dtype=None): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/pyboost/customize/ |
| D | max_min.cc | 27 void MinOrMaxGPUCall(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &input_tensor, in MinOrMaxGPUCall()
63 void MinGPUCustomize(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &input_tensor) { in MinGPUCustomize()
67 void MaxGPUCustomize(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &input_tensor) { in MaxGPUCustomize()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/int8/ |
| D | leaky_relu_int8.cc | 58 auto *input_tensor = in_tensors_.at(kInputIndex); in Prepare() local
79 auto *input_tensor = in_tensors_.at(kInputIndex); in ReSize() local
96 auto input_tensor = in_tensors_.at(kInputIndex); in DoExecute() local
|
| D | dynamic_gather_int8.cc | 71 auto input_tensor = in_tensors_.at(0); in Prepare() local
109 auto input_tensor = in_tensors_.at(0); in ReSize() local
176 auto input_tensor = in_tensors_.at(0); in DoGather() local
|
| D | crop_int8.cc | 40 auto *input_tensor = in_tensors_.at(kInputIndex); in Prepare() local
61 auto *input_tensor = in_tensors_.at(kInputIndex); in ReSize() local
|
| D | arithmetic_self_int8.cc | 40 auto *input_tensor = in_tensors_.at(kInputIndex); in Prepare() local
115 auto input_tensor = in_tensors_.at(0); in Run() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/dataset/ |
| D | jieba_tokenizer_op_test.cc | 48 std::shared_ptr<Tensor> input_tensor; in TEST_F() local
76 std::shared_ptr<Tensor> input_tensor; in TEST_F() local
98 std::shared_ptr<Tensor> input_tensor; in TEST_F() local
|
| D | tensor_row_test.cc | 138 std::shared_ptr<Tensor> input_tensor; in TEST_F() local
154 std::shared_ptr<Tensor> input_tensor; in TEST_F() local
170 std::shared_ptr<Tensor> input_tensor; in TEST_F() local
186 std::shared_ptr<Tensor> input_tensor; in TEST_F() local
202 std::shared_ptr<Tensor> input_tensor; in TEST_F() local
218 std::shared_ptr<Tensor> input_tensor; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ |
| D | test_ops_unfold.py | 26 def unfold_forward_func(input_tensor, kernel_size, dilation=1, padding=0, stride=1): argument
31 def unfold_backward_func(input_tensor, kernel_size, dilation=1, padding=0, stride=1): argument
|
| D | test_ops_fold.py | 26 def fold_forward_func(input_tensor, output_size, kernel_size, dilation=1, padding=0, stride=1): argument
31 def fold_backward_func(input_tensor, output_size, kernel_size, dilation=1, padding=0, stride=1): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/ |
| D | masked_fill.cc | 30 … const BaseTensorPtr &input_tensor, const BaseTensorPtr &mask_tensor, in MaskedFillAscendCall()
38 …r MaskedFillAscendCustomize(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &input_tensor, in MaskedFillAscendCustomize()
|
| D | upsample_nearest1d.cc | 28 const BaseTensorPtr &input_tensor, in UpsampleNearest1dAscendCall()
38 const BaseTensorPtr &input_tensor, in UpsampleNearest1DAscendCustomize()
|
| D | upsample_nearest2d.cc | 28 const BaseTensorPtr &input_tensor, in UpsampleNearest2dAscendCall()
38 const BaseTensorPtr &input_tensor, in UpsampleNearest2DAscendCustomize()
|
| D | reduce_all.cc | 30 … const BaseTensorPtr &input_tensor, const std::vector<int64_t> &axis, in ReduceAllAscendCall()
39 …tr ReduceAllAscendCustomize(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &input_tensor, in ReduceAllAscendCustomize()
|
| D | upsample_nearest3d.cc | 29 const BaseTensorPtr &input_tensor, in UpsampleNearest3DAscendCall()
45 const BaseTensorPtr &input_tensor, in UpsampleNearest3DAscendCustomize()
|
| D | upsample_linear1d.cc | 30 const BaseTensorPtr &input_tensor, in UpsampleLinear1DAscendCall()
44 const BaseTensorPtr &input_tensor, in UpsampleLinear1DAscendCustomize()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/ |
| D | scale.c | 118 TensorC *input_tensor = scale->base_.in_[FIRST_INPUT]; in ScaleCalculateParameter() local
202 TensorC *input_tensor = self->in_[i]; in ScaleCheckInputsOutputs() local
239 TensorC *input_tensor = self->in_[FIRST_INPUT]; in ScaleResize() local
267 TensorC *input_tensor = self->in_[FIRST_INPUT]; in ScaleCompute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | binary_cross_entropy_grad.cc | 46 const auto &input_tensor = input_values[kInputIndex1]->cast<tensor::BaseTensorPtr>(); in InferType() local
53 const auto &input_tensor = input_values[kInputIndex1]->cast<tensor::BaseTensorPtr>(); in InferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | cast_tests.cc | 64 …auto *input_tensor = new (std::nothrow) lite::Tensor(kNumberTypeFloat32, shape, mindspore::NHWC, t… in TEST_F() local
99 for (auto &input_tensor : inputs) { in TEST_F() local
173 …auto *input_tensor = new (std::nothrow) lite::Tensor(kNumberTypeFloat16, shape, mindspore::NHWC, t… in TEST_F() local
208 for (auto &input_tensor : inputs) { in TEST_F() local
|