Searched defs:tensor_index (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_node_parser.h | 50 …STATUS GetTfliteData(const int32_t tensor_index, const std::vector<std::unique_ptr<tflite::TensorT… in GetTfliteData()
132 STATUS TransTfliteDataToVec2D(const int32_t tensor_index, in TransTfliteDataToVec2D()
|
| /third_party/mindspore/mindspore/ops/composite/multitype_ops/ |
| D | getitem_impl.py | 107 def _tuple_getitem_by_tensor(data, tensor_index): argument
198 def _tensor_getitem_by_tensor(data, tensor_index): argument
|
| D | _compile_utils.py | 325 def tensor_index_by_tensor(data, tensor_index): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/control/ |
| D | tensorlist_getitem_infer.c | 43 TensorC *tensor_index = &input0->tensors_[index]; in TensorListGetItemInferShape() local
|
| /third_party/mindspore/mindspore/lite/src/ |
| D | tensorlist.cc | 277 int tensor_index = 2 + data[1] + 1; in Decode() local
|
| D | lite_session.cc | 135 int LiteSession::ConvertTensorsData(const lite::Model *model, size_t tensor_index, const schema::Te… in ConvertTensorsData()
|
| /third_party/mindspore/mindspore/lite/tools/converter/legacy_optimizer/graph/ |
| D | tensor_quant_pass.cc | 109 STATUS ComputeQuantTensorPerChannel(TensorT *tensor, const int &tensor_index, const schema::MetaGra… in ComputeQuantTensorPerChannel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/debug/ |
| D | print_gpu_kernel.h | 217 std::string GetString(size_t tensor_index, size_t original_index, void *input_host_data) { in GetString()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/ |
| D | trt_converter_context.cc | 159 for (size_t tensor_index = 0; tensor_index < nv_tensors.size(); ++tensor_index) { in StoreLayerOutput() local
|
| /third_party/mindspore/mindspore/lite/tools/common/ |
| D | node_util.cc | 268 size_t GetTensorInputIndexInCNode(const uint32_t &tensor_index, const schema::CNodeT &cnode) { in GetTensorInputIndexInCNode()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_maths.cc | 238 const size_t tensor_index = 2; in InferImplLinSpace() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/somas/ |
| D | somas.cc | 427 size_t tensor_index = 0; in InitSomasOutputAndWorkspaceTensors() local
1470 size_t tensor_index = 0; in DumpNodes() local
|