| /external/tensorflow/tensorflow/core/kernels/ |
| cwise_op_gpu_floor_div.cu.cc | 23 DEFINE_BINARY3(floor_div_real, Eigen::half, float, double);
|
| cwise_op_gpu_greater.cu.cc | 22 DEFINE_BINARY7(greater, Eigen::half, float, double, int64, uint8, int8, int16);
|
| cwise_op_gpu_greater_equal.cu.cc | 22 DEFINE_BINARY7(greater_equal, Eigen::half, float, double, int64, uint8, int8,
|
| cwise_op_gpu_less.cu.cc | 22 DEFINE_BINARY7(less, Eigen::half, float, double, int64, uint8, int8, int16);
|
| cwise_op_gpu_less_equal.cu.cc | 22 DEFINE_BINARY7(less_equal, Eigen::half, float, double, int64, uint8, int8,
|
| cwise_op_gpu_neg.cu.cc | 22 DEFINE_UNARY7(neg, Eigen::half, float, double, int32, int64, complex64,
|
| cwise_op_gpu_not_equal_to.cu.cc | 22 DEFINE_BINARY10(not_equal_to, float, Eigen::half, double, uint8, int8, int16,
|
| cwise_op_rsqrt.cc | 19 REGISTER5(UnaryOp, CPU, "Rsqrt", functor::rsqrt, float, Eigen::half, double,
23 REGISTER3(UnaryOp, GPU, "Rsqrt", functor::rsqrt, float, Eigen::half, double);
30 Eigen::half, double, complex64, complex128);
33 Eigen::half, double);
|
| cwise_op_sigmoid.cc | 20 REGISTER5(UnaryOp, CPU, "Sigmoid", functor::sigmoid, float, Eigen::half, double,
23 REGISTER3(UnaryOp, GPU, "Sigmoid", functor::sigmoid, float, Eigen::half,
31 Eigen::half, double, complex64, complex128);
34 Eigen::half, double);
|
| cwise_op_sqrt.cc | 19 REGISTER5(UnaryOp, CPU, "Sqrt", functor::sqrt, float, Eigen::half, double,
23 REGISTER3(UnaryOp, GPU, "Sqrt", functor::sqrt, float, Eigen::half, double);
31 Eigen::half, double, complex64, complex128);
34 Eigen::half, double);
|
| matrix_logarithm_op.cc | 18 #include "third_party/eigen3/Eigen/Core"
19 #include "third_party/eigen3/unsupported/Eigen/MatrixFunctions"
43 Eigen::Matrix<Scalar, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor>;
|
| reduction_ops_all.cc | 25 ReductionOp<CPUDevice, bool, int32, Eigen::internal::AndReducer>);
31 ReductionOp<CPUDevice, bool, int64, Eigen::internal::AndReducer>);
39 ReductionOp<GPUDevice, bool, int32, Eigen::internal::AndReducer>);
45 ReductionOp<GPUDevice, bool, int64, Eigen::internal::AndReducer>);
|
| reduction_ops_any.cc | 25 ReductionOp<CPUDevice, bool, int32, Eigen::internal::OrReducer>);
31 ReductionOp<CPUDevice, bool, int64, Eigen::internal::OrReducer>);
39 ReductionOp<GPUDevice, bool, int32, Eigen::internal::OrReducer>);
45 ReductionOp<GPUDevice, bool, int64, Eigen::internal::OrReducer>);
|
| slice_op.h | 21 #include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
31 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& slice_indices,
32 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& slice_sizes) {
33 bool use_64bit = (input.size() > Eigen::NumTraits<int>::highest());
35 Eigen::internal::is_same<Device, Eigen::GpuDevice>::value) {
36 Eigen::DSizes<int, NDIMS> indices;
40 Eigen::DSizes<int, NDIMS> sizes
[all...] |
| softmax_op_functor.h | 20 #include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
38 // Eigen code implementing SoftmaxFunctor::operator() or
55 Eigen::DSizes<int, 1> along_class(kClassDim);
56 Eigen::DSizes<int, 2> batch_by_one(batch_size, 1);
57 Eigen::DSizes<int, 2> one_by_class(1, num_classes);
59 Eigen::IndexList<Eigen::type2index<kClassDim> > along_class;
60 Eigen::IndexList<Eigen::type2index<1> > depth_dim;
61 Eigen::IndexList<int, Eigen::type2index<1> > batch_by_one
[all...] |
| eigen_pooling.h | 19 #include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
22 namespace Eigen {
42 const Eigen::DSizes<typename internal::traits<Input>::Index,
47 const Eigen::array<int, 2>,
52 const Eigen::DSizes<typename internal::traits<Input>::Index,
59 const Eigen::IndexList<Eigen::type2index<1>, Eigen::type2index<2> >,
60 const Eigen::IndexList<Eigen::type2index<2>
[all...] |
| /frameworks/native/libs/vr/libdvrcommon/include/private/dvr/ |
| range.h | 4 #include <private/dvr/eigen.h>
9 // TODO(skiazyk): Replace all instances of this with Eigen::AlignedBox
16 Range(Eigen::Vector<T, d> p1, Eigen::Vector<T, d> p2) : p1(p1), p2(p2) {}
18 static Range<T, d> FromSize(Eigen::Vector<T, d> p1,
19 Eigen::Vector<T, d> size) {
27 Eigen::Vector<T, d> GetMinPoint() const { return p1; }
29 Eigen::Vector<T, d> GetMaxPoint() const { return p2; }
31 Eigen::Vector<T, d> GetSize() const { return p2 - p1; }
33 Eigen::Vector<T, d> p1
[all...] |
| /external/eigen/unsupported/test/ |
| cxx11_tensor_index_list.cpp | 1 // This file is part of Eigen, a lightweight C++ template library
12 #include <Eigen/CXX11/Tensor>
54 typedef Eigen::IndexList<Eigen::type2index<0>> Dims0;
55 typedef Eigen::IndexList<Eigen::type2index<0>, Eigen::type2index<1>> Dims1;
56 typedef Eigen::IndexList<Eigen::type2index<0>, Eigen::type2index<1>, Eigen::type2index<2>> Dims2
[all...] |
| /external/eigen/bench/ |
| benchmarkSlice.cpp | 5 #include <Eigen/Core>
8 using namespace Eigen;
20 typedef Matrix<SCALAR, Eigen::Dynamic, Eigen::Dynamic> Mat;
27 r = Eigen::internal::random<int>(0,10);
28 c = Eigen::internal::random<int>(0,10);
29 nr = Eigen::internal::random<int>(50,80);
30 nc = Eigen::internal::random<int>(50,80);
|
| /external/eigen/doc/examples/ |
| QuickStart_example2_dynamic.cpp | 2 #include <Eigen/Dense>
4 using namespace Eigen;
|
| QuickStart_example2_fixed.cpp | 2 #include <Eigen/Dense>
4 using namespace Eigen;
|
| TutorialLinAlgExComputeSolveError.cpp | 2 #include <Eigen/Dense>
5 using namespace Eigen;
|
| TutorialLinAlgExSolveColPivHouseholderQR.cpp | 2 #include <Eigen/Dense>
5 using namespace Eigen;
|
| TutorialLinAlgExSolveLDLT.cpp | 2 #include <Eigen/Dense>
5 using namespace Eigen;
|
| TutorialLinAlgRankRevealing.cpp | 2 #include <Eigen/Dense>
5 using namespace Eigen;
|
