Searched defs:reduce_op (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/delegate/tensorrt/op/ |
| D | reduce_tensorrt.cc | 27 auto reduce_op = primitive->value_as_ReduceFusion(); in IsSupport() local
53 auto reduce_op = op_primitive_->value_as_ReduceFusion(); in AddInnerOp() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | unique_info.cc | 147 AnfNodePtr reduce_op = gen_g.PushBack({gen_g.NewOpInst(ALL_REDUCE, attrs), mul}); in ComputeReplaceGraph() local
|
| D | operator_info.cc | 310 static OperatorArgs CreateReduceCommunicationOpArgs(const std::string &reduce_op, const std::string… in CreateReduceCommunicationOpArgs()
324 Operator CreateAllReduceOp(const std::string &reduce_op, const std::string &group) { in CreateAllReduceOp()
333 Operator CreateReduceScatterOp(const std::string &reduce_op, const std::string &group) { in CreateReduceScatterOp()
|
| D | gather_v2_p_info.cc | 795 AnfNodePtr reduce_op; in ComputeReplaceGraph() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/ |
| D | nccl_collective_gpu_kernel.h | 224 auto reduce_op = prim->GetAttr(kAttrOp); in InferCommType() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | special_op_eliminate.h | 649 auto reduce_op = primitive->GetAttr("op"); in operator() local
|
| /third_party/mesa3d/src/amd/compiler/ |
| D | aco_lower_to_hw_instr.cpp | 474 emit_reduction(lower_context* ctx, aco_opcode op, ReduceOp reduce_op, unsigned cluster_size, in emit_reduction()
|
| D | aco_ir.h | 1683 ReduceOp reduce_op; member
|
| D | aco_instruction_selection.cpp | 7774 ReduceOp reduce_op = get_reduce_op(op, instr->src[0].ssa->bit_size); in emit_uniform_scan() local
8446 ReduceOp reduce_op = get_reduce_op(op, bit_size); in visit_intrinsic() local
|
| /third_party/python/Modules/ |
| D | _pickle.c | 3948 const char reduce_op = REDUCE; in save_reduce() local
|