Home
last modified time | relevance | path

Searched refs:FloorMod (Results 1 – 25 of 37) sorted by relevance

12

/third_party/mindspore/mindspore/core/ops/
Dfloor_mod.h28 class MS_CORE_API FloorMod : public PrimitiveC {
31 FloorMod() : PrimitiveC(kNameFloorMod) { InitIOName({"x", "y"}, {"output"}); } in FloorMod() function
33 ~FloorMod() = default;
34 MS_DECLARE_PARENT(FloorMod, PrimitiveC);
Dfloor_mod.cc20 REGISTER_PRIMITIVE_C(kNameFloorMod, FloorMod);
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Darithmetic_cpu_kernel.h48 void FloorMod(const T *input1, const T *input2, T *out) const;
102FloorMod, KernelAttr().AddInputAttr(kNumberTypeInt64).AddInputAttr(kNumberTypeInt64).AddOutputAttr…
105FloorMod, KernelAttr().AddInputAttr(kNumberTypeInt32).AddInputAttr(kNumberTypeInt32).AddOutputAttr…
108 FloorMod,
112 FloorMod,
Darithmetic_cpu_kernel.cc290 void ArithmeticCPUKernel<T>::FloorMod(const T *input1, const T *input2, T *out) const { in FloorMod() function in mindspore::kernel::ArithmeticCPUKernel
404 {prim::kPrimFloorMod->name(), &ArithmeticCPUKernel<T>::FloorMod}, in InitComputeFunc()
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/
Delewise_calculation_ops_declare.h245 DECLARE_OP_ADAPTER(FloorMod)
246 DECLARE_OP_USE_OUTPUT(FloorMod)
Delewise_calculation_ops_declare.cc184 INPUT_MAP(FloorMod) = {{1, INPUT_DESC(x1)}, {2, INPUT_DESC(x2)}};
185 ATTR_MAP(FloorMod) = EMPTY_ATTR_MAP;
186 OUTPUT_MAP(FloorMod) = {{0, OUTPUT_DESC(y)}};
187 REG_ADPT_DESC(FloorMod, kNameFloorMod, ADPT_DESC(FloorMod))
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/
Dbroadcast_gpu_kernel.cc60 FloorMod,
133 FloorMod,
215 FloorMod,
287FloorMod, KernelAttr().AddInputAttr(kNumberTypeInt32).AddInputAttr(kNumberTypeInt32).AddOutputAttr…
349FloorMod, KernelAttr().AddInputAttr(kNumberTypeInt64).AddInputAttr(kNumberTypeInt64).AddOutputAttr…
/third_party/mindspore/tests/st/ops/gpu/
Dtest_broadcast_op.py86 output_ms = P.FloorMod()(Tensor(x1_np), Tensor(x2_np))
149 output_ms = P.FloorMod()(Tensor(x1_np), Tensor(x2_np))
220 output_ms = P.FloorMod()(Tensor(x1_np), Tensor(x2_np))
291 output_ms = P.FloorMod()(Tensor(x1_np), Tensor(x2_np))
354 output_ms = P.FloorMod()(Tensor(x1_np), Tensor(x2_np))
/third_party/mindspore/mindspore/lite/src/delegate/npu/op/
Darithmetic_npu.cc71 op_ = CreateOperator<hiai::op::FloorMod>(name_); in Init()
161 SetInputs<hiai::op::FloorMod>(npu_inputs, op_); in SetNPUInputs()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/
Darithmetic_infer.c113 REG_INFER(FloorMod, PrimType_FloorMod, ArithmeticInferShape)
/third_party/mindspore/mindspore/lite/test/config/
Dut_arm64.cfg68 TestOpenCL_Arithmetic.FloorMod
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Darithmetic_tests.cc145 TEST_F(TestOpenCL_Arithmetic, FloorMod) { in TEST_F() argument
/third_party/mindspore/mindspore/lite/schema/
Dmodel_v0.fbs154 FloorMod,
Dops.fbs85 FloorMod,
567 table FloorMod {
Dops_v0.fbs874 table FloorMod {
Dmodel_generated.h659 const mindspore::schema::FloorMod *value_as_FloorMod() const { in value_as_FloorMod()
660 …::schema::PrimitiveType_FloorMod ? static_cast<const mindspore::schema::FloorMod *>(value()) : nul… in value_as_FloorMod()
1332 template<> inline const mindspore::schema::FloorMod *Primitive::value_as<mindspore::schema::FloorMo…
Dmodel_v0_generated.h1093 template<> struct PrimitiveTypeTraits<mindspore::schema::v0::FloorMod> {
2184 const mindspore::schema::v0::FloorMod *value_as_FloorMod() const {
2185 …::v0::PrimitiveType_FloorMod ? static_cast<const mindspore::schema::v0::FloorMod *>(value()) : nul…
2901 …emplate<> inline const mindspore::schema::v0::FloorMod *Primitive::value_as<mindspore::schema::v0:…
4139 auto ptr = reinterpret_cast<const mindspore::schema::v0::FloorMod *>(obj);
/third_party/mindspore/mindspore/lite/src/ops/
Dops_func_declare.h325 FUNC_MSOP2SCHEMAOP_DECLARE(FloorMod)
Dops_def.cc84 OP_TYPE(FloorMod)
567 OP_SCHEMA_DEF(FloorMod)
568 OP_SCHEMA_DEF_END(FloorMod)
/third_party/mindspore/tests/ut/python/ops/
Dtest_math_ops_check.py349 'block': (P.FloorMod(), {'exception': ValueError, 'error_keywords': ['FloorMod']}),
616 'block': P.FloorMod(),
/third_party/mindspore/mindspore/ops/operations/
D__init__.py53 … Cos, Div, DivNoNan, Equal, EqualCount, Exp, Expm1, Erf, Erfc, Floor, FloorDiv, FloorMod, Ceil,
/third_party/mindspore/mindspore/ops/
Dfunctional.py72 tensor_mod = P.FloorMod()
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dtf_arithmetic_parser.cc320 auto prim = std::make_unique<ops::FloorMod>(); in Parse()
/third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/
Dtflite_arithmetic_parser.cc139 auto prim = std::make_unique<ops::FloorMod>(); in Parse()
/third_party/mindspore/tests/st/ops/cpu/
Dtest_arithmetic_op.py66 self.floor_mod = P.FloorMod()

12