Browse Source

!27965 fix the adapter of DynamicGRUV2 and the max dimension of input logits of BCEWithLogitsLoss

Merge pull request !27965 from wangshuide/wsd_master
tags/v1.6.0
i-robot Gitee 4 years ago
parent
commit
f4a01b0283
3 changed files with 3 additions and 3 deletions
  1. +1
    -1
      mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/bce_with_logits_loss_impl.cuh
  2. +1
    -1
      mindspore/ccsrc/transform/graph_ir/op_declare/rnn_declare.cc
  3. +1
    -1
      mindspore/python/mindspore/ops/operations/math_ops.py

+ 1
- 1
mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/bce_with_logits_loss_impl.cuh View File

@@ -17,7 +17,7 @@
#ifndef MINDSPORE_CCSRC_BACKEND_KERNEL_COMPILER_GPU_CUDA_IMPL_BCE_WITH_LOGITS_LOSS_IMPL_CUH_
#define MINDSPORE_CCSRC_BACKEND_KERNEL_COMPILER_GPU_CUDA_IMPL_BCE_WITH_LOGITS_LOSS_IMPL_CUH_

#define MAX_LOGITS_DIMENSION 100
#define MAX_LOGITS_DIMENSION 8
#include "runtime/device/gpu/cuda_common.h"

template <typename T>


+ 1
- 1
mindspore/ccsrc/transform/graph_ir/op_declare/rnn_declare.cc View File

@@ -112,7 +112,7 @@ ATTR_MAP(DynamicGRUV2) = {{"direction", ATTR_DESC(direction, AnyTraits<std::stri
{"cell_clip", ATTR_DESC(cell_clip, AnyTraits<float>())},
{"num_proj", ATTR_DESC(num_proj, AnyTraits<int64_t>())},
{"time_major", ATTR_DESC(time_major, AnyTraits<bool>())},
{"activation", ATTR_DESC(direction, AnyTraits<std::string>())},
{"activation", ATTR_DESC(activation, AnyTraits<std::string>())},
{"gate_order", ATTR_DESC(gate_order, AnyTraits<std::string>())},
{"reset_after", ATTR_DESC(reset_after, AnyTraits<bool>())},
{"is_training", ATTR_DESC(is_training, AnyTraits<bool>())}};


+ 1
- 1
mindspore/python/mindspore/ops/operations/math_ops.py View File

@@ -4780,7 +4780,7 @@ class Atanh(Primitive):
Inputs:
- **x** (Tensor): The shape of tensor is
:math:`(N,*)` where :math:`*` means, any number of additional dimensions.
The data type should be one of the following types: float16, float32.
The data type should be one of the following types: float16, float32.

Outputs:
A Tensor, has the same type as the input.


Loading…
Cancel
Save