Searched refs:InvertPermutation (Results 1 – 23 of 23) sorted by relevance
/third_party/mindspore/mindspore/core/ops/ |
D | invert_permutation.h | 30 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>;
|
D | invert_permutation.cc | 22 REGISTER_PRIMITIVE_C(kNameInvertPermutation, InvertPermutation);
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
D | invert_permutation_fp32.h | 22 void InvertPermutation(const int *input, int *output, int num);
|
D | invert_permutation_fp32.c | 19 void InvertPermutation(const int *input, int *output, int num) { in InvertPermutation() function
|
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/ |
D | tf_invert_permutation_parser.cc | 29 auto prim = std::make_unique<ops::InvertPermutation>(); in Parse()
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
D | invert_permutation_infer.c | 43 REG_INFER(InvertPermutation, PrimType_InvertPermutation, InvertPermutationInferShape)
|
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
D | invert_permutation_fp32.cc | 61 InvertPermutation(input_ptr, output_ptr, in_tensors_[0]->ElementsNum()); in Run()
|
/third_party/mindspore/mindspore/ops/_grad/ |
D | grad_other_ops.py | 36 @bprop_getters.register(P.InvertPermutation)
|
D | grad_array_ops.py | 41 invert_permutation = P.InvertPermutation()
|
/third_party/mindspore/mindspore/lite/src/ops/ |
D | ops_func_declare.h | 441 FUNC_MSOP2SCHEMAOP_DECLARE(InvertPermutation)
|
D | ops_def.cc | 193 OP_TYPE(InvertPermutation) 1052 OP_SCHEMA_DEF(InvertPermutation) 1053 OP_SCHEMA_DEF_END(InvertPermutation)
|
D | ops_utils.cc | 333 auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::InvertPermutation>>(node); in InvertPermutationPrimitiveCreator()
|
/third_party/mindspore/mindspore/ops/operations/ |
D | __init__.py | 25 … Fill, Ones, Zeros, GatherNd, GatherV2, Gather, SparseGatherV2, InvertPermutation,
|
D | array_ops.py | 1551 class InvertPermutation(PrimitiveWithInfer): class
|
/third_party/mindspore/tests/vm_impl/ |
D | array_ops_vm_impl.py | 160 @vm_impl_getters.register(P.InvertPermutation)
|
/third_party/mindspore/mindspore/lite/schema/ |
D | ops.fbs | 194 InvertPermutation, 1052 table InvertPermutation {
|
D | model_generated.h | 986 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…
|
D | ops_generated.h | 530 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/ |
D | test_array_ops.py | 112 invert_permutation = P.InvertPermutation()
|
D | test_ops.py | 1464 'block': P.InvertPermutation(),
|
/third_party/mindspore/tests/ut/python/ir/ |
D | test_row_tensor.py | 56 invert_permutation = P.InvertPermutation()
|
/third_party/mindspore/mindspore/lite/schema/inner/ |
D | model_generated.h | 1135 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…
|
D | ops_generated.h | 702 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 …]
|