Searched refs:CrossEntropyLoss (Results 1 – 16 of 16) sorted by relevance
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_bias_add.py | 24 class CrossEntropyLoss(nn.Cell): class
26 super(CrossEntropyLoss, self).__init__()
91 loss = CrossEntropyLoss()
|
| D | test_reshape.py | 552 class CrossEntropyLoss(nn.Cell): class
554 super(CrossEntropyLoss, self).__init__()
576 loss = CrossEntropyLoss()
596 loss = CrossEntropyLoss()
637 loss = CrossEntropyLoss()
|
| D | test_parallel_transformer.py | 25 …VocabEmbedding, CrossEntropyLoss, OpParallelConfig, EmbeddingOpParallelConfig, FixedSparseAttention
74 self.loss = CrossEntropyLoss(parallel_config=config.dp_mp_config)
543 self.loss = CrossEntropyLoss(config_setting)
|
| /third_party/mindspore/tests/st/pynative/ |
| D | test_pynative_lenet.py | 95 class CrossEntropyLoss(nn.Cell): class
101 super(CrossEntropyLoss, self).__init__()
146 criterion = CrossEntropyLoss()
184 criterion = CrossEntropyLoss()
|
| D | test_pynative_resnet50_gpu.py | 351 class CrossEntropyLoss(nn.Cell): class
353 super(CrossEntropyLoss, self).__init__()
389 criterion = CrossEntropyLoss()
|
| D | test_pynative_resnet50_ascend.py | 389 class CrossEntropyLoss(nn.Cell): class
391 super(CrossEntropyLoss, self).__init__()
|
| /third_party/mindspore/tests/st/tbe_networks/ |
| D | test_resnet_cifar_1p.py | 86 class CrossEntropyLoss(nn.Cell): class
88 super(CrossEntropyLoss, self).__init__()
137 loss = CrossEntropyLoss()
|
| D | resnet_cifar.py | 108 class CrossEntropyLoss(nn.Cell): class
110 super(CrossEntropyLoss, self).__init__()
133 loss = CrossEntropyLoss()
|
| D | test_resnet_cifar_8p.py | 97 class CrossEntropyLoss(nn.Cell): class
99 super(CrossEntropyLoss, self).__init__()
159 loss = CrossEntropyLoss()
|
| /third_party/mindspore/tests/st/mem_reuse/ |
| D | resnet_cifar_memreuse.py | 107 class CrossEntropyLoss(nn.Cell): class
109 super(CrossEntropyLoss, self).__init__()
132 loss = CrossEntropyLoss()
|
| D | resnet_cifar_normal.py | 107 class CrossEntropyLoss(nn.Cell): class
109 super(CrossEntropyLoss, self).__init__()
132 loss = CrossEntropyLoss()
|
| /third_party/mindspore/tests/st/nontask_sink/ |
| D | test_lenet.py | 94 class CrossEntropyLoss(nn.Cell): class
100 super(CrossEntropyLoss, self).__init__()
137 criterion = CrossEntropyLoss()
|
| /third_party/mindspore/tests/st/pynative/ms_function/ |
| D | test_pynative_lenet_ms_function.py | 128 class CrossEntropyLoss(nn.Cell): class
134 super(CrossEntropyLoss, self).__init__()
179 criterion = CrossEntropyLoss()
|
| /third_party/mindspore/mindspore/parallel/nn/ |
| D | __init__.py | 23 from .loss import CrossEntropyLoss
|
| D | loss.py | 31 class CrossEntropyLoss(Cell): class
67 super(CrossEntropyLoss, self).__init__()
|
| /third_party/mindspore/tests/ut/python/nn/ |
| D | test_transformer.py | 21 …TransformerDecoder, TransformerDecoderLayer, Transformer, CrossEntropyLoss, AttentionMask, FixedSp…
231 model = CrossEntropyLoss()
|