Home | History | Annotate | Download | only in tensors

Lines Matching defs:b_

32     device_.deallocate(b_);
57 TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(b_, sizes);
88 const TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, sizes);
117 const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(b_, input_size);
134 const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(b_, input_size);
156 TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, size_b);
179 TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, size_b);
207 TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, size_b);
262 const TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, sizes);
280 const TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, sizes);
298 const TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, sizes);
315 const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(b_, input_size);
344 b_, input_size);
374 b_, input_size);
401 const TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, sizeB);
425 TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, kernel_sizes);
447 b_ = (T *) device_.allocate(k_ * n_ * sizeof(T));
453 device_.memset(b_, 23, k_ * n_ * sizeof(T));
474 T* b_;