Home
last modified time | relevance | path

Searched defs:indices_tensor (Results 1 – 25 of 26) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/opencl/kernel/
Dgather.cc45 auto indices_tensor = in_tensors_.at(SECOND_INPUT); in CheckSpecs() local
174 auto indices_tensor = in_tensors_.at(1); in ConvertTensorToweight() local
254 auto indices_tensor = in_tensors_.at(1); in InitWeights() local
288 auto indices_tensor = in_tensors_.at(1); in InitWeights() local
320 auto indices_tensor = in_tensors_[1]; in PreProcess() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/coreml/op/
Dgather_coreml.cc30 auto indices_tensor = in_tensors_[SECOND_INPUT]; in BuildLayer() local
44 auto indices_tensor = in_tensors_[SECOND_INPUT]; in SetMLOpInOut() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/int8/
Ddynamic_gather_int8.cc110 auto indices_tensor = in_tensors_.at(1); in ReSize() local
127 …nt8CPUKernel::AssignIndicesData(bool isIndicesInt32, int indices_num, lite::Tensor *indices_tensor, in AssignIndicesData()
177 auto indices_tensor = in_tensors_.at(1); in DoGather() local
229 auto indices_tensor = in_tensors_.at(1); in Run() local
DgatherNd_int8.cc72 auto indices_tensor = in_tensors_.at(1); in ReSize() local
100 auto indices_tensor = in_tensors_.at(1); in InitOffset() local
Dgather_int8.cc74 auto indices_tensor = in_tensors_.at(1); in DoGather() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/
Dgather_nd.c25 TensorC *indices_tensor = gather_nd->base_.in_[SECOND_INPUT]; in GatherNdInitOffset() local
129 TensorC *indices_tensor = self->in_[SECOND_INPUT]; in GatherNdResize() local
Dgather.c71 TensorC *indices_tensor = gather->base_.in_[SECOND_INPUT]; in AssignGatherIndicesData() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/
Dgather_nd_tensorrt.cc52 ITensorHelper indices_tensor = input(ctx, 1); in AddInnerOp() local
Dgather_tensorrt.cc72 ITensorHelper indices_tensor = input(ctx, 1); in AddInnerOp() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dsparse_reorder.cc66 Tensor *indices_tensor = ctx.Input(0); in ValidParam() local
158 Tensor *indices_tensor = ctx.Input(0); in Compute() local
Dsparse_apply_momentum.cc58 Tensor *indices_tensor = ctx.Input(4); in ValidParam() local
155 Tensor *indices_tensor = ctx.Input(4); in DoCompute() local
Dsparse_apply_proximal_gradient_descent.cc59 Tensor *indices_tensor = ctx.Input(5); in ValidParam() local
152 Tensor *indices_tensor = ctx.Input(5); in DoCompute() local
Dsparse_apply_adagrad_da.cc89 Tensor *indices_tensor = ctx.Input(4); in ValidParam() local
188 Tensor *indices_tensor = ctx.Input(4); in DoCompute() local
Dsparse_apply_centered_rms_prop.cc67 Tensor *indices_tensor = ctx.Input(9); in ValidParam() local
186 Tensor *indices_tensor = ctx.Input(9); in DoCompute() local
Dsparse_to_dense.cc81 Tensor *indices_tensor = ctx.Input(0); in ValidParam() local
268 Tensor *indices_tensor = ctx.Input(kInput0); in Compute() local
Dsparse_addmm.cc92 Tensor *indices_tensor = ctx.Input(0); in SparseAddmmCheck() local
172 auto *indices_tensor = ctx.Input(0); in ComputeRowAndCol1() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/
Dembedding_dense_backward.cc28 …d::shared_ptr<OpRunner> &op, const BaseTensorPtr &grad_tensor, const BaseTensorPtr &indices_tensor, in EmbeddingDenseBackwardAscendCustomize()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/int8/
Dgather_int8_coder.cc59 auto indices_tensor = input_tensors_.at(1); in DoCode() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp16_grad/
Dunsorted_segment_sum_fp16.cc85 auto indices_tensor = in_tensors_.at(1); in DoExecute() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32_grad/
Dunsorted_segment_sum.cc84 auto indices_tensor = in_tensors_.at(1); in DoExecute() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dslice_to_indices.cc73 …auto indices_tensor = std::make_shared<tensor::Tensor>(kNumberTypeInt64, ShapeVector{0}, &empty_st… in ConstSliceToIndices() local
97 …auto indices_tensor = std::make_shared<tensor::Tensor>(kNumberTypeInt64, indices_shp, indices.data… in ConstSliceToIndices() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Dseed_adapter.cc45 …tensor::TensorPtr indices_tensor = std::make_shared<tensor::Tensor>(type->type_id(), indices_shape… in CreateTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/expanders/
Dsolve_triangular.cc79 auto indices_tensor = in Expand() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/
Dpynative_utils.cc939 const auto &indices_tensor = coo_tensor->GetIndices(); in SetValueGradInfo() local
943 const auto &indices_tensor = csr_tensor->GetIndices(); in SetValueGradInfo() local
2128 auto indices_tensor = coo_tensor->GetIndices(); in WrapCOOTensor() local
2140 auto indices_tensor = csr_tensor->GetIndices(); in WrapCSRTensor() local
2355 const auto &indices_tensor = coo_tensor->GetIndices(); in SetGradInfoForInputs() local
2359 const auto &indices_tensor = csr_tensor->GetIndices(); in SetGradInfoForInputs() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Druntime_shape_fusion_pass.cc327 auto indices_tensor = src_tensors_->at(post_node->input_indices_.at(1)); in UpdateShapeFusionMatrix() local

12