| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_benchmark.h | 36 using Dimensions = Eigen::DSizes<Eigen::Index, 4>; variable
41 Eigen::Index BufferSize(const Dimensions& dims) { in BufferSize()
45 void SpatialConvolution(Dimensions input_dims, Dimensions filter_dims) { in SpatialConvolution()
46 Dimensions output_dims(input_dims[0], // batch in SpatialConvolution()
77 void SpatialConvolutionBackwardInput(Dimensions input_dims, in SpatialConvolutionBackwardInput()
78 Dimensions filter_dims) { in SpatialConvolutionBackwardInput()
82 Dimensions output_dims(input_dims[0], // batch in SpatialConvolutionBackwardInput()
118 void SpatialConvolutionBackwardKernel(Dimensions input_dims, in SpatialConvolutionBackwardKernel()
119 Dimensions filter_dims) { in SpatialConvolutionBackwardKernel()
123 Dimensions output_dims(input_dims[0], // batch in SpatialConvolutionBackwardKernel()
[all …]
|
| D | eigen_benchmark_cpu_test.cc | 44 typename Benchmark::Dimensions input_dims(input_batches, input_height, in SpatialConvolution()
46 typename Benchmark::Dimensions filter_dims(filter_height, filter_width, in SpatialConvolution()
73 typename Benchmark::Dimensions input_dims(input_batches, input_height, in SpatialConvolutionBackwardInput()
75 typename Benchmark::Dimensions filter_dims(filter_height, filter_width, in SpatialConvolutionBackwardInput()
101 typename Benchmark::Dimensions input_dims(input_batches, input_height, in SpatialConvolutionBackwardKernel()
103 typename Benchmark::Dimensions filter_dims(filter_height, filter_width, in SpatialConvolutionBackwardKernel()
266 typename Benchmark::Dimensions input_dims( in CuboidConvolution()
268 typename Benchmark::Dimensions filter_dims( in CuboidConvolution()
296 typename Benchmark::Dimensions input_dims( in CuboidConvolutionBackwardInput()
298 typename Benchmark::Dimensions filter_dims( in CuboidConvolutionBackwardInput()
[all …]
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorRef.h | 17 template <typename Dimensions, typename Scalar>
23 EIGEN_DEVICE_FUNC virtual const Dimensions& dimensions() const = 0;
42 template <typename Dimensions, typename Expr, typename Device>
43 class TensorLazyEvaluatorReadOnly : public TensorLazyBaseEvaluator<Dimensions, typename TensorEvalu…
56 EIGEN_DEVICE_FUNC virtual const Dimensions& dimensions() const { in dimensions()
73 Dimensions m_dims;
77 template <typename Dimensions, typename Expr, typename Device>
78 class TensorLazyEvaluatorWritable : public TensorLazyEvaluatorReadOnly<Dimensions, Expr, Device> {
80 typedef TensorLazyEvaluatorReadOnly<Dimensions, Expr, Device> Base;
93 template <typename Dimensions, typename Expr, typename Device>
[all …]
|
| D | TensorTraits.h | 65 template<typename Scalar_, typename Dimensions, int Options_, typename IndexType_>
66 struct traits<TensorFixedSize<Scalar_, Dimensions, Options_, IndexType_> >
71 static const int NumDimensions = array_size<Dimensions>::value;
133 template<typename Scalar_, typename Dimensions, int Options, typename IndexType_>
134 struct eval<TensorFixedSize<Scalar_, Dimensions, Options, IndexType_>, Eigen::Dense>
136 typedef const TensorFixedSize<Scalar_, Dimensions, Options, IndexType_>& type;
139 template<typename Scalar_, typename Dimensions, int Options, typename IndexType_>
140 struct eval<const TensorFixedSize<Scalar_, Dimensions, Options, IndexType_>, Eigen::Dense>
142 typedef const TensorFixedSize<Scalar_, Dimensions, Options, IndexType_>& type;
187 template <typename Scalar_, typename Dimensions, int Options, typename IndexType_>
[all …]
|
| D | TensorEvaluator.h | 34 typedef typename Derived::Dimensions Dimensions; typedef
54 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dims; } in dimensions()
118 Dimensions m_dims;
154 typedef typename Derived::Dimensions Dimensions;
175 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dims; }
217 Dimensions m_dims;
250 typedef typename TensorEvaluator<ArgType, Device>::Dimensions Dimensions;
252 EIGEN_DEVICE_FUNC const Dimensions& dimensions() const { return m_argImpl.dimensions(); }
315 typedef typename TensorEvaluator<ArgType, Device>::Dimensions Dimensions;
317 EIGEN_DEVICE_FUNC const Dimensions& dimensions() const { return m_argImpl.dimensions(); }
[all …]
|
| D | TensorSyclExtractFunctors.h | 37 typedef typename Evaluator::Dimensions Dimensions; typedef
38 const Dimensions m_dimensions;
39 const Dimensions& dimensions() const { return m_dimensions; } in dimensions()
157 …utDims==0, DSizes<typename Evaluator::Index, 1>, typename Evaluator::Dimensions >::type Dimensions;
158 const Dimensions m_dimensions;
159 const Dimensions& dimensions() const { return m_dimensions; }
161 : m_dimensions(DimConstr<Dimensions, Evaluator::NumOutputDims>::getDim(expr.dimensions())) {}
|
| D | TensorConcatenation.h | 114 …t NumDims = internal::array_size<typename TensorEvaluator<LeftArgType, Device>::Dimensions>::value;
115 …tNumDims = internal::array_size<typename TensorEvaluator<RightArgType, Device>::Dimensions>::value;
116 typedef DSizes<Index, NumDims> Dimensions;
135 const Dimensions& lhs_dims = m_leftImpl.dimensions();
136 const Dimensions& rhs_dims = m_rightImpl.dimensions();
177 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
213 const Dimensions& left_dims = m_leftImpl.dimensions();
230 const Dimensions& right_dims = m_rightImpl.dimensions();
281 Dimensions m_dimensions;
297 typedef typename Base::Dimensions Dimensions;
[all …]
|
| D | TensorContraction.h | 24 template<typename Dimensions, typename LhsXprType, typename RhsXprType>
25 struct traits<TensorContractionOp<Dimensions, LhsXprType, RhsXprType> >
41 …RhsXprType>::NumDimensions + traits<RhsXprType>::NumDimensions - 2 * array_size<Dimensions>::value;
49 template<typename Dimensions, typename LhsXprType, typename RhsXprType>
50 struct eval<TensorContractionOp<Dimensions, LhsXprType, RhsXprType>, Eigen::Dense>
52 typedef const TensorContractionOp<Dimensions, LhsXprType, RhsXprType>& type;
55 template<typename Dimensions, typename LhsXprType, typename RhsXprType>
56 struct nested<TensorContractionOp<Dimensions, LhsXprType, RhsXprType>, 1, typename eval<TensorContr…
58 typedef TensorContractionOp<Dimensions, LhsXprType, RhsXprType> type;
140 internal::array_size<typename TensorEvaluator<EvalLeftArgType, Device>::Dimensions>::value;
[all …]
|
| D | TensorArgMax.h | 83 typedef typename TensorEvaluator<ArgType, Device>::Dimensions Dimensions;
84 static const int NumDims = internal::array_size<Dimensions>::value;
98 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const {
208 …sorReductionOp<ReduceOp, Dims, const TensorIndexTupleOp<ArgType> >, Device>::Dimensions Dimensions;
209 …typedef typename TensorEvaluator<const TensorIndexTupleOp<ArgType> , Device>::Dimensions InputDime…
238 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const {
|
| D | TensorShuffling.h | 104 …t int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
105 typedef DSizes<Index, NumDims> Dimensions;
122 const typename TensorEvaluator<ArgType, Device>::Dimensions& input_dims = m_impl.dimensions();
151 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
210 Dimensions m_dimensions;
226 …t int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
227 typedef DSizes<Index, NumDims> Dimensions;
|
| D | TensorConvolution.h | 218 template<typename Dimensions, typename InputXprType, typename KernelXprType>
219 struct traits<TensorConvolutionOp<Dimensions, InputXprType, KernelXprType> >
240 template<typename Dimensions, typename InputXprType, typename KernelXprType>
241 struct eval<TensorConvolutionOp<Dimensions, InputXprType, KernelXprType>, Eigen::Dense>
243 typedef const TensorConvolutionOp<Dimensions, InputXprType, KernelXprType>& type;
246 template<typename Dimensions, typename InputXprType, typename KernelXprType>
247 …ruct nested<TensorConvolutionOp<Dimensions, InputXprType, KernelXprType>, 1, typename eval<TensorC…
249 typedef TensorConvolutionOp<Dimensions, InputXprType, KernelXprType> type;
295 … NumDims = internal::array_size<typename TensorEvaluator<InputArgType, Device>::Dimensions>::value;
298 typedef DSizes<Index, NumDims> Dimensions;
[all …]
|
| D | TensorGenerator.h | 85 typedef typename TensorEvaluator<ArgType, Device>::Dimensions Dimensions;
86 static const int NumDims = internal::array_size<Dimensions>::value;
118 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
178 Dimensions m_dimensions;
|
| D | TensorStorage.h | 34 template<typename T, typename Dimensions, int Options_> class TensorStorage;
74 typedef DSizes<IndexType, NumIndices_> Dimensions; typedef
114 … EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const {return m_dimensions;} in dimensions()
141 Dimensions m_dimensions;
|
| D | TensorIO.h | 61 typedef typename Evaluator::Dimensions Dimensions;
69 static const int rank = internal::array_size<Dimensions>::value;
|
| D | TensorDimensions.h | 43 template <typename Dimensions> EIGEN_DEVICE_FUNC
45 const Dimensions& dimensions) in run()
48 dget<RowMajor ? n - 1 : (NumIndices - n), Dimensions>::value * in run()
56 template <typename Dimensions> EIGEN_DEVICE_FUNC
57 static inline Index run(array<Index, NumIndices> const&, const Dimensions&)
66 template <typename Dimensions> EIGEN_DEVICE_FUNC
68 const Dimensions& dimensions)
79 template <typename Dimensions> EIGEN_DEVICE_FUNC
81 const Dimensions&)
|
| D | TensorInflation.h | 82 …t int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
83 typedef DSizes<Index, NumDims> Dimensions;
112 const typename TensorEvaluator<ArgType, Device>::Dimensions& input_dims = m_impl.dimensions();
130 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
219 Dimensions m_dimensions;
|
| D | TensorCustomOp.h | 85 typedef DSizes<Index, NumDims> Dimensions;
106 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
150 Dimensions m_dimensions;
241 typedef DSizes<Index, NumDims> Dimensions;
262 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
304 Dimensions m_dimensions;
|
| D | TensorReverse.h | 106 typedef DSizes<Index, NumDims> Dimensions;
143 const Dimensions& dimensions() const { return m_dimensions; }
228 Dimensions m_dimensions;
245 typedef DSizes<Index, NumDims> Dimensions;
264 const Dimensions& dimensions() const { return this->m_dimensions; }
|
| D | TensorMap.h | 50 typedef typename PlainObjectType::Dimensions Dimensions; typedef
99 template <typename Dimensions>
100 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorMap(PointerArgType dataPtr, const Dimensions& dimensio… in TensorMap()
113 EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; } in dimensions()
316 Dimensions m_dimensions;
|
| D | TensorLayoutSwap.h | 115 …t int NumDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
116 typedef DSizes<Index, NumDims> Dimensions;
138 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
168 Dimensions m_dimensions;
|
| D | TensorChipping.h | 132 … NumInputDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
135 typedef DSizes<Index, NumDims> Dimensions;
158 const typename TensorEvaluator<ArgType, Device>::Dimensions& input_dims = m_impl.dimensions();
186 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
300 Dimensions m_dimensions;
317 … NumInputDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
320 typedef DSizes<Index, NumDims> Dimensions;
|
| D | TensorMorphing.h | 102 typedef NewDimensions Dimensions;
125 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
167 typedef NewDimensions Dimensions;
324 for (std::size_t i = 0; i < internal::array_size<Dimensions>::value; ++i) {
328 const typename TensorEvaluator<ArgType, Device>::Dimensions& input_dims = m_impl.dimensions();
361 typedef Sizes Dimensions;
363 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dimensions; }
524 Dimensions m_dimensions;
554 typedef Sizes Dimensions;
713 for (size_t i = 0; i < internal::array_size<Dimensions>::value; ++i) {
[all …]
|
| /external/clang/test/SemaCUDA/ |
| D | cxx11-kernel-call.cu | 7 template<int ...Dimensions> void k1Wrapper() { in k1Wrapper()
8 …void (*f)() = [] { k1<<<Dimensions, Dimensions>>>(); }; // expected-error {{initializer contains u… in k1Wrapper()
9 void (*g[])() = { [] { k1<<<Dimensions, Dimensions>>>(); } ... }; // ok in k1Wrapper()
|
| /external/skqp/infra/bots/ |
| D | gen_tasks.go | 316 Dimensions: dimensions,
687 Dimensions: linuxGceDimensions(MACHINE_TYPE_SMALL),
744 Dimensions: linuxGceDimensions(MACHINE_TYPE_SMALL),
1043 task.Dimensions = append(task.Dimensions, "rack:2")
1045 task.Dimensions = append(task.Dimensions, "rack:1")
1139 task.Dimensions = append(task.Dimensions, "valgrind:1")
1191 task.Dimensions = append(task.Dimensions, "valgrind:1")
1207 task.Dimensions = append(task.Dimensions, "rack:2")
1209 task.Dimensions = append(task.Dimensions, "rack:1")
|
| /external/skia/infra/bots/ |
| D | gen_tasks.go | 315 Dimensions: dimensions,
680 Dimensions: linuxGceDimensions(MACHINE_TYPE_SMALL),
737 Dimensions: linuxGceDimensions(MACHINE_TYPE_SMALL),
1026 task.Dimensions = append(task.Dimensions, "rack:2")
1028 task.Dimensions = append(task.Dimensions, "rack:1")
1124 task.Dimensions = append(task.Dimensions, "valgrind:1")
1178 task.Dimensions = append(task.Dimensions, "valgrind:1")
1194 task.Dimensions = append(task.Dimensions, "rack:2")
1196 task.Dimensions = append(task.Dimensions, "rack:1")
|