Home
last modified time | relevance | path

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

123456

/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;
/external/tensorflow/tensorflow/lite/kernels/
Dsparse_to_dense.cc55 switch (NumDimensions(indices)) { in CheckDimensionsMatch()
58 if (NumDimensions(values) == 0) { in CheckDimensionsMatch()
67 if (NumDimensions(values) == 0) in CheckDimensionsMatch()
75 NumDimensions(indices)); in CheckDimensionsMatch()
90 switch (NumDimensions(indices)) { in GetIndicesVector()
122 NumDimensions(indices)); in GetIndicesVector()
162 TF_LITE_ASSERT(NumDimensions(indices) >= 0); in Prepare()
163 TF_LITE_ENSURE(context, NumDimensions(indices) < 3); in Prepare()
164 TF_LITE_ASSERT(NumDimensions(output_shape) >= 0); in Prepare()
165 TF_LITE_ENSURE_EQ(context, NumDimensions(output_shape), 1); in Prepare()
[all …]
Dhashtable_lookup.cc60 TF_LITE_ENSURE_EQ(context, NumDimensions(lookup), 1); in Prepare()
65 TF_LITE_ENSURE_EQ(context, NumDimensions(key), 1); in Prepare()
70 TF_LITE_ENSURE(context, NumDimensions(value) >= 1); in Prepare()
74 TF_LITE_ENSURE_EQ(context, NumDimensions(value), 1); in Prepare()
89 TfLiteIntArray* outputSize = TfLiteIntArrayCreate(NumDimensions(value)); in Prepare()
91 for (int i = 1; i < NumDimensions(value); i++) { in Prepare()
Dshape.cc34 for (int i = 0; i < NumDimensions(input); ++i) { in ExtractShape()
71 output_size->data[0] = NumDimensions(input); in Prepare()
74 TFLITE_DCHECK_EQ(NumDimensions(output), 1); in Prepare()
75 TFLITE_DCHECK_EQ(SizeOfDimension(output, 0), NumDimensions(input)); in Prepare()
Dslice.cc57 for (int idx = 0; idx < NumDimensions(input); ++idx) { in CalculateOutputShapeVector()
132 TF_LITE_ENSURE_EQ(context, NumDimensions(begin), 1); in Prepare()
133 TF_LITE_ENSURE_EQ(context, NumDimensions(size), 1); in Prepare()
135 TF_LITE_ENSURE_MSG(context, NumDimensions(input) <= kMaxDim, in Prepare()
170 for (int i = NumDimensions(input); i < kMaxDim; ++i) { in Eval()
176 GetBeginAndSizeVectors<int32_t>(NumDimensions(input), begin, size, &begins, in Eval()
179 GetBeginAndSizeVectors<int64_t>(NumDimensions(input), begin, size, &begins, in Eval()
Dembedding_lookup_sparse.cc88 TF_LITE_ENSURE_EQ(context, NumDimensions(ids), 1); in Prepare()
93 TF_LITE_ENSURE_EQ(context, NumDimensions(indices), 2); in Prepare()
98 TF_LITE_ENSURE_EQ(context, NumDimensions(shape), 1); in Prepare()
103 TF_LITE_ENSURE_EQ(context, NumDimensions(weights), 1); in Prepare()
113 TF_LITE_ENSURE(context, NumDimensions(value) >= 2); in Prepare()
163 const int embedding_rank = NumDimensions(value); in Eval()
Dembedding_lookup.cc51 TF_LITE_ENSURE_EQ(context, NumDimensions(lookup), 1); in Prepare()
56 TF_LITE_ENSURE(context, NumDimensions(value) >= 2); in Prepare()
60 TfLiteIntArray* outputSize = TfLiteIntArrayCreate(NumDimensions(value)); in Prepare()
64 for (int i = 2; i < NumDimensions(value); i++) { in Prepare()
104 for (int i = 1; i < NumDimensions(value); i++) { in EvalHybrid()
Dunpack.cc46 axis += NumDimensions(input); in Prepare()
48 TF_LITE_ENSURE(context, 0 <= axis && axis < NumDimensions(input)); in Prepare()
60 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(NumDimensions(input) - 1); in Prepare()
62 for (int index = 0; index < NumDimensions(input); ++index) { in Prepare()
Dsplit.cc56 axis_value += NumDimensions(input); in ResizeOutputTensors()
60 TF_LITE_ENSURE(context, axis_value < NumDimensions(input)); in ResizeOutputTensors()
120 axis_value += NumDimensions(op_context.input); in Eval()
124 TF_LITE_ENSURE(context, axis_value < NumDimensions(op_context.input)); in Eval()
/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions.h62 internal::traits<OutputBackward>::NumDimensions>,
82 internal::traits<OutputBackward>::NumDimensions>,
108 internal::traits<Kernel>::NumDimensions,
111 TensorRef<Tensor<OutScalar, internal::traits<OutputBackward>::NumDimensions,
122 static const int NumDims = internal::traits<OutputBackward>::NumDimensions;
337 internal::traits<Input>::NumDimensions>,
340 internal::traits<Input>::NumDimensions>,
343 internal::traits<Input>::NumDimensions>,
354 internal::traits<Input>::NumDimensions>,
365 internal::traits<Input>::NumDimensions>,
[all …]
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 …]
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/
Dwhitespace_tokenizer.cc95 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(NumDimensions(input) + 1); in WritePaddedOutput()
96 for (int i = 0; i < NumDimensions(input); ++i) { in WritePaddedOutput()
105 output_shape->data[NumDimensions(input)] = max_tokens; in WritePaddedOutput()
161 TF_LITE_ENSURE(context, NumDimensions(input) == in Prepare()
168 for (int i = 0; i < NumDimensions(input); ++i) { in Prepare()
185 for (int i = 0; i < NumDimensions(input); ++i) { in Eval()
203 nested_row_splits.reserve(NumDimensions(input)); in Eval()
204 for (int i = 0; i < NumDimensions(input); ++i) { in Eval()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dbatch_to_space_nd.cc48 TF_LITE_ENSURE(context, NumDimensions(input) >= kInputOutputMinDimensionNum); in Prepare()
49 TF_LITE_ENSURE(context, NumDimensions(output) >= kInputOutputMinDimensionNum); in Prepare()
50 TF_LITE_ENSURE(context, NumDimensions(input) <= kInputOutputMaxDimensionNum); in Prepare()
51 TF_LITE_ENSURE(context, NumDimensions(output) <= kInputOutputMaxDimensionNum); in Prepare()
Dspace_to_batch_nd.cc54 TF_LITE_ENSURE(context, NumDimensions(input) >= kInputOutputMinDimensionNum); in Prepare()
55 TF_LITE_ENSURE(context, NumDimensions(output) >= kInputOutputMinDimensionNum); in Prepare()
56 TF_LITE_ENSURE(context, NumDimensions(input) <= kInputOutputMaxDimensionNum); in Prepare()
57 TF_LITE_ENSURE(context, NumDimensions(output) <= kInputOutputMaxDimensionNum); in Prepare()
/external/openscreen/third_party/abseil/src/absl/types/
Dvariant_benchmark.cc143 template <std::size_t NumIndices, std::size_t NumDimensions = 1>
146 MakeVariantTuple<NumIndices, absl::make_index_sequence<NumDimensions>>:: in BM_RedundantVisit()
156 template <std::size_t NumIndices, std::size_t NumDimensions = 1>
159 integral_pow(NumIndices, NumDimensions); in BM_Visit()
162 MakeVariantTuple<NumIndices, absl::make_index_sequence<NumDimensions>>; in BM_Visit()
/external/rust/crates/grpcio-sys/grpc/third_party/abseil-cpp/absl/types/
Dvariant_benchmark.cc143 template <std::size_t NumIndices, std::size_t NumDimensions = 1>
146 MakeVariantTuple<NumIndices, absl::make_index_sequence<NumDimensions>>:: in BM_RedundantVisit()
156 template <std::size_t NumIndices, std::size_t NumDimensions = 1>
159 integral_pow(NumIndices, NumDimensions); in BM_Visit()
162 MakeVariantTuple<NumIndices, absl::make_index_sequence<NumDimensions>>; in BM_Visit()
/external/webrtc/third_party/abseil-cpp/absl/types/
Dvariant_benchmark.cc143 template <std::size_t NumIndices, std::size_t NumDimensions = 1>
146 MakeVariantTuple<NumIndices, absl::make_index_sequence<NumDimensions>>:: in BM_RedundantVisit()
156 template <std::size_t NumIndices, std::size_t NumDimensions = 1>
159 integral_pow(NumIndices, NumDimensions); in BM_Visit()
162 MakeVariantTuple<NumIndices, absl::make_index_sequence<NumDimensions>>; in BM_Visit()
/external/abseil-cpp/absl/types/
Dvariant_benchmark.cc143 template <std::size_t NumIndices, std::size_t NumDimensions = 1>
146 MakeVariantTuple<NumIndices, absl::make_index_sequence<NumDimensions>>:: in BM_RedundantVisit()
156 template <std::size_t NumIndices, std::size_t NumDimensions = 1>
159 integral_pow(NumIndices, NumDimensions); in BM_Visit()
162 MakeVariantTuple<NumIndices, absl::make_index_sequence<NumDimensions>>; in BM_Visit()
/external/libtextclassifier/abseil-cpp/absl/types/
Dvariant_benchmark.cc143 template <std::size_t NumIndices, std::size_t NumDimensions = 1>
146 MakeVariantTuple<NumIndices, absl::make_index_sequence<NumDimensions>>:: in BM_RedundantVisit()
156 template <std::size_t NumIndices, std::size_t NumDimensions = 1>
159 integral_pow(NumIndices, NumDimensions); in BM_Visit()
162 MakeVariantTuple<NumIndices, absl::make_index_sequence<NumDimensions>>; in BM_Visit()
/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())
/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;

123456