| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | nllloss_grad.cc | 32 void CheckNLLLossGradShapeValid(const PrimitivePtr &primitive, const ShapeVector &logits_shape, in CheckNLLLossGradShapeValid()
65 auto logits_shape = logits_shape_ptr->GetShapeVector(); in InferShape() local
|
| D | nllloss.cc | 31 void CheckNLLLossShapeValid(const PrimitivePtr &primitive, const ShapeVector &logits_shape, in CheckNLLLossShapeValid()
64 auto logits_shape = logits_shape_ptr->GetShapeVector(); in InferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | sparse_softmax_cross_entropy_with_logits_gpu_kernel.h | 97 auto logits_shape = inputs[kIndex0]->GetShapeVector(); in Resize() local
136 void InferInputOutputSize(const ShapeVector &logits_shape, const ShapeVector &labels_shape) { in InferInputOutputSize()
161 void CheckShapeValidation(const ShapeVector &logits_shape, const ShapeVector &labels_shape) { in CheckShapeValidation()
|
| D | softmax_cross_entropy_with_logits_gpu_kernel.h | 64 int CheckShapeValidation(const ShapeVector &logits_shape, const ShapeVector &labels_shape) { in CheckShapeValidation()
|
| D | nll_loss_gpu_kernel.cc | 47 auto logits_shape = inputs[kIndex0]->GetShapeVector(); in Resize() local
|
| D | softmax_cross_entropy_with_logits_gpu_kernel.cc | 45 auto logits_shape = inputs[kIndex0]->GetShapeVector(); in Resize() local
|
| D | nll_loss_grad_gpu_kernel.cc | 51 auto logits_shape = inputs[kIndex0]->GetShapeVector(); in Resize() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/ |
| D | test_sparseSoftmaxCrossEntropyWithLogits.py | 36 def np_sparse_softmax_cross_entropy_with_logits(labels_shape, logits_shape, logits_dtype): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32_grad/ |
| D | nllloss_grad.cc | 50 const auto logits_shape = in_tensors_[kLogitsIndex]->shape(); in Prepare() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | sigmoid_cross_entropy_with_logits.cc | 47 auto logits_shape = input_args[0]->GetShape(); in InferShape() local
|
| D | softmax_cross_entropy_with_logits.cc | 39 auto logits_shape = input_args[0]->GetShape(); in InferShape() local
|
| D | random_categorical.cc | 73 auto logits_shape = logits_shape_map[kShape]; in RandomCategoricalInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | nllloss_grad_cpu_kernel.cc | 54 auto logits_shape = inputs[0]->GetShapeVector(); in Resize() local
|
| D | nllloss_cpu_kernel.cc | 54 auto logits_shape = inputs[kIndex0]->GetShapeVector(); in Resize() local
|
| D | sparse_softmax_cross_entropy_with_logits_cpu_kernel.cc | 47 auto logits_shape = inputs[0]->GetDeviceShapeVector(); in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/random/ |
| D | random_categorical_gpu_kernel.h | 125 auto logits_shape = inputs[0]->GetShapeVector(); in Resize() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/ |
| D | test_ops_nllloss.cc | 37 ShapeVector logits_shape; member
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/ |
| D | sparse_softmax_cross_entropy_with_logits_unify_mindir.cc | 104 auto logits_shape = common::AnfAlgo::GetPrevNodeOutputInferShape(sparse_softmax_node, 0UL); in GetDepthAndBatchSizeFromSparseSoftmaxNode() local
650 auto logits_shape = common::AnfAlgo::GetPrevNodeOutputInferShape(sparse_softmax_node, 0UL); in Process() local
693 auto logits_shape = common::AnfAlgo::GetPrevNodeOutputInferShape(sparse_softmax_node, 0UL); in Process() local
781 auto logits_shape = common::AnfAlgo::GetPrevNodeOutputInferShape(sparse_softmax_node, 0UL); in Process() local
845 auto logits_shape = common::AnfAlgo::GetPrevNodeOutputInferShape(sparse_softmax_node, 0UL); in Process() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/python_input/gtest_input/pipeline/infer/ |
| D | primitive_test.py | 53 def infer_shape(self, logits_shape, label_shape): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | nllloss_fp32_test.cc | 36 std::vector<int> logits_shape = {3, 5}; in NLLLossInitArgs() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/ |
| D | random_ops_proto.cc | 273 ge::Shape logits_shape = logits_desc.GetShape(); in IMPLEMT_COMMON_INFERFUNC() local
308 ge::Shape logits_shape = logits_desc.GetShape(); in IMPLEMT_COMMON_INFERFUNC() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/ |
| D | nllloss_grad_fp32_test.cc | 37 std::vector<int> logits_shape = {3, 5}; in NLLLossGradInitArgs() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/loss/ |
| D | loss.py | 792 def _check_dice_shape(logits_shape, label_shape, prim_name=None): argument
871 def _check_shape(logits_shape, label_shape, prim_name=None): argument
2332 def _check_cross_entropy_inputs(logits_shape, label_shape, argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/ir/ |
| D | ir_pass.cc | 182 auto logits_shape = common::AnfAlgo::GetPrevNodeOutputInferShape(sparse_softmax_node, kIndex0); in Run() local
215 auto logits_shape = common::AnfAlgo::GetPrevNodeOutputInferShape(sparse_softmax_node, kIndex0); in GetDepthAndBatchSizeFromSparseSoftmaxNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/function/ |
| D | func_pass.cc | 97 auto logits_shape = BaseShapeToShape(inputs[kIndex0]->GetShape()); in GetDepthAndBatchSizeFromSparseSoftmaxNode() local
|