You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

all_ops.h 2.4 kB

5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
4 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485
  1. /**
  2. * Copyright 2019 Huawei Technologies Co., Ltd
  3. *
  4. * Licensed under the Apache License, Version 2.0 (the "License");
  5. * you may not use this file except in compliance with the License.
  6. * You may obtain a copy of the License at
  7. *
  8. * http://www.apache.org/licenses/LICENSE-2.0
  9. *
  10. * Unless required by applicable law or agreed to in writing, software
  11. * distributed under the License is distributed on an "AS IS" BASIS,
  12. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13. * See the License for the specific language governing permissions and
  14. * limitations under the License.
  15. */
  16. /*!
  17. * \file all_ops.h
  18. * \brief
  19. */
  20. #ifndef OPS_BUILT_IN_OP_PROTO_INC_ALL_OPS_H_
  21. #define OPS_BUILT_IN_OP_PROTO_INC_ALL_OPS_H_
  22. #include "aipp.h"
  23. #include "array_ops.h"
  24. #include "audio_ops.h"
  25. #include "batch_ops.h"
  26. #include "bitwise_ops.h"
  27. #include "boosted_trees_ops.h"
  28. #include "candidate_sampling_ops.h"
  29. #include "control_flow_ops.h"
  30. #include "ctc_ops.h"
  31. #include "data_flow_ops.h"
  32. #include "elewise_calculation_ops.h"
  33. #include "functional_ops.h"
  34. #include "get_data_ops.h"
  35. #include "hcom_ops.h"
  36. #include "hvd_ops.h"
  37. #include "image_ops.h"
  38. #include "internal_ops.h"
  39. #include "linalg_ops.h"
  40. #include "list_ops.h"
  41. #include "logging_ops.h"
  42. #include "lookup_ops.h"
  43. #include "map_ops.h"
  44. #include "math_ops.h"
  45. #include "matrix_calculation_ops.h"
  46. #include "nn_batch_norm_ops.h"
  47. #include "nn_calculation_ops.h"
  48. #include "nn_detect_ops.h"
  49. #include "nn_norm_ops.h"
  50. #include "nn_ops.h"
  51. #include "nn_pooling_ops.h"
  52. #include "nn_training_ops.h"
  53. #include "nonlinear_fuc_ops.h"
  54. #include "no_op.h"
  55. #include "npu_loss_scale_ops.h"
  56. #include "outfeed_ops.h"
  57. #include "pad_ops.h"
  58. #include "parsing_ops.h"
  59. #include "quantize_ops.h"
  60. #include "ragged_conversion_ops.h"
  61. #include "random_ops.h"
  62. #include "reduce_ops.h"
  63. #include "resource_variable_ops.h"
  64. #include "rnn.h"
  65. #include "rpn_ops.h"
  66. #include "randomdsa_ops.h"
  67. #include "save_ops.h"
  68. #include "selection_ops.h"
  69. #include "set_ops.h"
  70. #include "sparse_ops.h"
  71. #include "split_combination_ops.h"
  72. #include "stateful_random_ops.h"
  73. #include "stateless_random_ops.h"
  74. #include "state_ops.h"
  75. #include "string_ops.h"
  76. #include "swap_co_ops.h"
  77. #include "transformation_ops.h"
  78. #include "condtake_ops.h"
  79. #include "warp_perspective_ops.h"
  80. #include "vector_search.h"
  81. #include "deep_md.h"
  82. #include "encoding_ops.h"
  83. #endif // OPS_BUILT_IN_OP_PROTO_INC_ALL_OPS_H_

图引擎模块(GE)是MindSpore的一个子模块,其代码由C++实现,位于前端模块ME和底层硬件之间,起到承接作用。图引擎模块以ME下发的图作为输入,然后进行一系列的深度图优化操作,最后输出一张可以在底层硬件上高效运行的图。GE针对昇腾AI处理器的硬件结构特点,做了特定的优化工作,以此来充分发挥出昇腾AI处理器的强大算力。在进行模型训练/推理时,GE会被自动调用而用户并不感知。GE主要由GE API和GE Core两部分组成,详细的架构图如下所示