Home
last modified time | relevance | path

Searched refs:tensor_index (Results 1 – 15 of 15) sorted by relevance

/third_party/mindspore/mindspore/ops/composite/multitype_ops/
Dgetitem_impl.py107 def _tuple_getitem_by_tensor(data, tensor_index): argument
118 tensor_index = F.select(tensor_index >= 0, tensor_index, tensor_index + len(data))
119 return _tuple_get_item_tensor(data, tensor_index)
198 def _tensor_getitem_by_tensor(data, tensor_index): argument
209 return compile_utils.tensor_index_by_tensor(data, tensor_index)
D_compile_utils.py325 def tensor_index_by_tensor(data, tensor_index): argument
329 invalid = const_utils.check_type_invalid(F.dtype(tensor_index), mstype.int_type)
332 "The tensor index must be int type, but got {}.", F.dtype(tensor_index))
334 return F.gather(data, tensor_index, 0)
345 tensor_index = const_utils.sequence_to_index(list_index, data_shape[0])
346 if tensor_index is False:
348 return F.gather(data, tensor_index, 0)
422 tensor_index = F.scalar_to_tensor(int_index, mstype.int64)
423 tuple_index_new += (tensor_index,)
424 tensor_indexes.append(tensor_index)
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/control/
Dtensorlist_getitem_infer.c43 TensorC *tensor_index = &input0->tensors_[index]; in TensorListGetItemInferShape() local
44 NNACL_CHECK_NULL_RETURN_ERR(tensor_index); in TensorListGetItemInferShape()
46 if (tensor_index->data_type_ != kTypeUnknown) { in TensorListGetItemInferShape()
47 output->data_type_ = tensor_index->data_type_; in TensorListGetItemInferShape()
57 if (tensor_index->data_type_ != kTypeUnknown) { in TensorListGetItemInferShape()
58 … ShapeSet(output->shape_, &(output->shape_size_), tensor_index->shape_, tensor_index->shape_size_); in TensorListGetItemInferShape()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/
Dtrt_converter_context.cc159 for (size_t tensor_index = 0; tensor_index < nv_tensors.size(); ++tensor_index) { in StoreLayerOutput() local
160 if (nv_tensors[tensor_index] != nullptr) { in StoreLayerOutput()
161 const nvinfer1::Dims &dim = nv_tensors[tensor_index]->getDimensions(); in StoreLayerOutput()
163 output_map_[node][tensor_index] = LayerInput(nv_tensors[tensor_index], shape); in StoreLayerOutput()
166 oss << node->fullname_with_scope() << ", output: " << tensor_index << ": [ "; in StoreLayerOutput()
/third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/
Dtflite_node_parser.h50 …STATUS GetTfliteData(const int32_t tensor_index, const std::vector<std::unique_ptr<tflite::TensorT… in GetTfliteData() argument
54 CHECK_LESS_RETURN(tflite_tensors.size(), static_cast<size_t>(tensor_index + 1)); in GetTfliteData()
55 const auto &tensor = tflite_tensors[tensor_index]; in GetTfliteData()
132 STATUS TransTfliteDataToVec2D(const int32_t tensor_index, in TransTfliteDataToVec2D() argument
137 CHECK_LESS_RETURN(tflite_tensors.size(), static_cast<size_t>(tensor_index + 1)); in TransTfliteDataToVec2D()
138 const auto &tensor = tflite_tensors[tensor_index]; in TransTfliteDataToVec2D()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/debug/
Dprint_gpu_kernel.h217 std::string GetString(size_t tensor_index, size_t original_index, void *input_host_data) { in GetString() argument
219 size_t size_in_byte = std::get<0>(input_info_[tensor_index]); in GetString()
220 TypeId type_id = std::get<1>(input_info_[tensor_index]); in GetString()
221 (void)std::transform(input_shape_[tensor_index].begin(), input_shape_[tensor_index].end(), in GetString()
/third_party/mindspore/mindspore/lite/src/
Dtensorlist.cc277 int tensor_index = 2 + data[1] + 1; in Decode() local
279 int tensor_dims_size = data[tensor_index++]; in Decode()
282 shape[j] = data[tensor_index++]; in Decode()
Dlite_session.h93 …int ConvertTensorsData(const lite::Model *model, size_t tensor_index, const schema::Tensor *src_te…
Dlite_session.cc135 int LiteSession::ConvertTensorsData(const lite::Model *model, size_t tensor_index, const schema::Te… in ConvertTensorsData() argument
/third_party/mindspore/mindspore/lite/tools/converter/legacy_optimizer/graph/
Dtensor_quant_pass.cc109 STATUS ComputeQuantTensorPerChannel(TensorT *tensor, const int &tensor_index, const schema::MetaGra… in ComputeQuantTensorPerChannel() argument
112 auto used_nodes_idx = GetLinkedPostIdx(graph, tensor_index); in ComputeQuantTensorPerChannel()
119 int input_index = GetTensorInputIndexInCNode(tensor_index, *used_node); in ComputeQuantTensorPerChannel()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/somas/
Dsomas.cc427 size_t tensor_index = 0; in InitSomasOutputAndWorkspaceTensors() local
442 auto output_tensor_index = tensor_index; in InitSomasOutputAndWorkspaceTensors()
443 tensor_index++; in InitSomasOutputAndWorkspaceTensors()
469 auto workspace_tensor_index = tensor_index; in InitSomasOutputAndWorkspaceTensors()
470 tensor_index++; in InitSomasOutputAndWorkspaceTensors()
1470 size_t tensor_index = 0; in DumpNodes() local
1477 oss << "%" << node->input_tensors_[tensor_index]->GetId() << "T" in DumpNodes()
1479 tensor_index++; in DumpNodes()
/third_party/mindspore/mindspore/lite/tools/common/
Dnode_util.cc268 size_t GetTensorInputIndexInCNode(const uint32_t &tensor_index, const schema::CNodeT &cnode) { in GetTensorInputIndexInCNode() argument
271 if (cnode.inputIndex.at(i) == tensor_index) { in GetTensorInputIndexInCNode()
Dnode_util.h87 size_t GetTensorInputIndexInCNode(const uint32_t &tensor_index, const schema::CNodeT &cnode);
/third_party/mindspore/mindspore/core/abstract/
Dprim_maths.cc238 const size_t tensor_index = 2; in InferImplLinSpace() local
239 auto abs_num = args_spec_list[tensor_index]; in InferImplLinSpace()
/third_party/mindspore/third_party/proto/tensorflow/lite/
Dschema_2.4.1.fbs1094 tensor_index:uint;