Home
last modified time | relevance | path

Searched refs:NumDimensions (Results 1 – 25 of 102) sorted by relevance

12345

/external/tensorflow/tensorflow/lite/kernels/
Dsparse_to_dense.cc59 switch (NumDimensions(indices)) { in CheckDimensionsMatch()
62 if (NumDimensions(values) == 0) { in CheckDimensionsMatch()
71 if (NumDimensions(values) == 0) in CheckDimensionsMatch()
79 NumDimensions(indices)); in CheckDimensionsMatch()
94 switch (NumDimensions(indices)) { in GetIndicesVector()
126 NumDimensions(indices)); in GetIndicesVector()
160 TF_LITE_ASSERT(NumDimensions(indices) >= 0); in Prepare()
161 TF_LITE_ENSURE(context, NumDimensions(indices) < 3); in Prepare()
162 TF_LITE_ASSERT(NumDimensions(output_shape) >= 0); in Prepare()
163 TF_LITE_ENSURE_EQ(context, NumDimensions(output_shape), 1); in Prepare()
[all …]
Dunpack.cc53 TF_LITE_ENSURE(context, NumDimensions(input) <= 4); in Prepare()
54 TF_LITE_ENSURE(context, NumDimensions(input) > 1); in Prepare()
57 axis += NumDimensions(input); in Prepare()
59 TF_LITE_ENSURE(context, 0 <= axis && axis < NumDimensions(input)); in Prepare()
69 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(NumDimensions(input) - 1); in Prepare()
71 for (int index = 0; index < NumDimensions(input); ++index) { in Prepare()
Dhashtable_lookup.cc63 TF_LITE_ENSURE_EQ(context, NumDimensions(lookup), 1); in Prepare()
67 TF_LITE_ENSURE_EQ(context, NumDimensions(key), 1); in Prepare()
71 TF_LITE_ENSURE(context, NumDimensions(value) >= 1); in Prepare()
75 TF_LITE_ENSURE_EQ(context, NumDimensions(value), 1); in Prepare()
88 TfLiteIntArray* outputSize = TfLiteIntArrayCreate(NumDimensions(value)); in Prepare()
90 for (int i = 1; i < NumDimensions(value); i++) { in Prepare()
Dshape.cc31 for (int i = 0; i < NumDimensions(input); ++i) { in ExtractShape()
60 output_size->data[0] = NumDimensions(input); in Prepare()
67 TFLITE_DCHECK_EQ(NumDimensions(output), 1); in Eval()
68 TFLITE_DCHECK_EQ(SizeOfDimension(output, 0), NumDimensions(input)); in Eval()
Dslice.cc50 for (int idx = 0; idx < NumDimensions(input); ++idx) { in CalculateOutputShapeVector()
120 TF_LITE_ENSURE_EQ(context, NumDimensions(begin), 1); in Prepare()
121 TF_LITE_ENSURE_EQ(context, NumDimensions(size), 1); in Prepare()
122 TF_LITE_ENSURE_MSG(context, NumDimensions(input) <= kMaxDim, in Prepare()
153 GetBeginAndSizeVectors<int32_t>(NumDimensions(input), begin, size, &begins, in Eval()
156 GetBeginAndSizeVectors<int64_t>(NumDimensions(input), begin, size, &begins, in Eval()
164 for (int i = NumDimensions(input); i < kMaxDim; ++i) { in Eval()
Dembedding_lookup_sparse.cc85 TF_LITE_ENSURE_EQ(context, NumDimensions(ids), 1); in Prepare()
89 TF_LITE_ENSURE_EQ(context, NumDimensions(indices), 2); in Prepare()
93 TF_LITE_ENSURE_EQ(context, NumDimensions(shape), 1); in Prepare()
97 TF_LITE_ENSURE_EQ(context, NumDimensions(weights), 1); in Prepare()
106 TF_LITE_ENSURE(context, NumDimensions(value) >= 2); in Prepare()
149 const int embedding_rank = NumDimensions(value); in Eval()
Dembedding_lookup.cc55 TF_LITE_ENSURE_EQ(context, NumDimensions(lookup), 1); in Prepare()
59 TF_LITE_ENSURE(context, NumDimensions(value) >= 2); in Prepare()
62 TfLiteIntArray* outputSize = TfLiteIntArrayCreate(NumDimensions(value)); in Prepare()
66 for (int i = 2; i < NumDimensions(value); i++) { in Prepare()
103 for (int i = 1; i < NumDimensions(value); i++) { in EvalHybrid()
Dgather.cc66 TF_LITE_ENSURE_EQ(context, NumDimensions(input), 1); in Prepare()
76 axis += NumDimensions(input); in Prepare()
78 TF_LITE_ENSURE(context, 0 <= axis && axis < NumDimensions(input)); in Prepare()
81 NumDimensions(input) + NumDimensions(positions) - 1; in Prepare()
Dreverse_sequence.cc38 TF_LITE_ENSURE_EQ(context, NumDimensions(seq_lengths), 1); in Prepare()
78 TF_LITE_ENSURE(context, seq_dim < NumDimensions(input)); in ReverseSequenceImpl()
79 TF_LITE_ENSURE(context, batch_dim < NumDimensions(input)); in ReverseSequenceImpl()
82 for (int i = 0; i < NumDimensions(seq_lengths_tensor); ++i) { in ReverseSequenceImpl()
Dmirror_pad.cc141 const int input_dims = NumDimensions(input); in GetPaddedOutputShape()
165 const int input_dims = NumDimensions(input_tensor); in Eval()
245 TF_LITE_ENSURE_EQ(context, NumDimensions(padding_matrix), 2); in Prepare()
247 NumDimensions(input_tensor)); in Prepare()
249 int num_elements = NumElements(input_tensor) * NumDimensions(input_tensor); in Prepare()
Darg_min_max.cc36 axis_value += NumDimensions(input); in ResizeOutput()
40 TfLiteIntArray* output_dims = TfLiteIntArrayCreate(NumDimensions(input) - 1); in ResizeOutput()
42 for (int i = 0; i < NumDimensions(input); ++i) { in ResizeOutput()
95 TF_LITE_ENSURE(context, NumDimensions(input) >= 1); in Prepare()
Dactivations.cc245 const int num_dims = NumDimensions(input); in SoftmaxPrepare()
750 if (NumDimensions(input) == 1) { in SoftmaxEval()
754 if (NumDimensions(input) == 2) { in SoftmaxEval()
758 if (NumDimensions(input) == 3) { in SoftmaxEval()
762 if (NumDimensions(input) == 4) { in SoftmaxEval()
768 NumDimensions(input)); in SoftmaxEval()
772 if (NumDimensions(input) == 1) { in SoftmaxEval()
776 if (NumDimensions(input) == 2) { in SoftmaxEval()
780 if (NumDimensions(input) == 3) { in SoftmaxEval()
784 if (NumDimensions(input) == 4) { in SoftmaxEval()
[all …]
Dreverse.cc38 TF_LITE_ENSURE_EQ(context, NumDimensions(axis), 1); in Prepare()
39 TF_LITE_ENSURE(context, NumDimensions(input) >= NumElements(axis)); in Prepare()
72 TF_LITE_ENSURE(context, axis >= 0 && axis < NumDimensions(input)); in Eval()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorInitializer.h32 … Eigen::array<typename traits<Derived>::Index, traits<Derived>::NumDimensions>* indices, in run()
36 (*indices)[traits<Derived>::NumDimensions - N] = i++; in run()
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);
DTensorBase.h34 static const int NumDimensions = DerivedTraits::NumDimensions; variable
530 …nOp<internal::SumReducer<CoeffReturnType>, const DimensionList<Index, NumDimensions>, const Derive…
532 DimensionList<Index, NumDimensions> in_dims; in sum()
533 …nOp<internal::SumReducer<CoeffReturnType>, const DimensionList<Index, NumDimensions>, const Derive… in sum()
542 …Op<internal::MeanReducer<CoeffReturnType>, const DimensionList<Index, NumDimensions>, const Derive…
544 DimensionList<Index, NumDimensions> in_dims; in mean()
545 …Op<internal::MeanReducer<CoeffReturnType>, const DimensionList<Index, NumDimensions>, const Derive… in mean()
554 …Op<internal::ProdReducer<CoeffReturnType>, const DimensionList<Index, NumDimensions>, const Derive…
556 DimensionList<Index, NumDimensions> in_dims; in prod()
557 …Op<internal::ProdReducer<CoeffReturnType>, const DimensionList<Index, NumDimensions>, const Derive… in prod()
[all …]
DTensorExpr.h39 static const int NumDimensions = XprTraits::NumDimensions;
90 static const int NumDimensions = XprTraits::NumDimensions;
162 static const int NumDimensions = XprTraits::NumDimensions;
239 static const int NumDimensions = XprTraits::NumDimensions;
315 static const int NumDimensions = XprTraits::NumDimensions;
DTensorTraits.h53 static const int NumDimensions = NumIndices_;
71 static const int NumDimensions = array_size<Dimensions>::value;
91 static const int NumDimensions = BaseTraits::NumDimensions;
112 static const int NumDimensions = BaseTraits::NumDimensions;
DTensorCustomOp.h31 static const int NumDimensions = traits<XprType>::NumDimensions;
84 static const int NumDims = internal::traits<ArgType>::NumDimensions;
181 static const int NumDimensions = traits<LhsXprType>::NumDimensions;
240 static const int NumDims = internal::traits<XprType>::NumDimensions;
/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_cuboid_convolutions.h51 internal::traits<OutputBackward>::NumDimensions>,
70 internal::traits<OutputBackward>::NumDimensions>,
96 internal::traits<Kernel>::NumDimensions,
101 internal::traits<OutputBackward>::NumDimensions,
112 static const int NumDims = internal::traits<OutputBackward>::NumDimensions;
329 internal::traits<Input>::NumDimensions>,
332 internal::traits<Input>::NumDimensions>,
335 internal::traits<Input>::NumDimensions>,
345 internal::traits<Input>::NumDimensions>,
356 internal::traits<Input>::NumDimensions>,
[all …]
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, in CuboidMaxPooling()
190 internal::traits<Input>::NumDimensions, in CuboidMaxPooling()
205 Eigen::DSizes<DenseIndex, internal::traits<Input>::NumDimensions> in CuboidMaxPooling()
[all …]
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;
357 internal::traits<Input>::NumDimensions,
360 TensorRef<Tensor<OutScalar, internal::traits<OutputBackward>::NumDimensions,
374 static const int NumDims = internal::traits<Input>::NumDimensions;
375 EIGEN_STATIC_ASSERT(internal::traits<Input>::NumDimensions ==
376 internal::traits<OutputBackward>::NumDimensions,
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/
Dsoftmax.cc160 if (NumDimensions(input) == 1) { in SoftmaxEval()
164 if (NumDimensions(input) == 2) { in SoftmaxEval()
168 if (NumDimensions(input) == 4) { in SoftmaxEval()
174 NumDimensions(input)); in SoftmaxEval()
178 if (NumDimensions(input) == 1) { in SoftmaxEval()
182 if (NumDimensions(input) == 2) { in SoftmaxEval()
186 if (NumDimensions(input) == 4) { in SoftmaxEval()
192 NumDimensions(input)); in SoftmaxEval()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
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;
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Deigen_spatial_convolutions.h71 internal::traits<Input>::NumDimensions>,
84 internal::traits<Input>::NumDimensions>,
102 internal::traits<Input>::NumDimensions,
106 internal::traits<Kernel>::NumDimensions,
115 const int NumDims = internal::traits<Input>::NumDimensions;
/external/tensorflow/tensorflow/go/
Dshape_test.go53 if got, want := test.shape.NumDimensions(), len(test.slice); got != want {
72 if s.NumDimensions() != -1 {
73 t.Error(s.NumDimensions())

12345