/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
H A D | TensorInitializer.h | 32 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 D | TensorTraits.h | 53 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 D | TensorBase.h | 34 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 D | TensorExpr.h | 39 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 D | TensorCustomOp.h | 31 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 D | embedding_lookup.cc | 55 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 D | hashtable_lookup.cc | 64 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 D | gather.cc | 46 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 D | div.cc | 48 TF_LITE_ENSURE_EQ(context, NumDimensions(input1), NumDimensions(input2)); 49 for (int i = 0; i < NumDimensions(input1); ++i) {
|
H A D | sub.cc | 48 TF_LITE_ENSURE_EQ(context, NumDimensions(input1), NumDimensions(input2)); 49 for (int i = 0; i < NumDimensions(input1); ++i) {
|
H A D | split.cc | 53 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 D | kernel_util.h | 23 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 D | embedding_lookup_sparse.cc | 85 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 D | batch_to_space_nd.cc | 62 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 D | lsh_projection.cc | 81 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 D | space_to_batch_nd.cc | 62 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 D | transpose.cc | 47 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 D | activations.cc | 107 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 D | shape_test.go | 53 if got, want := test.shape.NumDimensions(), len(test.slice); got != want { 72 if s.NumDimensions() != -1 { 73 t.Error(s.NumDimensions())
|
H A D | shape.go | 48 // 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 D | eigen_pooling.h | 43 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 D | eigen_cuboid_convolution.h | 50 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 D | eigen_backward_cuboid_convolutions.h | 51 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 D | eigen_backward_spatial_convolutions.h | 62 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 D | eigen_spatial_convolutions.h | 74 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;
|