Searched refs:ONEHOT (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/ |
| D | rec_parse_graph.h | 51 {ONEHOT, OperatorType::kRecOneHot},
|
| D | rec_generate_strategy.cc | 593 } else if (type == ONEHOT) { in PrepareStrategy()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | onehot_info.cc | 193 …auto onehot = gen_g.PushBack({gen_g.NewOpInst(ONEHOT, attrs_onehot), sub2, CreatInt64Imm(classes_e… in ComputeReplaceGraph()
|
| D | ops_utils.h | 288 constexpr char ONEHOT[] = "OneHot"; variable
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_auto_parallel.cc | 163 …FLOORDIV, L2_NORMALIZE, ADD, MAXPOOL, AVGPOOL, MAXPOOLV2, VIRTUAL_DATA_SET, RELU, ONEHOT, DROPOUT_… in IsSplittableOperator()
|