Skip to content

Commit 9945292

Browse files
tests
1 parent b7357c6 commit 9945292

File tree

3 files changed

+6
-4
lines changed

3 files changed

+6
-4
lines changed

src/tensor-array/core/extern_type_map.hh

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,9 @@ namespace tensor_array
3737
{
3838
namespace value
3939
{
40-
extern std::unordered_map<std::type_index, std::size_t> TENSOR_ARRAY_API dynamic_type_size;
40+
/**
41+
* Map of data types.
42+
*/
43+
extern TENSOR_ARRAY_API std::unordered_map<std::type_index, std::size_t> dynamic_type_size;
4144
}
4245
}

src/tensor-array/core/tensor.hh

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,8 +39,7 @@ namespace tensor_array
3939
{
4040
namespace value
4141
{
42-
bool TENSOR_ARRAY_API is_use_grad();
43-
void TENSOR_ARRAY_API set_use_grad(bool use_grad);
42+
extern TENSOR_ARRAY_API bool use_grad;
4443

4544
#ifdef TENSOR_CONTENT
4645
void* create_mem_101(std::size_t s, const void* dat);

src/tensor-array/layers/normalization.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ namespace tensor_array
4040
value::Tensor NormalizationImpl::calculate(const value::Tensor& input)
4141
{
4242
value::Tensor normal;
43-
if (tensor_array::value::is_use_grad())
43+
if (tensor_array::value::use_grad)
4444
{
4545
value::Tensor temp_mean = input.mean(this->dims_mean);
4646
value::Tensor temp_variance = input.variance(this->dims_mean);

0 commit comments

Comments
 (0)