Searched refs:NumDimensions (Results 1 - 25 of 73) sorted by relevance

123

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorInitializer.h32 Eigen::array<typename traits<Derived>::Index, traits<Derived>::NumDimensions>* indices,
36 (*indices)[traits<Derived>::NumDimensions - N] = i++;
47 Eigen::array<typename traits<Derived>::Index, traits<Derived>::NumDimensions>* indices,
52 (*indices)[traits<Derived>::NumDimensions - 1] = i++;
63 Eigen::array<typename traits<Derived>::Index, traits<Derived>::NumDimensions>*,
72 const typename Initializer<Derived, traits<Derived>::NumDimensions>::InitList& vals) {
73 Eigen::array<typename traits<Derived>::Index, traits<Derived>::NumDimensions> indices;
74 Initializer<Derived, traits<Derived>::NumDimensions>::run(tensor, &indices, vals);
H A DTensorTraits.h53 static const int NumDimensions = NumIndices_; member in struct:Eigen::internal::traits
71 static const int NumDimensions = array_size<Dimensions>::value; member in struct:Eigen::internal::traits
91 static const int NumDimensions = BaseTraits::NumDimensions; member in struct:Eigen::internal::traits
112 static const int NumDimensions = BaseTraits::NumDimensions; member in struct:Eigen::internal::traits
H A DTensorBase.h34 static const int NumDimensions = DerivedTraits::NumDimensions; member in class:Eigen::TensorBase
530 const TensorReductionOp<internal::SumReducer<CoeffReturnType>, const DimensionList<Index, NumDimensions>, const Derived>
532 DimensionList<Index, NumDimensions> in_dims;
533 return TensorReductionOp<internal::SumReducer<CoeffReturnType>, const DimensionList<Index, NumDimensions>, const Derived>(derived(), in_dims, internal::SumReducer<CoeffReturnType>());
542 const TensorReductionOp<internal::MeanReducer<CoeffReturnType>, const DimensionList<Index, NumDimensions>, const Derived>
544 DimensionList<Index, NumDimensions> in_dims;
545 return TensorReductionOp<internal::MeanReducer<CoeffReturnType>, const DimensionList<Index, NumDimensions>, const Derived>(derived(), in_dims, internal::MeanReducer<CoeffReturnType>());
554 const TensorReductionOp<internal::ProdReducer<CoeffReturnType>, const DimensionList<Index, NumDimensions>, const Derived>
556 DimensionList<Index, NumDimensions> in_dim
833 static const int NumDimensions = DerivedTraits::NumDimensions; member in class:Eigen::TensorBase
[all...]
H A DTensorExpr.h39 static const int NumDimensions = XprTraits::NumDimensions; member in struct:Eigen::internal::traits
90 static const int NumDimensions = XprTraits::NumDimensions; member in struct:Eigen::internal::traits
162 static const int NumDimensions = XprTraits::NumDimensions; member in struct:Eigen::internal::traits
239 static const int NumDimensions = XprTraits::NumDimensions; member in struct:Eigen::internal::traits
315 static const int NumDimensions = XprTraits::NumDimensions; member in struct:Eigen::internal::traits
[all...]
H A DTensorCustomOp.h31 static const int NumDimensions = traits<XprType>::NumDimensions; member in struct:Eigen::internal::traits
84 static const int NumDims = internal::traits<ArgType>::NumDimensions;
181 static const int NumDimensions = traits<LhsXprType>::NumDimensions; member in struct:Eigen::internal::traits
240 static const int NumDims = internal::traits<XprType>::NumDimensions;
/external/tensorflow/tensorflow/contrib/lite/kernels/
H A Dembedding_lookup.cc55 TF_LITE_ENSURE_EQ(context, NumDimensions(lookup), 1);
59 TF_LITE_ENSURE(context, NumDimensions(value) >= 2);
62 TfLiteIntArray* outputSize = TfLiteIntArrayCreate(NumDimensions(value));
66 for (int i = 2; i < NumDimensions(value); i++) {
H A Dhashtable_lookup.cc64 TF_LITE_ENSURE_EQ(context, NumDimensions(lookup), 1);
68 TF_LITE_ENSURE_EQ(context, NumDimensions(key), 1);
72 TF_LITE_ENSURE(context, NumDimensions(value) >= 1);
76 TF_LITE_ENSURE_EQ(context, NumDimensions(value), 1);
89 TfLiteIntArray* outputSize = TfLiteIntArrayCreate(NumDimensions(value));
91 for (int i = 1; i < NumDimensions(value); i++) {
H A Dgather.cc46 TF_LITE_ENSURE(context, NumDimensions(positions) <= 1);
59 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 1);
67 NumDimensions(input) + NumDimensions(positions) - 1;
87 const int input_rank = NumDimensions(input);
H A Ddiv.cc48 TF_LITE_ENSURE_EQ(context, NumDimensions(input1), NumDimensions(input2));
49 for (int i = 0; i < NumDimensions(input1); ++i) {
H A Dsub.cc48 TF_LITE_ENSURE_EQ(context, NumDimensions(input1), NumDimensions(input2));
49 for (int i = 0; i < NumDimensions(input1); ++i) {
H A Dsplit.cc53 axis_value += NumDimensions(input);
109 axis_value += NumDimensions(op_context.input);
111 axis_value = RemapDim(NumDimensions(op_context.input), axis_value);
119 if (axis_value == NumDimensions(op_context.input)) { \
H A Dkernel_util.h23 inline int NumDimensions(const TfLiteTensor* t) { return t->dims->size; } function in namespace:tflite
40 for (int i = 0; i < NumDimensions(t); ++i) {
H A Dembedding_lookup_sparse.cc85 TF_LITE_ENSURE_EQ(context, NumDimensions(ids), 1);
89 TF_LITE_ENSURE_EQ(context, NumDimensions(indices), 2);
93 TF_LITE_ENSURE_EQ(context, NumDimensions(shape), 1);
97 TF_LITE_ENSURE_EQ(context, NumDimensions(weights), 1);
106 TF_LITE_ENSURE(context, NumDimensions(value) >= 2);
149 const int embedding_rank = NumDimensions(value);
H A Dbatch_to_space_nd.cc62 TF_LITE_ENSURE_EQ(context, NumDimensions(op_context->block_shape),
66 TF_LITE_ENSURE_EQ(context, NumDimensions(op_context->crops),
100 TF_LITE_ENSURE_EQ(context, NumDimensions(op_context.input),
H A Dlsh_projection.cc81 TF_LITE_ENSURE_EQ(context, NumDimensions(hash), 2);
86 TF_LITE_ENSURE(context, NumDimensions(input) >= 1);
90 TF_LITE_ENSURE_EQ(context, NumDimensions(weight), 1);
H A Dspace_to_batch_nd.cc62 TF_LITE_ENSURE_EQ(context, NumDimensions(op_context->block_shape),
66 TF_LITE_ENSURE_EQ(context, NumDimensions(op_context->paddings),
95 TF_LITE_ENSURE_EQ(context, NumDimensions(op_context.input),
H A Dtranspose.cc47 int dims = NumDimensions(op_context->input);
51 TF_LITE_ENSURE_EQ(context, NumDimensions(op_context->perm), 1);
75 TF_LITE_ENSURE_MSG(context, NumDimensions(op_context.input) <= 4,
H A Dactivations.cc107 NumDimensions(input) == 2 || NumDimensions(input) == 4);
308 if (NumDimensions(input) == 2) {
312 if (NumDimensions(input) == 4) {
321 if (NumDimensions(input) == 2) {
325 if (NumDimensions(input) == 4) {
/external/tensorflow/tensorflow/go/
H A Dshape_test.go53 if got, want := test.shape.NumDimensions(), len(test.slice); got != want {
72 if s.NumDimensions() != -1 {
73 t.Error(s.NumDimensions())
H A Dshape.go48 // NumDimensions returns the number of dimensions represented by s, or -1 if
50 func (s Shape) NumDimensions() int {
60 // REQUIRES: 0 <= dim < s.NumDimensions()
62 if dim < 0 || dim >= s.NumDimensions() {
/external/tensorflow/tensorflow/core/kernels/
H A Deigen_pooling.h43 internal::traits<Input>::NumDimensions>,
53 internal::traits<Input>::NumDimensions>,
68 EIGEN_STATIC_ASSERT(internal::traits<Input>::NumDimensions == 4,
73 internal::traits<Input>::NumDimensions,
92 Eigen::DSizes<TensorIndex, internal::traits<Input>::NumDimensions>
161 const Eigen::DSizes<DenseIndex, internal::traits<Input>::NumDimensions>,
171 const Eigen::DSizes<DenseIndex, internal::traits<Input>::NumDimensions>,
184 EIGEN_STATIC_ASSERT(internal::traits<Input>::NumDimensions == 5,
190 internal::traits<Input>::NumDimensions,
205 Eigen::DSizes<DenseIndex, internal::traits<Input>::NumDimensions>
[all...]
H A Deigen_cuboid_convolution.h50 internal::traits<Input>::NumDimensions>,
62 internal::traits<Input>::NumDimensions>,
79 internal::traits<Input>::NumDimensions,
83 internal::traits<Kernel>::NumDimensions,
91 static const int NumDims = internal::traits<Input>::NumDimensions;
H A Deigen_backward_cuboid_convolutions.h51 internal::traits<OutputBackward>::NumDimensions>,
66 internal::traits<OutputBackward>::NumDimensions>,
89 internal::traits<Kernel>::NumDimensions,
94 internal::traits<OutputBackward>::NumDimensions,
105 static const int NumDims = internal::traits<OutputBackward>::NumDimensions;
376 internal::traits<Input>::NumDimensions,
380 internal::traits<OutputBackward>::NumDimensions,
390 static const int NumDims = internal::traits<Input>::NumDimensions;
391 EIGEN_STATIC_ASSERT(internal::traits<Input>::NumDimensions ==
392 internal::traits<OutputBackward>::NumDimensions,
[all...]
H A Deigen_backward_spatial_convolutions.h62 internal::traits<OutputBackward>::NumDimensions>,
81 internal::traits<OutputBackward>::NumDimensions>,
107 internal::traits<Kernel>::NumDimensions,
110 TensorRef<Tensor<OutScalar, internal::traits<OutputBackward>::NumDimensions,
121 static const int NumDims = internal::traits<OutputBackward>::NumDimensions;
359 internal::traits<Input>::NumDimensions,
362 TensorRef<Tensor<OutScalar, internal::traits<OutputBackward>::NumDimensions,
376 static const int NumDims = internal::traits<Input>::NumDimensions;
377 EIGEN_STATIC_ASSERT(internal::traits<Input>::NumDimensions ==
378 internal::traits<OutputBackward>::NumDimensions,
[all...]
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
H A Deigen_spatial_convolutions.h74 internal::traits<Input>::NumDimensions>,
87 internal::traits<Input>::NumDimensions>,
105 internal::traits<Input>::NumDimensions,
109 internal::traits<Kernel>::NumDimensions,
118 const int NumDims = internal::traits<Input>::NumDimensions;

Completed in 3149 milliseconds

123