2021-02-01 13:31:45 +01:00
|
|
|
/* ******************************************************************************
|
|
|
|
*
|
2019-06-06 14:21:15 +02:00
|
|
|
*
|
|
|
|
* This program and the accompanying materials are made available under the
|
|
|
|
* terms of the Apache License, Version 2.0 which is available at
|
|
|
|
* https://www.apache.org/licenses/LICENSE-2.0.
|
|
|
|
*
|
2021-02-01 13:31:45 +01:00
|
|
|
* See the NOTICE file distributed with this work for additional
|
|
|
|
* information regarding copyright ownership.
|
2019-06-06 14:21:15 +02:00
|
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
|
|
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
|
|
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
|
|
* License for the specific language governing permissions and limitations
|
|
|
|
* under the License.
|
|
|
|
*
|
|
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
|
|
******************************************************************************/
|
|
|
|
|
|
|
|
//
|
|
|
|
// @author raver119@gmail.com
|
|
|
|
//
|
|
|
|
|
|
|
|
|
|
|
|
#ifndef LIBND4J_OP_ENUMS_H
|
|
|
|
#define LIBND4J_OP_ENUMS_H
|
|
|
|
|
|
|
|
#include <loops/legacy_ops.h>
|
2020-03-02 10:49:41 +01:00
|
|
|
#include <system/type_boilerplate.h>
|
|
|
|
#include <system/enum_boilerplate.h>
|
2019-06-06 14:21:15 +02:00
|
|
|
|
2020-03-02 10:49:41 +01:00
|
|
|
namespace sd {
|
2019-06-06 14:21:15 +02:00
|
|
|
namespace random {
|
|
|
|
enum Ops {
|
|
|
|
BUILD_ENUMERATION(RANDOM_OPS)
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
namespace transform {
|
|
|
|
enum FloatOps {
|
|
|
|
BUILD_ENUMERATION(TRANSFORM_FLOAT_OPS)
|
|
|
|
};
|
|
|
|
|
|
|
|
enum SameOps {
|
|
|
|
BUILD_ENUMERATION(TRANSFORM_SAME_OPS)
|
|
|
|
};
|
|
|
|
|
|
|
|
enum BoolOps {
|
|
|
|
BUILD_ENUMERATION(TRANSFORM_BOOL_OPS)
|
|
|
|
};
|
|
|
|
|
|
|
|
enum AnyOps {
|
|
|
|
BUILD_ENUMERATION(TRANSFORM_ANY_OPS)
|
|
|
|
};
|
|
|
|
|
|
|
|
enum StrictOps {
|
|
|
|
BUILD_ENUMERATION(TRANSFORM_STRICT_OPS)
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
namespace pairwise {
|
|
|
|
enum Ops {
|
|
|
|
BUILD_ENUMERATION(PAIRWISE_TRANSFORM_OPS)
|
|
|
|
};
|
|
|
|
|
|
|
|
enum BoolOps {
|
|
|
|
BUILD_ENUMERATION(PAIRWISE_BOOL_OPS)
|
|
|
|
};
|
2019-08-30 09:12:40 +02:00
|
|
|
|
|
|
|
enum IntOps {
|
|
|
|
BUILD_ENUMERATION(PAIRWISE_INT_OPS)
|
|
|
|
};
|
2019-06-06 14:21:15 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
namespace scalar {
|
|
|
|
enum Ops {
|
|
|
|
BUILD_ENUMERATION(SCALAR_OPS)
|
|
|
|
};
|
|
|
|
|
|
|
|
enum BoolOps {
|
|
|
|
BUILD_ENUMERATION(SCALAR_BOOL_OPS)
|
|
|
|
};
|
2019-08-30 09:12:40 +02:00
|
|
|
|
|
|
|
enum IntOps {
|
|
|
|
BUILD_ENUMERATION(SCALAR_INT_OPS)
|
|
|
|
};
|
2019-06-06 14:21:15 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
namespace reduce {
|
|
|
|
enum FloatOps {
|
|
|
|
BUILD_ENUMERATION(REDUCE_FLOAT_OPS)
|
|
|
|
};
|
|
|
|
|
|
|
|
enum SameOps {
|
|
|
|
BUILD_ENUMERATION(REDUCE_SAME_OPS)
|
|
|
|
};
|
|
|
|
|
|
|
|
enum BoolOps {
|
|
|
|
BUILD_ENUMERATION(REDUCE_BOOL_OPS)
|
|
|
|
};
|
|
|
|
|
|
|
|
enum LongOps {
|
|
|
|
BUILD_ENUMERATION(REDUCE_LONG_OPS)
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
namespace reduce3 {
|
|
|
|
enum Ops {
|
|
|
|
BUILD_ENUMERATION(REDUCE3_OPS)
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
namespace indexreduce {
|
|
|
|
enum Ops {
|
|
|
|
BUILD_ENUMERATION(INDEX_REDUCE_OPS)
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
namespace broadcast {
|
|
|
|
enum Ops {
|
|
|
|
BUILD_ENUMERATION(BROADCAST_OPS)
|
|
|
|
};
|
|
|
|
|
|
|
|
enum BoolOps {
|
|
|
|
BUILD_ENUMERATION(BROADCAST_BOOL_OPS)
|
|
|
|
};
|
2019-08-30 09:12:40 +02:00
|
|
|
|
|
|
|
enum IntOps {
|
|
|
|
BUILD_ENUMERATION(BROADCAST_INT_OPS)
|
|
|
|
};
|
2019-06-06 14:21:15 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
namespace variance {
|
|
|
|
enum Ops {
|
|
|
|
BUILD_ENUMERATION(SUMMARY_STATS_OPS)
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
namespace logic {
|
|
|
|
enum Ops {
|
|
|
|
While = 0,
|
|
|
|
Scope = 10,
|
|
|
|
Conditional = 20,
|
|
|
|
Switch = 30,
|
|
|
|
Return = 40,
|
|
|
|
Expose = 50,
|
|
|
|
Merge = 60,
|
|
|
|
LoopCond = 70,
|
|
|
|
NextIteration = 80,
|
|
|
|
Exit = 90,
|
|
|
|
Enter = 100,
|
|
|
|
};
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#endif
|