Home
last modified time | relevance | path

Searched defs:IndexSelect (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/symbol_ops_impl/
Dindex_select.cc22 class MS_CORE_API IndexSelect : public InferShapeOp { class
25 IndexSelect(const SymbolPtr &data, const SymbolPtr &idx) : InferShapeOp({data, idx}) {} in IndexSelect() function in mindspore::symshape::ops::IndexSelect
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/auto_generate/
Dgen_lite_ops.h1267 IndexSelect() : BaseOperator(kNameIndexSelect) {} in IndexSelect() function