Home
last modified time | relevance | path

Searched defs:in_tensor (Results 1 – 25 of 154) sorted by relevance

1234567

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/
Ddelegate_utils.h32 for (auto in_tensor : cur_op->inputs()) { in FindPreOps() local
60 for (auto in_tensor : op->inputs()) { in FindPreNextOps() local
69 for (auto in_tensor : op->inputs()) { in FindPreNextOps() local
91 for (auto in_tensor : inputs) { in GetGraphInOutOps() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/coreml/op/
Dconcat_coreml.cc44 for (const auto &in_tensor : in_tensors_) { in BuildLayer() local
63 for (const auto &in_tensor : in_tensors_) { in SetMLOpInOut() local
Darithmetic_coreml.cc91 for (const auto &in_tensor : in_tensors_) { in BuildBroadcastableArithmetic() local
105 for (const auto &in_tensor : in_tensors_) { in SetMLOpInOut() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp16/
Dquant_dtype_cast_fp16.cc35 auto in_tensor = in_tensors_.front(); in Prepare() local
82 auto in_tensor = in_tensors_.front(); in ReSize() local
185 auto in_tensor = inputs.front(); in CpuQuantDTypeCastFp16KernelCreator() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/common/
Dstrided_slice_tests.cc28 void InitStridedSliceParam(StridedSliceParameter *param, const lite::Tensor *in_tensor, in InitStridedSliceParam()
53 lite::Tensor in_tensor(kNumberTypeFloat32, {1, 2, 4}); in TEST_F() local
96 lite::Tensor in_tensor(kNumberTypeFloat32, {1, 2, 4, 1, 1, 1, 1}); variable
143 lite::Tensor in_tensor(kNumberTypeInt8, {2, 2, 2, 3, 1, 1, 1, 1}); variable
188 lite::Tensor in_tensor(kNumberTypeFloat32, {1, 2, 4, 1, 1, 1, 1}); in TEST_F() local
233 lite::Tensor in_tensor(kNumberTypeFloat32, {1, 2, 4, 1, 1, 1, 1}); in TEST_F() local
278 lite::Tensor in_tensor(kNumberTypeInt8, {2, 3, 4}); in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/
Ddelegate_utils.h63 for (auto in_tensor : op->inputs()) { in GetGraphInTensors() local
124 for (auto in_tensor : op->inputs()) { in GraphInTensors() local
181 for (auto in_tensor : cur_op->inputs()) { in FindPreOps() local
217 for (auto in_tensor : inputs) { in GetGraphInOutOps() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/executor/
Dkernel_exec.cc30 return std::all_of(in_tensors.begin(), in_tensors.end(), [&](lite::Tensor *in_tensor) { in IsReady()
86 for (auto &in_tensor : in_tensors()) { in DoExecute() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/int8/
Dsplit_int8.cc50 auto in_tensor = in_tensors_.at(kInputIndex); in Prepare() local
107 auto in_tensor = in_tensors_.at(kInputIndex); in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Didentityn.cc40 Tensor *in_tensor = ctx.Input(idx); in IdentityNParamCheck() local
64 Tensor *in_tensor = ctx.Input(idx); in Compute() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/
Dcrop.c49 TensorC *in_tensor = self->in_[FIRST_INPUT]; in CropResize() local
64 TensorC *in_tensor = self->in_[FIRST_INPUT]; in CropCompute() local
Dstrided_slice.c95 TensorC *in_tensor = strided_slice->base_.in_[FIRST_INPUT]; in StridedSliceMatchInOutShapeEqualPattern() local
122 TensorC *in_tensor = strided_slice->base_.in_[FIRST_INPUT]; in StridedSliceSoftCopyInputToOutput() local
167 TensorC *in_tensor = strided_slice->base_.in_[FIRST_INPUT]; in StridedSliceMatchFastPattern() local
Daddn.c75 TensorC *in_tensor = addn->base_.in_[i]; in AddNComputeNoParallel() local
109 TensorC *in_tensor = self->in_[i]; in AddnCompute() local
Dsoftmax.c67 TensorC *in_tensor = softmax->base_.in_[FIRST_INPUT]; in InitSoftmaxParam() local
109 TensorC *in_tensor = self->in_[FIRST_INPUT]; in SoftmaxResize() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/
Dslice_int8_tests.cc31 lite::Tensor in_tensor(kNumberTypeInt8, {1, 3, 2, 3}); in TEST_F() local
82 lite::Tensor in_tensor(kNumberTypeInt8, {2, 1, 3, 2, 3}); in TEST_F() local
134 lite::Tensor in_tensor(kNumberTypeInt8, {2, 1, 1, 3, 2, 3}); in TEST_F() local
186 lite::Tensor in_tensor(kNumberTypeInt8, {2, 1, 1, 1, 3, 2, 3}); in TEST_F() local
238 lite::Tensor in_tensor(kNumberTypeInt8, {2, 1, 1, 1, 1, 3, 2, 3}); in TEST_F() local
291 lite::Tensor in_tensor(kNumberTypeInt8, {2, 1, 1, 1, 1, 3, 2, 3}); in TEST_F() local
344 lite::Tensor in_tensor(kNumberTypeInt8, {2, 1, 1, 1, 1, 3, 2, 3}); in TEST_F() local
399 lite::Tensor in_tensor(kNumberTypeInt8, {2, 1, 1, 1, 1, 3, 2, 3}); in TEST_F() local
Dl2_norm_int8_tests.cc31 lite::Tensor in_tensor(kNumberTypeInt8, {2, 1, 1, 5}); in TEST_F() local
76 lite::Tensor in_tensor(kNumberTypeInt8, {1, 1, 1, 51}); in TEST_F() local
Drelux_int8_tests.cc31 lite::Tensor in_tensor(kNumberTypeInt8, {2, 2}); in TEST_F() local
77 lite::Tensor in_tensor(kNumberTypeInt8, {2, 4}); in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/
Dshape_fusion_infer.c20 int CalculateOutput(const TensorC *in_tensor, const TensorC *matrix_tensor, TensorC *out_tensor, si… in CalculateOutput()
68 const TensorC *in_tensor = inputs[0]; in ShapeFusionInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/
Dtile_fp32_tests.cc30 lite::Tensor in_tensor(kNumberTypeFloat32, {2, 2}); in TEST_F() local
74 lite::Tensor in_tensor(kNumberTypeFloat32, {2, 2}); in TEST_F() local
118 lite::Tensor in_tensor(kNumberTypeFloat32, {2, 2}); in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/opencl/kernel/
Dgl_to_cl.cc48 auto in_tensor = in_tensors_.front(); in Prepare() local
96 auto in_tensor = in_tensors_.front(); in Run() local
Dscale.cc50 auto in_tensor = in_tensors_.at(INPUT_DATA); in CheckSpecs() local
93 auto *in_tensor = in_tensors_[INPUT_DATA]; in InitWeights() local
167 auto in_tensor = in_tensors_.at(INPUT_DATA); in Prepare() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/base/
Dsplit_base.cc43 int SplitBaseCPUKernel::CheckAndInitSplitParam(const lite::Tensor &in_tensor, SplitParameter *param… in CheckAndInitSplitParam()
103 auto in_tensor = in_tensors_.front(); in ReSize() local
Dgroup_convolution_base.cc99 auto in_tensor = in_tensors_.front(); in PreProcess() local
138 auto in_tensor = in_tensors_.front(); in InitGroupParam() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/npu/
Dnpu_subgraph.cc49 void NPUSubGraph::set_input(mindspore::MSTensor in_tensor, int index) { in set_input()
105 for (const auto &in_tensor : cur_op->inputs()) { in FindPreOps() local
169 auto in_tensor = op->inputs()[i]; in BuildNPUInputOp() local
Dtranspose_kernel.cc33 mindspore::MSTensor in_tensor = inputs()[0]; in Execute() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/nnacl/
Dnnacl_reshape.cc34 auto in_tensor = in_tensors().front(); in Run() local

1234567