File tree Expand file tree Collapse file tree 3 files changed +6
-4
lines changed
Expand file tree Collapse file tree 3 files changed +6
-4
lines changed Original file line number Diff line number Diff 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}
Original file line number Diff line number Diff 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);
Original file line number Diff line number Diff 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 );
You can’t perform that action at this time.
0 commit comments