Searched refs:op_prim (Results 1 – 3 of 3) sorted by relevance
29 PrimitivePtr op_prim = std::make_shared<Primitive>(op_name); in ConstructKernelGraphBasedOnSingleOp() local30 MS_EXCEPTION_IF_NULL(op_prim); in ConstructKernelGraphBasedOnSingleOp()31 inputs.push_back(std::make_shared<ValueNode>(op_prim)); in ConstructKernelGraphBasedOnSingleOp()
2134 PrimitivePtr op_prim = op_run_info.primitive; in ConstructSingleOpGraph() local2135 MS_EXCEPTION_IF_NULL(op_prim); in ConstructSingleOpGraph()2137 inputs.push_back(std::make_shared<ValueNode>(std::make_shared<Primitive>(*op_prim))); in ConstructSingleOpGraph()
349 const auto &op_prim = op_exec_info->py_primitive; in GetSingleOpGraphInfo() local350 MS_EXCEPTION_IF_NULL(op_prim); in GetSingleOpGraphInfo()351 const auto &attr_map = op_prim->attrs(); in GetSingleOpGraphInfo()389 …rtConstInputToAttr(const py::object &input_object, size_t input_index, const PrimitivePtr &op_prim, in RunOpConvertConstInputToAttr() argument391 MS_EXCEPTION_IF_NULL(op_prim); in RunOpConvertConstInputToAttr()392 const auto &input_names_value = op_prim->GetAttr(kAttrInputNames); in RunOpConvertConstInputToAttr()404 op_prim->AddAttr(input_name, value); in RunOpConvertConstInputToAttr()410 void PlantTensorTupleToVector(const py::tuple &tuple_inputs, const PrimitivePtr &op_prim, in PlantTensorTupleToVector() argument412 MS_EXCEPTION_IF_NULL(op_prim); in PlantTensorTupleToVector()422 …op_prim->set_attr(kAttrDynInputSizes, MakeValue(std::vector<int64_t>{SizeToLong(tuple_inputs.size(… in PlantTensorTupleToVector()[all …]