Home
last modified time | relevance | path

Searched defs:op_name (Results 1 – 25 of 228) sorted by relevance

12345678910

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/ir/
Dvalidators.cc21 Status ValidateProbability(const std::string &op_name, const double probability) { in ValidateProbability()
31 Status ValidateIntScalarPositive(const std::string &op_name, const std::string &scalar_name, int32_… in ValidateIntScalarPositive()
36 Status ValidateFloatScalarPositive(const std::string &op_name, const std::string &scalar_name, floa… in ValidateFloatScalarPositive()
41 Status ValidateFloatScalarNonNegative(const std::string &op_name, const std::string &scalar_name, f… in ValidateFloatScalarNonNegative()
46 Status ValidateVectorFillvalue(const std::string &op_name, const std::vector<uint8_t> &fill_value) { in ValidateVectorFillvalue()
58 Status ValidateVectorColorAttribute(const std::string &op_name, const std::string &attr_name, in ValidateVectorColorAttribute()
80 Status ValidateVectorMeanStd(const std::string &op_name, const std::vector<float> &mean, in ValidateVectorMeanStd()
106 Status ValidateVectorOdd(const std::string &op_name, const std::string &vec_name, const std::vector… in ValidateVectorOdd()
118 Status ValidateVectorPadding(const std::string &op_name, const std::vector<int32_t> &padding) { in ValidateVectorPadding()
131 Status ValidateVectorPositive(const std::string &op_name, const std::string &vec_name, in ValidateVectorPositive()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dconst_input_to_attr_registry.h28 explicit ConstInputToAttrInfoRegister(const std::string &op_name = "") : op_name_(op_name) {} in op_name_() argument
71 #define REG_CONST_INPUT_TO_ATTR(op_name) REG_CONST_INPUT_TO_ATTR_UNIQ_HELPER(__COUNTER__, op_name) argument
72 #define REG_CONST_INPUT_TO_ATTR_UNIQ_HELPER(ctr, op_name) REG_CONST_INPUT_TO_ATTR_UNIQ(ctr, op_name) argument
73 #define REG_CONST_INPUT_TO_ATTR_UNIQ(ctr, op_name) … argument
Dconst_input_to_attr_registry.cc95 auto op_name = reg.GetOpName(); in Register() local
102 void ConstInputToAttrInfoRegistry::Register(const std::string &op_name, in Register()
112 bool ConstInputToAttrInfoRegistry::GetRegisterByOpName(const std::string &op_name, in GetRegisterByOpName()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/profiling/reporter/
Dprofiling_desc.h29 explicit ProfDesc(std::string op_name) : op_name_(std::move(op_name)) {} in ProfDesc()
39 TaskDesc(std::string op_name, uint32_t task_id, uint32_t block_dim, uint32_t stream_id) in TaskDesc()
59 GraphDesc(std::string op_name, std::string op_type, std::vector<DataElement> input_data_list, in GraphDesc()
77 …PointDesc(std::string op_name, uint32_t point_id) : ProfDesc(std::move(op_name)), point_id_(point_… in PointDesc()
87 …TaskStreamOpNameDesc(std::string op_name, std::vector<std::pair<uint32_t, uint32_t>> stream_id_tas… in TaskStreamOpNameDesc()
/third_party/boost/libs/compute/include/boost/compute/container/
Dvalarray.hpp270 #define BOOST_COMPUTE_DEFINE_VALARRAY_COMPOUND_ASSIGNMENT(op, op_name, assert) \ argument
291 #define BOOST_COMPUTE_DEFINE_VALARRAY_COMPOUND_ASSIGNMENT_ANY(op, op_name) \ argument
303 #define BOOST_COMPUTE_DEFINE_VALARRAY_COMPOUND_ASSIGNMENT_NO_FP(op, op_name) \ argument
341 #define BOOST_COMPUTE_DEFINE_VALARRAY_BINARY_OPERATOR(op, op_name, assert) \ argument
380 #define BOOST_COMPUTE_DEFINE_VALARRAY_BINARY_OPERATOR_ANY(op, op_name) \ argument
392 #define BOOST_COMPUTE_DEFINE_VALARRAY_BINARY_OPERATOR_NO_FP(op, op_name) \ argument
424 #define BOOST_COMPUTE_DEFINE_VALARRAY_COMPARISON_OPERATOR(op, op_name) \ argument
487 #define BOOST_COMPUTE_DEFINE_VALARRAY_LOGICAL_OPERATOR(op, op_name) \ argument
/third_party/boost/boost/compute/container/
Dvalarray.hpp270 #define BOOST_COMPUTE_DEFINE_VALARRAY_COMPOUND_ASSIGNMENT(op, op_name, assert) \ argument
291 #define BOOST_COMPUTE_DEFINE_VALARRAY_COMPOUND_ASSIGNMENT_ANY(op, op_name) \ argument
303 #define BOOST_COMPUTE_DEFINE_VALARRAY_COMPOUND_ASSIGNMENT_NO_FP(op, op_name) \ argument
341 #define BOOST_COMPUTE_DEFINE_VALARRAY_BINARY_OPERATOR(op, op_name, assert) \ argument
380 #define BOOST_COMPUTE_DEFINE_VALARRAY_BINARY_OPERATOR_ANY(op, op_name) \ argument
392 #define BOOST_COMPUTE_DEFINE_VALARRAY_BINARY_OPERATOR_NO_FP(op, op_name) \ argument
424 #define BOOST_COMPUTE_DEFINE_VALARRAY_COMPARISON_OPERATOR(op, op_name) \ argument
487 #define BOOST_COMPUTE_DEFINE_VALARRAY_LOGICAL_OPERATOR(op, op_name) \ argument
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/
Dtrt_op_factory.h44 void Register(const std::string &op_name, const ConvertFunc &func) { in Register()
51 ConvertFunc GetConvertFunc(const std::string &op_name) const { in GetConvertFunc()
70 …TrtOpRegister(const std::string &op_name, ConvertFunc func) { TrtOpFactory::GetInstance().Register… in TrtOpRegister()
/third_party/boost/libs/compute/test/
Dtest_valarray.cpp81 #define BOOST_COMPUTE_TEST_VALARRAY_COMPOUND_ASSIGNMENT(op, op_name) \ argument
126 #define BOOST_COMPUTE_TEST_VALARRAY_COMPOUND_ASSIGNMENT_NO_FP(op, op_name) \ argument
228 #define BOOST_COMPUTE_TEST_VALARRAY_BINARY_OPERATOR(op, op_name) \ argument
270 #define BOOST_COMPUTE_TEST_VALARRAY_BINARY_OPERATOR_NO_FP(op, op_name) \ argument
311 #define BOOST_COMPUTE_TEST_VALARRAY_COMPARISON_OPERATOR(op, op_name) \ argument
351 #define BOOST_COMPUTE_TEST_VALARRAY_LOGICAL_OPERATOR(op, op_name) \ argument
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/audio/ir/
Dvalidators.h39 Status ValidateScalarNotZero(const std::string &op_name, const std::string &scalar_name, const T sc… in ValidateScalarNotZero()
50 Status ValidateVectorNotEmpty(const std::string &op_name, const std::string &vec_name, const std::v… in ValidateVectorNotEmpty()
61 Status ValidateVectorSameSize(const std::string &op_name, const std::string &vec1_name, const std::… in ValidateVectorSameSize()
Dvalidators.cc20 Status ValidateIntScalarNonNegative(const std::string &op_name, const std::string &scalar_name, int… in ValidateIntScalarNonNegative()
25 Status ValidateFloatScalarNotNan(const std::string &op_name, const std::string &scalar_name, float … in ValidateFloatScalarNotNan()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/
Dtask_info.h52 std::string op_name() const { return op_name_; } in op_name() function
56 TaskInfo(const std::string &op_name, uint32_t stream_id, TaskInfoType type, bool dump_flag) in TaskInfo()
68 …TbeTaskInfo(const std::string &op_name, uint32_t stream_id, const std::string &stub_func, uint32_t… in TbeTaskInfo()
119 AicpuTaskInfo(const std::string &op_name, uint32_t stream_id, const std::string &so_name, in AicpuTaskInfo()
150 LabelSetTaskInfo(const std::string &op_name, uint32_t stream_id, uint32_t label_id) in LabelSetTaskInfo()
161 LabelGotoTaskInfo(const std::string &op_name, uint32_t stream_id, uint32_t label_id) in LabelGotoTaskInfo()
172 LabelSwitchTaskInfo(const std::string &op_name, uint32_t stream_id, uint32_t label_size, in LabelSwitchTaskInfo()
194 …EventTaskInfo(const std::string &op_name, uint32_t stream_id, TaskInfoType type, uint32_t event_id) in EventTaskInfo()
203 EventRecordTaskInfo(const std::string &op_name, uint32_t stream_id, uint32_t event_id) in EventRecordTaskInfo()
210 EventWaitTaskInfo(const std::string &op_name, uint32_t stream_id, uint32_t event_id) in EventWaitTaskInfo()
[all …]
/third_party/mindspore/mindspore/core/abstract/
Dprim_others.cc205 const std::string op_name = primitive->name(); in InferImplMakeRowTensor() local
268 const std::string op_name = primitive->name(); in InferImplRowTensorGetValues() local
278 const std::string op_name = primitive->name(); in InferImplRowTensorGetIndices() local
288 const std::string op_name = primitive->name(); in InferImplRowTensorGetDenseShape() local
298 const std::string op_name = primitive->name(); in InferImplRowTensorAdd() local
312 const std::string op_name = primitive->name(); in InferImplMakeSparseTensor() local
371 const std::string op_name = primitive->name(); in InferImplSparseTensorGetValues() local
381 const std::string op_name = primitive->name(); in InferImplSparseTensorGetIndices() local
391 const std::string op_name = primitive->name(); in InferImplSparseTensorGetDenseShape() local
400 const std::string op_name = primitive->name(); in InferImplAllSwap() local
[all …]
Dprim_structures.cc38 const std::string op_name = primitive->name(); in InferImplMakeDict() local
68 const std::string op_name = primitive->name(); in InferImplMakeKwarg() local
84 const std::string op_name = primitive->name(); in InferImplExtractKwarg() local
159 AbstractBasePtr InferTupleOrListGetItem(const std::string &op_name, const AbstractBasePtrList &args… in InferTupleOrListGetItem()
192 AbstractBasePtr InferTupleOrListSetItem(const std::string &op_name, const AbstractBasePtrList &args… in InferTupleOrListSetItem()
240 const std::string op_name = primitive->name(); in InferImplDictGetItem() local
263 const std::string op_name = primitive->name(); in InferImplDictSetItem() local
292 const std::string op_name = primitive->name(); in InferImplDictGetKeys() local
305 const std::string op_name = primitive->name(); in InferImplDictGetValues() local
318 const std::string op_name = primitive->name(); in InferImplListAppend() local
[all …]
Dprim_arrays.cc32 const std::string op_name = primitive->name(); in InferImplScalarToArray() local
41 const std::string op_name = primitive->name(); in InferImplArrayToScalar() local
55 const std::string op_name = primitive->name(); in InferImplBroadCastShape() local
95 const std::string op_name = primitive->name(); in InferImplStack() local
137 const std::string op_name = primitive->name(); in InferImplUnique() local
182 const std::string op_name = primitive->name(); in InferImplPadAndShift() local
204 const std::string op_name = primitive->name(); in InferImplUniqueGrad() local
228 const std::string op_name = primitive->name(); in InferImplUnsortedSegmentSum() local
284 const std::string op_name = primitive->name(); in InferImplUnsortedSegmentMax() local
335 const std::string op_name = primitive->name(); in InferImplUnsortedSegmentMin() local
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dinput_to_output_registry.h32 : op_name_(op_name), pre_check_func_(pre_check_func) {} in op_name_() argument
38 const std::string &op_name() const { return op_name_; } in op_name() function
Dinput_to_output_registry.cc96 auto op_name = reg.op_name(); in Register() local
103 void InputToOutputRegistry::Register(const std::string &op_name, const std::vector<size_t> &input_i… in Register()
113 bool InputToOutputRegistry::GetRegisterByOpName(const std::string &op_name, InputToOutputRegister *… in GetRegisterByOpName() argument
/third_party/mindspore/mindspore/ccsrc/profiler/device/
Dprofiling.cc54 void Profiler::SetRunTimeData(const std::string &op_name, const float time_elapsed) { in SetRunTimeData()
61 void Profiler::SetRunTimeData(const std::string &op_name, const uint64_t start, const float duratio… in SetRunTimeData()
75 void Profiler::RecordOneStepStartEndInfo(const std::string op_name) { in RecordOneStepStartEndInfo()
/third_party/mindspore/mindspore/core/ops/
Dnot_equal.cc31 auto op_name = prim->name(); in InferType() local
42 auto op_name = primitive->name(); in NotEqualInfer() local
Dones_like.cc31 auto op_name = primitive->name(); in InferShape() local
38 auto op_name = primitive->name(); in InferType() local
Dzeros_like.cc31 auto op_name = primitive->name(); in InferShape() local
37 auto op_name = primitive->name(); in InferType() local
Dminimum.cc32 auto op_name = primitive->name(); in InferShape() local
38 auto op_name = prim->name(); in InferType() local
Dctcloss.cc31 void CheckCTCLossInputs(const std::vector<AbstractBasePtr> &input_args, const std::string &op_name)… in CheckCTCLossInputs()
70 auto op_name = primitive->name(); in InferShape() local
94 auto op_name = primitive->name(); in InferType() local
Dgelu.cc32 auto op_name = primitive->name(); in InferShape() local
46 auto op_name = prim->name(); in InferType() local
/third_party/mindspore/mindspore/ops/composite/
Darray_ops.py27 def _check_is_int(arg_value, arg_name, op_name): argument
33 def _check_positive_int(arg_value, arg_name, op_name): argument
39 def _check_axis_range(arg_value, limit, arg_name, op_name): argument
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/
Daicpu_kernel_metadata.cc31 std::string op_name = AnfAlgo::GetCNodeName(kernel_node); in AicpuMetadataInfo() local
56 auto op_name = AnfAlgo::GetCNodeName(kernel_node); in AicpuMetadataInfoForSpecialNodes() local

12345678910