Home
last modified time | relevance | path

Searched refs:InvertPermutation (Results 1 – 23 of 23) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Dinvert_permutation.h30 class MS_CORE_API InvertPermutation : public PrimitiveC {
33 InvertPermutation() : PrimitiveC(kNameInvertPermutation) {} in InvertPermutation() function
35 ~InvertPermutation() = default;
36 MS_DECLARE_PARENT(InvertPermutation, PrimitiveC);
39 using PrimInvertPermutationPtr = std::shared_ptr<InvertPermutation>;
Dinvert_permutation.cc22 REGISTER_PRIMITIVE_C(kNameInvertPermutation, InvertPermutation);
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dinvert_permutation_fp32.h22 void InvertPermutation(const int *input, int *output, int num);
Dinvert_permutation_fp32.c19 void InvertPermutation(const int *input, int *output, int num) { in InvertPermutation() function
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dtf_invert_permutation_parser.cc29 auto prim = std::make_unique<ops::InvertPermutation>(); in Parse()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/
Dinvert_permutation_infer.c43 REG_INFER(InvertPermutation, PrimType_InvertPermutation, InvertPermutationInferShape)
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dinvert_permutation_fp32.cc61 InvertPermutation(input_ptr, output_ptr, in_tensors_[0]->ElementsNum()); in Run()
/third_party/mindspore/mindspore/ops/_grad/
Dgrad_other_ops.py36 @bprop_getters.register(P.InvertPermutation)
Dgrad_array_ops.py41 invert_permutation = P.InvertPermutation()
/third_party/mindspore/mindspore/lite/src/ops/
Dops_func_declare.h441 FUNC_MSOP2SCHEMAOP_DECLARE(InvertPermutation)
Dops_def.cc193 OP_TYPE(InvertPermutation)
1052 OP_SCHEMA_DEF(InvertPermutation)
1053 OP_SCHEMA_DEF_END(InvertPermutation)
Dops_utils.cc333 auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::InvertPermutation>>(node); in InvertPermutationPrimitiveCreator()
/third_party/mindspore/mindspore/ops/operations/
D__init__.py25 … Fill, Ones, Zeros, GatherNd, GatherV2, Gather, SparseGatherV2, InvertPermutation,
Darray_ops.py1551 class InvertPermutation(PrimitiveWithInfer): class
/third_party/mindspore/tests/vm_impl/
Darray_ops_vm_impl.py160 @vm_impl_getters.register(P.InvertPermutation)
/third_party/mindspore/mindspore/lite/schema/
Dops.fbs194 InvertPermutation,
1052 table InvertPermutation {
Dmodel_generated.h986 const mindspore::schema::InvertPermutation *value_as_InvertPermutation() const { in value_as_InvertPermutation()
987 …mitiveType_InvertPermutation ? static_cast<const mindspore::schema::InvertPermutation *>(value()) … in value_as_InvertPermutation()
1768 …late<> inline const mindspore::schema::InvertPermutation *Primitive::value_as<mindspore::schema::I…
Dops_generated.h530 struct InvertPermutation;
1937 template<> struct PrimitiveTypeTraits<mindspore::schema::InvertPermutation> {
10945 struct InvertPermutation FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table {
10954 typedef InvertPermutation Table;
10961 flatbuffers::Offset<InvertPermutation> Finish() {
10963 auto o = flatbuffers::Offset<InvertPermutation>(end);
10968 inline flatbuffers::Offset<InvertPermutation> CreateInvertPermutation(
13354 auto ptr = reinterpret_cast<const mindspore::schema::InvertPermutation *>(obj);
/third_party/mindspore/tests/ut/python/ops/
Dtest_array_ops.py112 invert_permutation = P.InvertPermutation()
Dtest_ops.py1464 'block': P.InvertPermutation(),
/third_party/mindspore/tests/ut/python/ir/
Dtest_row_tensor.py56 invert_permutation = P.InvertPermutation()
/third_party/mindspore/mindspore/lite/schema/inner/
Dmodel_generated.h1135 const mindspore::schema::InvertPermutation *value_as_InvertPermutation() const { in value_as_InvertPermutation()
1136 …mitiveType_InvertPermutation ? static_cast<const mindspore::schema::InvertPermutation *>(value()) … in value_as_InvertPermutation()
1923 …late<> inline const mindspore::schema::InvertPermutation *Primitive::value_as<mindspore::schema::I…
Dops_generated.h702 struct InvertPermutation;
2537 template<> struct PrimitiveTypeTraits<mindspore::schema::InvertPermutation> {
16668 typedef InvertPermutation TableType;
16671 struct InvertPermutation FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table {
16683 …static flatbuffers::Offset<InvertPermutation> Pack(flatbuffers::FlatBufferBuilder &_fbb, const Inv…
16687 typedef InvertPermutation Table;
16694 flatbuffers::Offset<InvertPermutation> Finish() {
16696 auto o = flatbuffers::Offset<InvertPermutation>(end);
16701 inline flatbuffers::Offset<InvertPermutation> CreateInvertPermutation(
16707 flatbuffers::Offset<InvertPermutation> CreateInvertPermutation(flatbuffers::FlatBufferBuilder &_fbb…
[all …]