Home
last modified time | relevance | path

Searched refs:operator_info (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_auto_parallel.cc250 void SetStrategyToOperator(const OperatorInfoPtr &operator_info, const PrimitivePtr &prim, in SetStrategyToOperator() argument
266 if (operator_info->SetCostUnderStrategy(strategyPtr) != SUCCESS) { in SetStrategyToOperator()
270 int64_t used_devices = operator_info->used_devices(); in SetStrategyToOperator()
275 (void)configured_stra_ops_.emplace(operator_info, strategyPtr); in SetStrategyToOperator()
287 (void)configured_stra_ops_.emplace(operator_info, strategyPtr); in SetStrategyToOperator()
301 OperatorInfoPtr operator_info = NewOperatorInstance(prim, attrs, shape_list); in CreateTheOperatorInfo() local
302 MS_EXCEPTION_IF_NULL(operator_info); in CreateTheOperatorInfo()
305 if (operator_info->set_is_parameter(parameter_info) != SUCCESS) { in CreateTheOperatorInfo()
306 …MS_LOG(ERROR) << "Initializing parameter information failed for operator: " << operator_info->name… in CreateTheOperatorInfo()
316 …if (operator_info->SetInputAndOutputTypeLength(inputs_type_length, outputs_type_length) != SUCCESS… in CreateTheOperatorInfo()
[all …]
Dstep_parallel.cc2196 OperatorInfoPtr operator_info = cnode->user_data<OperatorInfo>(); in ReshapeInit() local
2197 if (operator_info == nullptr) { in ReshapeInit()
2210 auto reshape_info_ptr = std::dynamic_pointer_cast<ReshapeInfo>(operator_info); in ReshapeInit()
2216 auto reshape_info_ptr = std::dynamic_pointer_cast<ReshapeInfo>(operator_info); in ReshapeInit()
2219 auto reshape_info_ptr = std::dynamic_pointer_cast<ReshapeInfo>(operator_info); in ReshapeInit()
2222 if (operator_info->Init(nullptr) == FAILED) { in ReshapeInit()
2326 OperatorInfoPtr operator_info = loss_cnode->user_data<OperatorInfo>(); in GetLossNodeGradOutputLayout() local
2327 MS_EXCEPTION_IF_NULL(operator_info); in GetLossNodeGradOutputLayout()
2329 size_t op_output_size = operator_info->outputs_tensor_info().size(); in GetLossNodeGradOutputLayout()
2330 MS_LOG(INFO) << "The loss name is " << operator_info->name() << ", the has tuple item is " in GetLossNodeGradOutputLayout()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dcpu_session.cc325 std::stringstream operator_info; in KernelNotSupportException() local
326 operator_info << "Operator[" << kernel_name << "] "; in KernelNotSupportException()
329 operator_info << "is not support."; in KernelNotSupportException()
330 MS_LOG(EXCEPTION) << operator_info.str(); in KernelNotSupportException()
334 operator_info << "is not support."; in KernelNotSupportException()
335 MS_LOG(EXCEPTION) << operator_info.str(); in KernelNotSupportException()
339 operator_info << " input("; in KernelNotSupportException()
341 operator_info << TypeIdLabel(kernel_build_Info->GetInputDeviceType(i)); in KernelNotSupportException()
343 operator_info << ","; in KernelNotSupportException()
346 operator_info << ") "; in KernelNotSupportException()
[all …]
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dkernel_select_cpu.cc193 std::stringstream operator_info; in KernelNotSupportException() local
194 operator_info << "Operator[" << kernel_name << "] "; in KernelNotSupportException()
197 operator_info << " input("; in KernelNotSupportException()
199 operator_info << TypeIdLabel(input_types[i]); in KernelNotSupportException()
201 operator_info << ","; in KernelNotSupportException()
204 operator_info << ") "; in KernelNotSupportException()
208 operator_info << "output("; in KernelNotSupportException()
210 operator_info << TypeIdLabel(infer_output_types[i]); in KernelNotSupportException()
212 operator_info << ","; in KernelNotSupportException()
215 operator_info << ") "; in KernelNotSupportException()
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/allreduce_fusion/
Dallreduce_fusion.cc101 auto operator_info = cnode->user_data<OperatorInfo>(); in FindCNode() local
103 << " operator_info: " << (operator_info != nullptr); in FindCNode()
105 if (IsParallelCareNode(cnode) && (operator_info != nullptr)) { in FindCNode()
106 auto cost = operator_info->GetForwardMemoryCostFromCNode(); in FindCNode()
/third_party/mindspore/mindspore/_extends/parallel_compile/tbe_compiler/
Dcompiler.py22 import tbe.common.context.op_info as operator_info namespace
126 op_info = operator_info.OpInfo(op_type, op_type)
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dnode_info.cc292 auto operator_info = cnode->user_data<OperatorInfo>(); in FindReshape() local
293 std::string op_info_name = operator_info->name(); in FindReshape()
/third_party/mindspore/mindspore/ccsrc/debug/
Danf_ir_dump.cc298 auto operator_info = node->user_data<parallel::OperatorInfo>(); in DumpParallelInfo() local
299 if (operator_info == nullptr) { in DumpParallelInfo()
303 auto strategy = operator_info->strategy(); in DumpParallelInfo()
/third_party/mindspore/mindspore/profiler/parser/
Dhccl_parser.py290 operator_info = json.load(src_file)
294 trace_events = operator_info.get("traceEvents")
/third_party/mindspore/mindspore/lite/tools/converter/
DCMakeLists.txt106 ../optimizer/parallel/operator_info.cc