| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | convert_tensor_eliminate.h | 36 constexpr size_t tensor_index = 1; in operator() local
75 constexpr size_t tensor_index = 1; in operator() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_node_parser.h | 53 …STATUS GetTfliteData(const int32_t tensor_index, const std::vector<std::unique_ptr<tflite::TensorT… in GetTfliteData()
144 STATUS TransTfliteDataToVec2D(const int32_t tensor_index, in TransTfliteDataToVec2D()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pybind_api/ir/ |
| D | tensor_index_py.cc | 41 std::ostream &operator<<(std::ostream &stream, const TensorIndex &tensor_index) { in operator <<()
238 …auto tensor_index = IsStubTensor(array_like) ? ConvertStubTensor(array_like) : py::cast<TensorPtr>… in DeepTensorToNdArray() local
369 TensorIndex TensorIndex::SliceToArray(const TensorIndex &tensor_index, const ShapeVector &shape) { in SliceToArray()
467 …ple TensorIndex::GenerateNonZeroIndex(const ShapeVector &data_shape, const TensorPtr &tensor_index, in GenerateNonZeroIndex()
490 … const TensorPtr &tensor_index, bool check_align) { in GenerateNonZeroIndexTensorList()
505 …:TensorGetitemByTupleParseTensorIndex(const ShapeVector &data_shape, const TensorPtr &tensor_index, in TensorGetitemByTupleParseTensorIndex()
682 TensorPtr tensor_index = std::make_shared<Tensor>(int_index); in TensorGetitemByTuple() local
688 TensorPtr tensor_index = sequence_list.tensor(); in TensorGetitemByTuple() local
693 const TensorPtr &tensor_index = index.tensor(); in TensorGetitemByTuple() local
721 ….begin(), tensor_indexes.end(), std::back_inserter(tensor_indexes_shapes), [](auto &tensor_index) { in TensorGetitemByTuple()
[all …]
|
| D | tensor_index_py.h | 197 auto tensor_index = index.cast<TensorPtr>(); in InitByNone() local
212 auto tensor_index = IsStubTensor(index) ? ConvertStubTensor(index) : index.cast<TensorPtr>(); in NormalizeIndex() local
434 static inline TensorIndex SequenceToTensor(const TensorIndex &tensor_index, int64_t dim_size) { in SequenceToTensor()
546 static inline TensorIndex UnpackTuple(const TensorIndex &tensor_index) { in UnpackTuple()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/control/ |
| D | tensorlist_getitem_infer.c | 48 TensorC *tensor_index = input0->tensors_[index]; in TensorListGetItemInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/lite_exporter/ |
| D | anf_exporter.cc | 375 size_t tensor_index; in SetSubGraphInputIndex() local
407 size_t tensor_index; in SetSubGraphOutputIndex() local
1030 auto tensor_index = NewFbTensor(meta_graphT, schema_tensor.release()); in ConvertInputValueNode() local
1057 size_t tensor_index; in SetOpInputNode() local
1130 auto tensor_index = NewFbTensor(meta_graphT, ms_tensor); in SetOpOutputNode() local
1139 auto tensor_index = NewFbTensor(meta_graphT, ms_tensor); in SetOpOutputNode() local
1166 auto tensor_index = NewFbTensor(meta_graphT, ms_tensor); in SetOpOutputNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/composite/multitype_ops/ |
| D | getitem_impl.py | 170 def _tuple_getitem_by_tensor(data, tensor_index): argument
323 def _tensor_getitem_by_tensor(data, tensor_index): argument
|
| D | _compile_utils.py | 633 def _check_dim_shape_valid(data, tensor_index): argument
643 def tensor_index_by_bool_tensor(data, tensor_index): argument
651 def tensor_index_by_tensor(data, tensor_index): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/train/optimizer/fusion/ |
| D | gru_fusion_pass.cc | 73 … [&graph](uint32_t tensor_index) { return graph->allTensors.size() > tensor_index; }) && in CheckCommon()
75 … [&graph](uint32_t tensor_index) { return graph->allTensors.size() > tensor_index; }); in CheckCommon()
506 for (auto tensor_index : in_tensor_indexes) { in FillRealPattern() local
519 for (auto tensor_index : out_tensor_indexes) { in FillRealPattern() local
721 auto CommonCheck = [&link_infos](uint32_t tensor_index) { in MatchPatten()
735 for (auto tensor_index : stack_node->inputIndex) { in MatchPatten() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/nnrt/ |
| D | nnrt_delegate.cc | 513 uint32_t tensor_index; // the index of node in/out tensors which the tensor is located at. member
632 uint32_t tensor_index = iter - node->input_indices_.begin(); in CreateNNRTSubgraphKernels() local
654 uint32_t tensor_index = iter - node->output_indices_.begin(); in CreateNNRTSubgraphKernels() local
871 auto tensor_index = node->input_indices_[i]; in DequantNodeInputs() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/executor/ |
| D | sub_graph_kernel.cc | 168 …hKernel::InsertInEdge(KernelExec *kernel, KernelExec *replace_kernel, const size_t &tensor_index) { in InsertInEdge()
187 …Kernel::InsertOutEdge(KernelExec *kernel, KernelExec *replace_kernel, const size_t &tensor_index) { in InsertOutEdge()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/ |
| D | tensorlist.cc | 267 int tensor_index = kOffset + data[1] + 1; in Decode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/debug/ |
| D | print_gpu_kernel.cc | 210 std::string PrintGpuKernelMod::GetString(size_t tensor_index, size_t original_index, void *input_ho… in GetString()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/ |
| D | debug_info_manager.cc | 264 const OpParameter *op_parameter, int tensor_index, in SetQuantStaticInfo()
312 … size_t tensor_index, const mindspore::lite::Tensor *origin_tensor, in AddOriginInfo()
349 … const OpParameter *op_parameter, bool is_input, size_t tensor_index, in AddComparedInfo()
|
| /third_party/mindspore/mindspore-src/source/tests/st/fallback/ |
| D | test_graph_fallback_runtime.py | 615 def _check_dim_shape_valid(data, tensor_index): argument
646 def _check_dim_shape_valid(data, tensor_index): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/abstract/ops/ |
| D | prim_maths.cc | 156 const size_t tensor_index = 2; in InferImplLinSpace() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/trt_pass/ |
| D | trt_converter_context.cc | 161 for (size_t tensor_index = 0; tensor_index < nv_tensors.size(); ++tensor_index) { in StoreLayerOutput() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/ |
| D | pack_weight.cc | 136 …:ReplaceOriginTensorData(const void *model_buf, std::vector<Tensor *> *tensors, int tensor_index) { in ReplaceOriginTensorData()
|
| D | pack_weight_manager.cc | 161 auto tensor_index = node->input_indices_[j]; in StoreOriginTensorData() local
|
| D | lite_mindrt.cc | 427 void LiteOpActor::AddResultIndex(size_t index, size_t tensor_index) { in AddResultIndex()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/common/ |
| D | node_util.cc | 231 size_t GetTensorInputIndexInCNode(const uint32_t &tensor_index, const schema::CNodeT &cnode) { in GetTensorInputIndexInCNode()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/providers/nnie/src/ |
| D | nnie_common.cc | 907 …nieCommGetOutputData(NnieRunCfg *nnie_run_cfg, float *data, HI_U32 output_size, int tensor_index) { in NnieCommGetOutputData()
925 int NnieCommFillData(NnieRunCfg *nnie_run_cfg, void *data, HI_U32 input_size, int tensor_index) { in NnieCommFillData()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/data/ |
| D | parse_example_op.cc | 1089 for (auto tensor_index = start; tensor_index < end; ++tensor_index) { in ParallelParseExample() local
1138 const int32_t column_index, const size_t tensor_index, in ParseSerializedKnownShapeColumn()
1328 const size_t tensor_index) { in ParseSerializedExample()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | tensor_layout.cc | 172 size_t tensor_index = device_arrangement.size() - 1 - static_cast<size_t>(tensor_map[i][j]); in InitFromExtendVector() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/ |
| D | mem_swap_scheduler.cc | 129 for (auto &tensor_index : tensor_indexes) { in GetActionTensors() local
|