| /third_party/mindspore/tests/st/numpy_native/ |
| D | test_array_ops.py | 115 def mnp_transpose(input_tensor): argument
146 def mnp_expand_dims(input_tensor): argument
177 def mnp_squeeze(input_tensor): argument
218 def mnp_rollaxis(input_tensor): argument
253 def mnp_swapaxes(input_tensor): argument
288 def mnp_reshape(input_tensor): argument
331 def mnp_ravel(input_tensor): argument
356 def mnp_concatenate(input_tensor): argument
448 def mnp_column_stack(input_tensor): argument
473 def mnp_hstack(input_tensor): argument
[all …]
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | leaky_relu_int8.cc | 48 auto *input_tensor = in_tensors_.at(kInputIndex); in Init() local
79 auto *input_tensor = in_tensors_.at(kInputIndex); in ReSize() local
122 auto input_tensor = in_tensors_.at(kInputIndex); in DoExecute() local
|
| D | crop_int8.cc | 33 auto *input_tensor = in_tensors_.at(kInputIndex); in Init() local
65 auto input_tensor = in_tensors_.at(kInputIndex); in DoExecute() local
|
| D | arithmetic_self_int8.cc | 32 auto *input_tensor = in_tensors_.at(kInputIndex); in Init() local
106 auto input_tensor = in_tensors_.at(0); in Run() local
|
| D | space_to_batch_int8.cc | 29 auto input_tensor = in_tensors_.at(0); in Run() local
|
| /third_party/mindspore/tests/ut/cpp/dataset/ |
| D | jieba_tokenizer_op_test.cc | 45 std::shared_ptr<Tensor> input_tensor; in TEST_F() local
70 std::shared_ptr<Tensor> input_tensor; in TEST_F() local
89 std::shared_ptr<Tensor> input_tensor; in TEST_F() local
|
| D | tensor_row_test.cc | 117 std::shared_ptr<Tensor> input_tensor; in TEST_F() local
130 std::shared_ptr<Tensor> input_tensor; in TEST_F() local
143 std::shared_ptr<Tensor> input_tensor; in TEST_F() local
156 std::shared_ptr<Tensor> input_tensor; in TEST_F() local
169 std::shared_ptr<Tensor> input_tensor; in TEST_F() local
182 std::shared_ptr<Tensor> input_tensor; in TEST_F() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | gather_fp16.cc | 45 auto input_tensor = in_tensors_.at(0); in Init() local
69 auto input_tensor = in_tensors_.at(0); in DoGather() local
140 auto input_tensor = in_tensors_.at(0); in Run() local
|
| /third_party/mindspore/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
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | space_to_batch_fp32.cc | 29 auto input_tensor = in_tensors_.at(FIRST_INPUT); in ProcessInput() local
73 auto input_tensor = in_tensors_.at(FIRST_INPUT); in ReSize() local
|
| D | prelu_fp32.cc | 74 auto input_tensor = in_tensors_.at(0); in ReSize() local
90 auto input_tensor = in_tensors_[0]; in Run() local
|
| D | cumsum_fp32.cc | 36 auto input_tensor = kernel->in_tensors().at(0); in CumsumLaunch() local
61 auto input_tensor = in_tensors_.at(0); in ReSize() local
|
| /third_party/mindspore/tests/mindspore_test_framework/apps/ |
| D | bert_attention_submodules.py | 128 def construct(self, input_tensor, query_layer): argument
180 def construct(self, input_tensor, attention_mask): argument
211 def construct(self, input_tensor): argument
330 def construct(self, input_tensor, attention_probs): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/sponge/pme/ |
| D | fft_3d_impl.cu | 20 void FFT3D(int Nfft, T *input_tensor, Complex<T> *output_tensor, const cufftHandle &FFT_plan_r2c, c… in FFT3D()
|
| D | ifft_3d_impl.cu | 20 void IFFT3D(int Nfft, Complex<T> *input_tensor, T *output_tensor, const cufftHandle &FFT_plan_c2r, in IFFT3D()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | strided_slice_fp32_tests.cc | 139 lite::Tensor input_tensor; in TEST_F() local
193 lite::Tensor input_tensor; in TEST_F() local
254 lite::Tensor input_tensor; in TEST_F() local
315 lite::Tensor input_tensor; in TEST_F() local
368 lite::Tensor input_tensor; in TEST_F() local
429 lite::Tensor input_tensor; in TEST_F() local
575 lite::Tensor input_tensor; in TEST_F() local
|
| /third_party/mindspore/mindspore/lite/tools/converter/quantizer/quant_helper/ |
| D | quant_dtype_cast_quant_param_propogator.cc | 26 auto &input_tensor = graph->allTensors.at(node.inputIndex.at(0)); in PropogateQuantParams() local
|
| D | conv_quant_type_determiner.cc | 26 auto &input_tensor = graph.allTensors.at(node->inputIndex.at(kInputIndex)); in DetermineQuantWeight() local
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/image/ |
| D | adjust_gamma_op.cc | 34 std::shared_ptr<Tensor> input_tensor; in Compute() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | softmax_base.cc | 42 auto input_tensor = in_tensors_.front(); in ReSize() local
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/ |
| D | quant_dtype_cast_tests.cc | 42 lite::Tensor input_tensor; in TEST_F() local
94 lite::Tensor input_tensor; in TEST_F() local
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/audio/kernels/ |
| D | vol_op.cc | 26 std::shared_ptr<Tensor> input_tensor; in Compute() local
|
| D | amplitude_to_db_op.cc | 32 std::shared_ptr<Tensor> input_tensor; in Compute() local
|
| D | time_masking_op.cc | 41 std::shared_ptr<Tensor> input_tensor; in Compute() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/ |
| D | data_prepare_actor.cc | 211 const auto &input_tensor = tensors[j]; in PrepareDataForDeviceTensorStore() local
242 const auto &input_tensor = tensors[j]; in PrepareDataForHostTensorQueue() local
286 const auto &input_tensor = tensors[j]; in PrepareDataForStepMode() local
521 const auto &input_tensor = tensors[i]; in PrepareDeviceTensorStoreForControlNode() local
547 const auto &input_tensor = tensors[i]; in PrepareHostTensorQueueForControlNode() local
|