| /third_party/mindspore/mindspore/lite/src/ops/compat/v0/ |
| D | strided_slice_compat_v0.cc | 22 int TransferStridedSliceAttr(LiteGraph::Node *node, std::vector<schema::Tensor *> *dst_tensors, in TransferStridedSliceAttr() argument
24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferStridedSliceAttr()
28 dst_tensors->clear(); in TransferStridedSliceAttr()
47 dst_tensors->push_back(dst_begins_tensor); in TransferStridedSliceAttr()
57 dst_tensors->push_back(dst_ends_tensor); in TransferStridedSliceAttr()
68 dst_tensors->push_back(dst_strides_tensor); in TransferStridedSliceAttr()
76 …if (std::any_of(dst_tensors->begin(), dst_tensors->end(), [](schema::Tensor *tensor) { return tens… in TransferStridedSliceAttr()
|
| D | slice_compat_v0.cc | 22 int TransferSliceAttr(LiteGraph::Node *node, std::vector<schema::Tensor *> *dst_tensors, in TransferSliceAttr() argument
24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferSliceAttr()
32 MS_ASSERT(dst_tensors->size() == 0); in TransferSliceAttr()
53 dst_tensors->push_back(begin_tensor); in TransferSliceAttr()
67 dst_tensors->push_back(size_tensor); in TransferSliceAttr()
|
| D | nchw2nhwc_compat_v0.cc | 22 int TransferNchw2NhwcAttr(LiteGraph::Node *node, std::vector<schema::Tensor *> *dst_tensors, in TransferNchw2NhwcAttr() argument
24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferNchw2NhwcAttr()
32 dst_tensors->clear(); in TransferNchw2NhwcAttr()
39 dst_tensors->push_back(dst_shape_tensor); in TransferNchw2NhwcAttr()
|
| D | nhwc2nchw_compat_v0.cc | 22 int TransferNhwc2NchwAttr(LiteGraph::Node *node, std::vector<schema::Tensor *> *dst_tensors, in TransferNhwc2NchwAttr() argument
24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferNhwc2NchwAttr()
32 dst_tensors->clear(); in TransferNhwc2NchwAttr()
39 dst_tensors->push_back(dst_shape_tensor); in TransferNhwc2NchwAttr()
|
| D | gather_compat_v0.cc | 22 int TransferGatherAttr(LiteGraph::Node *node, std::vector<schema::Tensor *> *dst_tensors, in TransferGatherAttr() argument
24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferGatherAttr()
28 dst_tensors->clear(); in TransferGatherAttr()
42 dst_tensors->push_back(axis_tensor); in TransferGatherAttr()
|
| D | cast_compat_v0.cc | 22 int TransferCastAttr(LiteGraph::Node *node, std::vector<schema::Tensor *> *dst_tensors, in TransferCastAttr() argument
24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferCastAttr()
28 dst_tensors->clear(); in TransferCastAttr()
42 dst_tensors->push_back(dst_type_tensor); in TransferCastAttr()
|
| D | power_compat_v0.cc | 22 int TransferPowerToAttr(LiteGraph::Node *node, std::vector<schema::Tensor *> *dst_tensors, in TransferPowerToAttr() argument
24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferPowerToAttr()
32 dst_tensors->clear(); in TransferPowerToAttr()
46 dst_tensors->push_back(power_tensor); in TransferPowerToAttr()
|
| D | expand_dims_compat_v0.cc | 23 int TransferExpandDimsAttr(LiteGraph::Node *node, std::vector<schema::Tensor *> *dst_tensors, in TransferExpandDimsAttr() argument
25 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferExpandDimsAttr()
30 MS_CHECK_TRUE_RET(dst_tensors->size() == 0, RET_ERROR); in TransferExpandDimsAttr()
45 dst_tensors->push_back(dim_tensor); in TransferExpandDimsAttr()
|
| D | topk_compat_v0.cc | 22 int TransferTopkAttr(LiteGraph::Node *node, std::vector<schema::Tensor *> *dst_tensors, in TransferTopkAttr() argument
24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferTopkAttr()
32 MS_ASSERT(dst_tensors->size() == 0); in TransferTopkAttr()
46 dst_tensors->push_back(k_tensor); in TransferTopkAttr()
|
| D | transpose_compat_v0.cc | 22 int TransferTransposeAttr(LiteGraph::Node *node, std::vector<schema::Tensor *> *dst_tensors, in TransferTransposeAttr() argument
24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferTransposeAttr()
32 dst_tensors->clear(); in TransferTransposeAttr()
51 dst_tensors->push_back(dst_shape_tensor); in TransferTransposeAttr()
|
| D | reduce_compat_v0.cc | 22 int TransferReduceToAttr(LiteGraph::Node *node, std::vector<schema::Tensor *> *dst_tensors, in TransferReduceToAttr() argument
24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferReduceToAttr()
32 dst_tensors->clear(); in TransferReduceToAttr()
51 dst_tensors->push_back(axes_tensor); in TransferReduceToAttr()
|
| D | reshape_compat_v0.cc | 23 int TransferReshapeAttr(LiteGraph::Node *node, std::vector<schema::Tensor *> *dst_tensors, in TransferReshapeAttr() argument
25 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferReshapeAttr()
33 dst_tensors->clear(); in TransferReshapeAttr()
52 dst_tensors->push_back(dst_shape_tensor); in TransferReshapeAttr()
|
| D | broadcast_to_compat_v0.cc | 22 int TransferBroadcastToAttr(LiteGraph::Node *node, std::vector<schema::Tensor *> *dst_tensors, in TransferBroadcastToAttr() argument
24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferBroadcastToAttr()
32 dst_tensors->clear(); in TransferBroadcastToAttr()
51 dst_tensors->push_back(dst_shape_tensor); in TransferBroadcastToAttr()
|
| D | fill_compat_v0.cc | 22 int TransferFillToAttr(LiteGraph::Node *node, std::vector<schema::Tensor *> *dst_tensors, in TransferFillToAttr() argument
24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferFillToAttr()
32 dst_tensors->clear(); in TransferFillToAttr()
51 dst_tensors->push_back(dims_tensor); in TransferFillToAttr()
|
| D | pad_compat_v0.cc | 22 int TransferPadAttr(LiteGraph::Node *node, std::vector<schema::Tensor *> *dst_tensors, in TransferPadAttr() argument
24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferPadAttr()
28 dst_tensors->clear(); in TransferPadAttr()
51 dst_tensors->push_back(paddings_tensor); in TransferPadAttr()
|
| D | tile_compat_v0.cc | 23 int TransferTileToAttr(LiteGraph::Node *node, std::vector<schema::Tensor *> *dst_tensors, in TransferTileToAttr() argument
25 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferTileToAttr()
33 dst_tensors->clear(); in TransferTileToAttr()
52 dst_tensors->push_back(multiples_tensor); in TransferTileToAttr()
|
| D | permute_compat_v0.cc | 22 int TransferPermuteAttr(LiteGraph::Node *node, std::vector<schema::Tensor *> *dst_tensors, in TransferPermuteAttr() argument
24 …if (node == nullptr || node->primitive_ == nullptr || dst_tensors == nullptr || tensor_bufs == nul… in TransferPermuteAttr()
32 dst_tensors->clear(); in TransferPermuteAttr()
54 dst_tensors->push_back(dst_shape_tensor); in TransferPermuteAttr()
|
| /third_party/mindspore/mindspore/lite/src/ |
| D | lite_model.cc | 73 std::vector<schema::Tensor *> dst_tensors; in ConvertAttrToTensors() local
74 int status = ConvertAttrs(this->graph_.all_nodes_[index], &dst_tensors); in ConvertAttrToTensors()
79 if (dst_tensors.empty()) { in ConvertAttrToTensors()
89 for (auto tensor : dst_tensors) { in ConvertAttrToTensors()
|