| /external/tensorflow/tensorflow/core/kernels/ |
| sparse_utils.cc | 31 Tindices end = indices_mat.dimension(0);
80 const Tindices num_entries_in_sparse_tensor = indices_mat.dimension(0);
121 for (size_t i = 0; i < vec.dimension(0); ++i) {
|
| eigen_attention.h | 40 * batch dimension. The normalized boolean indicates if incoming coordinates are
42 * height and width dimension. The centered boolean indicates if incoming
44 * correspond to minimum and maximum of each dimension while 0.0 corresponds to
77 dims[0] = in.dimension(0);
80 dims[3] = in.dimension(3);
92 const Index num_channels = in.dimension(0);
93 const Index input_width = in.dimension(1);
94 const Index input_height = in.dimension(2);
95 const Index batch_size = in.dimension(3);
|
| unique_op.cc | 136 for (Eigen::Index i = 0; i < Tin.dimension(0); i++) {
137 for (Eigen::Index j = 0; j < Tin.dimension(2); j++) {
146 for (Eigen::Index i = 0; i < Tin.dimension(0); i++) {
147 for (Eigen::Index j = 0; j < Tin.dimension(2); j++) {
159 uniq.reserve(2 * Tin.dimension(1));
161 for (int64 i = 0, j = 0; i < Tin.dimension(1); ++i) {
|
| conv_2d_gpu.h | 125 // A dimension type with compile-time known size.
127 struct Dimension : Array<int, IndexCount, 1> {
129 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Dimension() : Base() {}
130 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Dimension(int a0) : Base(a0) {}
131 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Dimension(int a0, int a1)
133 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Dimension(int a0, int a1, int a2)
135 EIGEN_STRONG_INLINE Dimension(const std::array<int, IndexCount>& array)
153 const Index<IndexCount>& index, const Dimension<IndexCount>& dims) {
164 int index, const Dimension<IndexCount>& dims) {
176 // <sp0, sp1, sp2> shuffles dimensions such that input dimension 0 goes to sp0
[all...] |
| cwise_op_gpu_select.cu.cc | 48 const int size = then_flat.dimension(0);
65 const int all_but_batch = then_flat_outer_dims.dimension(1);
|
| resize_bilinear_op_test.cc | 70 const int batch = images.dimension(0);
71 const int64 in_height = images.dimension(1);
72 const int64 in_width = images.dimension(2);
73 const int channels = images.dimension(3);
75 ASSERT_EQ(batch, output.dimension(0));
76 ASSERT_EQ(channels, output.dimension(3));
78 const int64 out_height = output.dimension(1);
79 const int64 out_width = output.dimension(2);
|
| sparse_tensor_dense_matmul_op.cc | 89 "Cannot multiply A and B because inner dimension does not match: ",
252 const std::size_t rhs_right = (ADJ_B ? b.dimension(0) : b.dimension(1));
253 const std::size_t lhs_right = (ADJ_B ? b.dimension(1) : b.dimension(0));
273 if (!FastBoundsCheck(m, out.dimension(0))) {
274 return MOutOfBoundsError(m, i, lhs_index_a, out.dimension(0));
294 if (!FastBoundsCheck(m, out.dimension(0))) { \
295 return MOutOfBoundsError(m, i, lhs_index_a, out.dimension(0)); \
304 Eigen::array<int, 2> shuffle(1, 0); // preserve dimension orde
[all...] |
| scale_and_translate_op_test.cc | 99 const int64 in_height = images.dimension(1);
100 const int64 in_width = images.dimension(2);
101 const int channels = images.dimension(3);
166 const int batch = images.dimension(0);
167 const int channels = images.dimension(3);
169 ASSERT_EQ(batch, output.dimension(0));
170 ASSERT_EQ(channels, output.dimension(3));
172 const int64 out_height = output.dimension(1);
173 const int64 out_width = output.dimension(2);
174 const int64 in_height = images.dimension(1)
[all...] |
| depthtospace_op.cc | 97 errors::InvalidArgument("Input depth dimension ", input_depth,
153 const int batch_size = output.dimension(0);
154 const int output_height = output.dimension(1);
155 const int output_width = output.dimension(2);
156 const int output_depth = output.dimension(3);
|
| gather_functor.h | 44 const SliceIndex indices_size = static_cast<SliceIndex>(indices.dimension(0));
45 const SliceIndex batch_size = static_cast<SliceIndex>(params.dimension(0));
46 const Index limit = static_cast<Index>(params.dimension(1));
119 const int64 slice_size = out.dimension(2);
|
| matrix_diag_op.cc | 150 for (int64 r = 0; r < output.dimension(0); ++r) {
151 for (int64 d = 0; d < output.dimension(1); ++d) {
163 for (int64 r = 0; r < output.dimension(0); ++r) {
164 for (int64 d = 0; d < output.dimension(1); ++d) {
|
| parameterized_truncated_normal_op_gpu.cu.cc | 248 num_elements, means.data(), means.dimension(0) == 1, stddevs.data(),
249 stddevs.dimension(0) == 1, minvals.data(), minvals.dimension(0) == 1,
250 maxvals.data(), maxvals.dimension(0) == 1, kMaxIterations));
|
| /external/eigen/unsupported/test/ |
| cxx11_tensor_scan.cpp | 25 VERIFY_IS_EQUAL(tensor.dimension(0), result.dimension(0));
|
| cxx11_tensor_contraction.cpp | 167 VERIFY_IS_EQUAL(result.dimension(0), 5);
168 VERIFY_IS_EQUAL(result.dimension(1), 7);
169 VERIFY_IS_EQUAL(result.dimension(2), 7);
170 VERIFY_IS_EQUAL(result.dimension(3), 11);
171 VERIFY_IS_EQUAL(result.dimension(4), 13);
202 VERIFY_IS_EQUAL(result.dimension(0), 2);
211 VERIFY_IS_EQUAL(result.dimension(0), 2);
236 VERIFY_IS_EQUAL(result.dimension(0), 2);
237 VERIFY_IS_EQUAL(result.dimension(1), 2);
476 VERIFY_IS_EQUAL(result.dimension(0), 2)
[all...] |
| cxx11_tensor_convolution.cpp | 89 VERIFY_IS_EQUAL(valid.dimension(0), 1);
97 VERIFY_IS_EQUAL(same.dimension(0), 3);
107 VERIFY_IS_EQUAL(full.dimension(0), 5);
132 VERIFY_IS_EQUAL(result.dimension(0), 2);
|
| /external/mesa3d/src/gallium/auxiliary/tgsi/ |
| tgsi_build.c | 108 declaration.Dimension = 0;
125 unsigned dimension,
143 declaration.Dimension = dimension;
424 full_decl->Declaration.Dimension,
444 if (full_decl->Declaration.Dimension) {
849 src_register.Dimension = 0;
865 unsigned dimension,
888 src_register.Dimension = dimension;
937 struct tgsi_dimension dimension; local
967 struct tgsi_dimension dimension; local
[all...] |
| /external/virglrenderer/src/gallium/auxiliary/tgsi/ |
| tgsi_build.c | 108 declaration.Dimension = 0;
125 unsigned dimension,
143 declaration.Dimension = dimension;
413 full_decl->Declaration.Dimension,
433 if (full_decl->Declaration.Dimension) {
831 src_register.Dimension = 0;
847 unsigned dimension,
870 src_register.Dimension = dimension;
919 struct tgsi_dimension dimension; local
949 struct tgsi_dimension dimension; local
[all...] |
| /external/webrtc/talk/app/webrtc/java/android/org/webrtc/ |
| GlShader.java | 96 * |buffer| with |dimension| number of components per vertex.
98 public void setVertexAttribArray(String label, int dimension, FloatBuffer buffer) {
104 GLES20.glVertexAttribPointer(location, dimension, GLES20.GL_FLOAT, false, 0, buffer);
|
| /external/webrtc/webrtc/modules/audio_processing/vad/ |
| pitch_based_vad.cc | 26 "noise and voice gmm dimension not equal");
49 noise_gmm_.dimension = kNoiseGmmDim;
56 voice_gmm_.dimension = kVoiceGmmDim;
|
| /external/deqp/modules/gles31/functional/ |
| es31fFramebufferDefaultStateQueryTests.cpp | 109 FramebufferDimensionTest (Context& context, QueryType verifier, DimensionType dimension, const char* name, const char* desc);
117 FramebufferDimensionTest::FramebufferDimensionTest (Context& context, QueryType verifier, DimensionType dimension, const char* name, const char* desc)
119 , m_dimension (dimension)
121 DE_ASSERT(dimension < DIMENSION_LAST);
|
| /external/freetype/src/pshinter/ |
| pshrec.h | 114 /* dimension-specific hints descriptor */
125 /* dimension 0 => X coordinates + vertical hints/stems */
126 /* dimension 1 => Y coordinates + horizontal hints/stems */
133 PS_DimensionRec dimension[2]; member in struct:PS_HintsRec_
|
| /external/python/google-api-python-client/samples/adexchangeseller/ |
| generate_report.py | 65 dimension=['DATE'],
|
| generate_report_with_paging.py | 66 dimension=['DATE'],
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| shape.cc | 27 for (const int64 dimension : shape_proto.dimensions()) {
28 add_dimensions(dimension);
39 "fields does not match number of dimension fields";
62 for (const int64 dimension : dimensions()) {
63 proto.add_dimensions(dimension);
|
| shape.h | 32 // dimension, and the primitive component type. For tuples, shape describes the
61 // Returns true if no array dimension in the shape is dynamically sized. Tuple
65 // Returns true if the given dimension is dynamically-sized.
66 bool is_dynamic_dimension(int dimension) const {
67 return dynamic_dimensions_.at(dimension);
70 // Sets whether or not the given dimension is dynamically-sized.
71 void set_dynamic_dimension(int dimension, bool is_dynamic) {
72 dynamic_dimensions_[dimension] = is_dynamic;
81 // Removes the given dimension form the shape. Layout, if it exists, is
203 // shapes. For a dynamically-sized dimension, the respective value in thi
[all...] |
