Searched refs:EMBEDDING_LOOKUP (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/core/protos/ |
| D | ps.proto | 33 EMBEDDING_LOOKUP = 8; enumerator
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel_utils.cc | 203 if (replace_op.first == EMBEDDING_LOOKUP) { in ReplaceOpInput()
|
| D | step_auto_parallel.cc | 169 …EMBEDDING_LOOKUP, FUSE_BATCH_NORM_EX, SPLIT, BROADCAST_TO, ABS, ACOSH, ASIN, ASINH, ATAN, ATANH, C… in IsSplittableOperator()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | ops_utils.h | 248 constexpr char EMBEDDING_LOOKUP[] = "EmbeddingLookup"; variable
|
| D | gather_v2_p_info.cc | 843 OperatorName op_name = EMBEDDING_LOOKUP; in ComputeReplaceOp()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | schema.fbs | 225 EMBEDDING_LOOKUP = 7,
|
| /third_party/mindspore/third_party/proto/tensorflow/lite/ |
| D | schema.fbs | 225 EMBEDDING_LOOKUP = 7,
|
| D | schema_2.4.1.fbs | 228 EMBEDDING_LOOKUP = 7,
|