| /third_party/mindspore/mindspore/core/ops/ |
| D | floor_div.h | 28 class MS_CORE_API FloorDiv : public PrimitiveC {
31 FloorDiv() : PrimitiveC(kNameFloorDiv) { InitIOName({"x", "y"}, {"output"}); } in FloorDiv() function
33 ~FloorDiv() = default;
34 MS_DECLARE_PARENT(FloorDiv, PrimitiveC);
|
| D | floor_div.cc | 20 REGISTER_PRIMITIVE_C(kNameFloorDiv, FloorDiv);
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | arithmetic_cpu_kernel.h | 46 void FloorDiv(const T *input1, const T *input2, T *out) const;
83 …FloorDiv, KernelAttr().AddInputAttr(kNumberTypeInt64).AddInputAttr(kNumberTypeInt64).AddOutputAttr…
86 …FloorDiv, KernelAttr().AddInputAttr(kNumberTypeInt32).AddInputAttr(kNumberTypeInt32).AddOutputAttr…
89 FloorDiv,
|
| D | arithmetic_cpu_kernel.cc | 239 void ArithmeticCPUKernel<T>::FloorDiv(const T *input1, const T *input2, T *out) const { in FloorDiv() function in mindspore::kernel::ArithmeticCPUKernel
406 {prim::kPrimFloorDiv->name(), &ArithmeticCPUKernel<T>::FloorDiv}, in InitComputeFunc()
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_auto_parallel_arithmetic.py | 65 self.floordiv = P.FloorDiv()
94 self.floordiv = P.FloorDiv()
123 self.floordiv = P.FloorDiv()
152 self.floordiv = P.FloorDiv()
|
| D | test_auto_parallel_multi_graph.py | 44 self.floor_div = P.FloorDiv()
|
| D | test_mul_div_bn.py | 46 self.floor_div = P.FloorDiv()
|
| D | test_arithmetic.py | 548 self.floordiv = P.FloorDiv().shard(strategy2)
572 self.floordiv = P.FloorDiv().shard(strategy2)
596 self.floordiv = P.FloorDiv().shard(strategy2)
|
| /third_party/mindspore/tests/syntax/simple_expression/ |
| D | test_math_ops.py | 492 class FloorDiv(nn.Cell): class
494 super(FloorDiv, self).__init__()
495 self.floordiv = P.FloorDiv()
506 floordiv_net = FloorDiv()
517 floordiv_net = FloorDiv()
528 floordiv_net = FloorDiv()
539 floordiv_net = FloorDiv()
550 floordiv_net = FloorDiv()
561 floordiv_net = FloorDiv()
572 floordiv_net = FloorDiv()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | elewise_calculation_ops_declare.h | 242 DECLARE_OP_ADAPTER(FloorDiv)
243 DECLARE_OP_USE_OUTPUT(FloorDiv)
|
| D | elewise_calculation_ops_declare.cc | 178 INPUT_MAP(FloorDiv) = {{1, INPUT_DESC(x1)}, {2, INPUT_DESC(x2)}};
179 ATTR_MAP(FloorDiv) = EMPTY_ATTR_MAP;
180 OUTPUT_MAP(FloorDiv) = {{0, OUTPUT_DESC(y)}};
181 REG_ADPT_DESC(FloorDiv, kNameFloorDiv, ADPT_DESC(FloorDiv))
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/ |
| D | broadcast_gpu_kernel.cc | 115 FloorDiv,
197 FloorDiv,
269 …FloorDiv, KernelAttr().AddInputAttr(kNumberTypeInt32).AddInputAttr(kNumberTypeInt32).AddOutputAttr…
334 …FloorDiv, KernelAttr().AddInputAttr(kNumberTypeInt64).AddInputAttr(kNumberTypeInt64).AddOutputAttr…
|
| /third_party/mindspore/mindspore/lite/src/delegate/npu/op/ |
| D | arithmetic_npu.cc | 74 op_ = CreateOperator<hiai::op::FloorDiv>(name_); in Init()
164 op_ = CreateOperator<hiai::op::FloorDiv>(name_); in SetNPUInputs()
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_ops_reid.py | 71 'block': P.FloorDiv(),
|
| D | test_math_ops_check.py | 336 'block': (P.FloorDiv(), {'exception': ValueError, 'error_keywords': ['FloorDiv']}),
611 'block': P.FloorDiv(),
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | arithmetic_infer.c | 112 REG_INFER(FloorDiv, PrimType_FloorDiv, ArithmeticInferShape)
|
| /third_party/mindspore/mindspore/_extends/parse/ |
| D | resources.py | 44 ast.FloorDiv: (trope_ns, 'floordiv'),
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_floordiv_op.py | 27 self.floordiv = P.FloorDiv()
|
| /third_party/mindspore/mindspore/lite/schema/ |
| D | model_v0.fbs | 153 FloorDiv,
|
| D | ops.fbs | 84 FloorDiv,
564 table FloorDiv {
|
| /third_party/mindspore/mindspore/lite/src/ops/ |
| D | ops_func_declare.h | 324 FUNC_MSOP2SCHEMAOP_DECLARE(FloorDiv)
|
| D | ops_def.cc | 83 OP_TYPE(FloorDiv)
564 OP_SCHEMA_DEF(FloorDiv)
565 OP_SCHEMA_DEF_END(FloorDiv)
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | __init__.py | 53 … Cos, Div, DivNoNan, Equal, EqualCount, Exp, Expm1, Erf, Erfc, Floor, FloorDiv, FloorMod, Ceil,
|
| /third_party/mindspore/mindspore/nn/ |
| D | learning_rate_schedule.py | 188 p = P.FloorDiv()(p, self.decay_steps) * self.decay_steps
|
| /third_party/python/Parser/ |
| D | Python.asdl | 100 | RShift | BitOr | BitXor | BitAnd | FloorDiv
|