Home
last modified time | relevance | path

Searched defs:dims (Results 1 – 25 of 262) sorted by relevance

1234567891011

/external/tensorflow/tensorflow/lite/experimental/micro/testing/
Dtest_utils.h31 inline int ElementCount(const TfLiteIntArray& dims) { in ElementCount()
107 inline TfLiteTensor CreateFloatTensor(const float* data, TfLiteIntArray* dims, in CreateFloatTensor()
118 TfLiteIntArray* dims, const char* name) { in CreateFloatTensor()
123 TfLiteIntArray* dims, in CreateQuantizedTensor()
138 TfLiteIntArray* dims, in CreateQuantizedTensor()
145 TfLiteIntArray* dims, in CreateQuantized32Tensor()
160 TfLiteIntArray* dims, in CreateQuantized32Tensor()
/external/tensorflow/tensorflow/lite/kernels/
Dfill.cc35 TfLiteStatus ResizeOutputImpl(TfLiteContext* context, const TfLiteTensor* dims, in ResizeOutputImpl()
49 TfLiteStatus ResizeOutput(TfLiteContext* context, const TfLiteTensor* dims, in ResizeOutput()
72 const TfLiteTensor* dims = GetInput(context, node, kDimsTensor); in Prepare() local
102 const TfLiteTensor* dims = GetInput(context, node, kDimsTensor); in Eval() local
/external/eigen/unsupported/test/
Dcxx11_tensor_contraction.cpp90 Eigen::array<DimPair, 1> dims = {{DimPair(0, 0)}}; in test_scalar() local
111 Eigen::array<DimPair, 2> dims = {{DimPair(1, 2), DimPair(2, 3)}}; in test_multidims() local
165 Eigen::array<DimPair, 2> dims = {{DimPair(0, 0), DimPair(3, 4)}}; in test_holes() local
200 Eigen::array<DimPair, 2> dims = {{DimPair(0, 0), DimPair(1, 1)}}; in test_full_redux() local
230 Eigen::array<DimPair, 1> dims = {{DimPair(1, 0)}}; in test_contraction_of_contraction() local
261 Eigen::array<DimPair, 1> dims = {{DimPair(1, 0)}}; in test_expr() local
281 Eigen::array<DimPair, 2> dims = {{DimPair(2, 0), DimPair(0, 2)}}; in test_out_of_order_contraction() local
372 Eigen::array<DimPair, 2> dims = {{DimPair(2, 0), DimPair(3, 1)}}; in test_large_contraction() local
400 Eigen::array<DimPair, 1> dims{{DimPair(1, 0)}}; in test_matrix_vector() local
452 Eigen::array<DimPair, 2> dims = {{DimPair(2, 0), DimPair(3, 1)}}; in test_small_blocking_factors() local
[all …]
Dcxx11_tensor_convolution.cpp54 Eigen::array<ptrdiff_t, 2> dims; in test_expr() local
80 Eigen::array<ptrdiff_t, 1> dims; in test_modes() local
122 Eigen::array<ptrdiff_t, 1> dims; in test_strides() local
Dcxx11_tensor_argmax.cpp65 DimensionList<DenseIndex, 4> dims; in test_argmax_tuple_reducer() local
97 DimensionList<DenseIndex, 4> dims; in test_argmin_tuple_reducer() local
164 std::vector<int> dims {2, 3, 5, 7}; in test_argmax_dim() local
222 std::vector<int> dims {2, 3, 5, 7}; in test_argmin_dim() local
Dcxx11_tensor_thread_pool.cpp78 Eigen::array<DimPair, 2> dims({{DimPair(2, 0), DimPair(3, 1)}}); in test_multithread_contraction() local
119 Eigen::array<DimPair, 1> dims{{DimPair(0, 0)}}; in test_contraction_corner_cases() local
214 Eigen::array<DimPair, 1> dims({{DimPair(1, 2)}}); in test_multithread_contraction_agrees_with_singlethread() local
253 Eigen::array<DimPair, 2> dims({{DimPair(0, 0), DimPair(1, 1)}}); in test_full_contraction() local
/external/tensorflow/tensorflow/contrib/android/java/org/tensorflow/contrib/android/
DTensorFlowInferenceInterface.java301 public void feed(String inputName, boolean[] src, long... dims) { in feed()
317 public void feed(String inputName, float[] src, long... dims) { in feed()
327 public void feed(String inputName, int[] src, long... dims) { in feed()
337 public void feed(String inputName, long[] src, long... dims) { in feed()
347 public void feed(String inputName, double[] src, long... dims) { in feed()
357 public void feed(String inputName, byte[] src, long... dims) { in feed()
388 public void feed(String inputName, FloatBuffer src, long... dims) { in feed()
399 public void feed(String inputName, IntBuffer src, long... dims) { in feed()
410 public void feed(String inputName, LongBuffer src, long... dims) { in feed()
421 public void feed(String inputName, DoubleBuffer src, long... dims) { in feed()
[all …]
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc138 std::vector<int64> dims; member
169 std::vector<int64> dims) { in RandomInput()
180 std::vector<int64> dims) { in RandomUniqueInput()
435 bool OpTest::TensorSizeIsOk(absl::Span<const int64> dims) { in TensorSizeIsOk()
449 std::vector<int64> dims(rank); in RandomDims() local
606 std::vector<int64> dims) { in BroadcastableDims()
677 std::vector<int64> dims; in ImageDims() local
839 std::vector<int64> dims; in ExpectTfAndXlaOutputsAreClose() local
975 auto dims = BroadcastableDims(); in TEST_F() local
1034 auto dims = BroadcastableDims(); in TEST_F() local
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dreverse_sequence_op_gpu.cu.cc27 #define DEFINE_GPU_SPEC(T, Tlen, dims) \ argument
31 #define DEFINE_GPU_SPEC_LEN(T, dims) \ argument
Dmkl_conv_ops_test.cc77 static Tensor GetRandomInputTensor(const Conv2DDimensions& dims) { in GetRandomInputTensor()
83 static Tensor GetRandomFilterTensor(const Conv2DDimensions& dims) { in GetRandomFilterTensor()
89 static Tensor GetRandomOutputTensor(const Conv2DDimensions& dims) { in GetRandomOutputTensor()
95 static Tensor GetInputSizesTensor(const Conv2DDimensions& dims) { in GetInputSizesTensor()
101 static Tensor GetFilterSizesTensor(const Conv2DDimensions& dims) { in GetFilterSizesTensor()
120 static Graph* DefaultConv2D(const Conv2DDimensions& dims) { in DefaultConv2D()
142 static Graph* MklConv2D(const Conv2DDimensions& dims) { in MklConv2D()
169 static Graph* DefaultConv2DBwdInput(const Conv2DDimensions& dims) { in DefaultConv2DBwdInput()
197 static Graph* MklConv2DBwdInput(const Conv2DDimensions& dims) { in MklConv2DBwdInput()
231 static Graph* DefaultConv2DBwdFilter(const Conv2DDimensions& dims) { in DefaultConv2DBwdFilter()
[all …]
Dreduction_ops_common.cc36 const int dims = data_reshape_.size(); in shuffled_shape() local
48 const int dims = data_reshape_.size(); in permutation() local
Dscan_ops.h34 Eigen::array<bool, 3> dims; in operator() local
/external/tensorflow/tensorflow/core/framework/
Dtensor.h207 int dims() const { return shape().dims(); } in dims() function
751 Eigen::array<Eigen::DenseIndex, NDIMS> dims; in reinterpret_last_dimension() local
767 Eigen::array<Eigen::DenseIndex, NDIMS> dims; in reinterpret_last_dimension() local
777 Eigen::array<Eigen::DenseIndex, NDIMS>* dims) const { in FillDimsAndValidateCompatibleShape()
790 Eigen::array<Eigen::DenseIndex, NDIMS>* dims) const { in FillDimsAndValidateCompatibleShape()
815 Eigen::array<Eigen::DenseIndex, NDIMS> dims; in shaped() local
824 Eigen::array<Eigen::DenseIndex, NDIMS> dims; in bit_casted_shaped() local
833 Eigen::array<Eigen::DenseIndex, NDIMS> dims; in unaligned_shaped() local
843 Eigen::array<Eigen::DenseIndex, NDIMS> dims; in shaped() local
852 Eigen::array<Eigen::DenseIndex, NDIMS> dims; in bit_casted_shaped() local
[all …]
Dshape_inference.cc377 std::vector<DimensionHandle> dims; in WithRank() local
500 std::vector<DimensionHandle> dims; in MergePrefix() local
549 std::vector<DimensionHandle> dims(rank); in Relax() local
611 std::vector<DimensionHandle> dims(rank, nullptr); in Merge() local
689 std::vector<DimensionHandle> dims; in Subshape() local
704 std::vector<DimensionHandle> dims; in Concatenate() local
726 std::vector<DimensionHandle> dims(s->dims_); in ReplaceDim() local
732 const std::vector<DimensionHandle>& dims) { in MakeShape()
737 std::initializer_list<DimensionOrConstant> dims) { in MakeShape()
757 std::vector<DimensionHandle> dims(rank); in UnknownShapeOfRank() local
[all …]
/external/tensorflow/tensorflow/c/eager/
Dc_api_test_util.cc61 int64_t dims[] = {2, 2}; in DoubleTestMatrixTensorHandle() local
75 int64_t dims[] = {2, 2}; in TestMatrixTensorHandle() local
89 int64_t dims[] = {3, 2}; in DoubleTestMatrixTensorHandle3X2() local
103 int64_t dims[] = {3, 2}; in TestMatrixTensorHandle3X2() local
145 int64_t dims[] = {1}; in TestAxisTensorHandle() local
/external/mesa3d/src/intel/compiler/
Dbrw_fs_surface_builder.cpp40 unsigned dims, unsigned arg, unsigned rsize, in emit_send()
67 unsigned dims, unsigned size, in emit_untyped_read()
82 unsigned dims, unsigned size, in emit_untyped_write()
98 unsigned dims, unsigned rsize, unsigned op, in emit_untyped_atomic()
120 const fs_reg &addr, unsigned dims, unsigned size) in emit_typed_read()
134 unsigned dims, unsigned size) in emit_typed_write()
149 unsigned dims, unsigned rsize, unsigned op, in emit_typed_atomic()
167 unsigned dims, unsigned size, in emit_byte_scattered_read()
177 unsigned dims, unsigned size, in emit_byte_scattered_write()
445 const fs_reg &addr, unsigned dims) in emit_bounds_check()
[all …]
Dbrw_vec4_surface_builder.cpp164 unsigned dims, unsigned size, in emit_untyped_read()
181 unsigned dims, unsigned size, in emit_untyped_write()
203 unsigned dims, unsigned rsize, unsigned op, in emit_untyped_atomic()
265 const src_reg &addr, unsigned dims, unsigned size) in emit_typed_read()
289 unsigned dims, unsigned size) in emit_typed_write()
311 unsigned dims, unsigned rsize, unsigned op, in emit_typed_atomic()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddiag_op.cc88 auto dims = input_shape.dim_sizes(); in Compile() local
127 auto dims = input_shape.dim_sizes(); in Compile() local
172 auto dims = input_shape.dim_sizes(); in Compile() local
198 auto dims = input_shape.dim_sizes(); in Compile() local
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorBase.h483 contract(const OtherDerived& other, const Dimensions& dims) const { in contract()
490 convolve(const KernelDerived& kernel, const Dimensions& dims) const { in convolve()
526 sum(const Dims& dims) const { in sum()
538 mean(const Dims& dims) const { in mean()
550 prod(const Dims& dims) const { in prod()
562 maximum(const Dims& dims) const { in maximum()
574 minimum(const Dims& dims) const { in minimum()
586 all(const Dims& dims) const { in all()
599 any(const Dims& dims) const { in any()
664 reduce(const Dims& dims, const Reducer& reducer) const { in reduce()
/external/tensorflow/tensorflow/lite/testing/kernel_test/
Dinput_generator.cc32 std::vector<T> GenerateRandomTensor(TfLiteIntArray* dims, in GenerateRandomTensor()
47 std::vector<T> GenerateUniform(TfLiteIntArray* dims, float min, float max) { in GenerateUniform()
62 std::vector<T> GenerateGaussian(TfLiteIntArray* dims, float min, float max) { in GenerateGaussian()
/external/tensorflow/tensorflow/python/tpu/
Dtpu_feed.py40 def partition_or_replicate_on_host(tensor, dims): argument
86 def _tag_sharding_attribute_for_dequeued_tensor(tensor, dims): argument
105 def tag_sharding_attribute_for_dequeued_tensors(dequeues, dims): argument
868 def _check_input_partition_dims(self, tensor, dims): argument
906 def _check_dims_and_partition_or_replicate_on_host(self, tensor, dims): argument
/external/tensorflow/tensorflow/lite/
Dinterpreter.cc129 const std::vector<int>& dims) { in ResizeInputTensor()
157 const std::vector<int>& dims, TfLiteQuantization quantization, in SetTensorParametersReadOnly()
166 const std::vector<int>& dims, TfLiteQuantization quantization, in SetTensorParametersReadWrite()
175 const int* dims, TfLiteQuantizationParams quantization, const char* buffer, in SetTensorParametersReadOnly()
189 const int* dims, TfLiteQuantizationParams quantization, bool is_variable) { in SetTensorParametersReadWrite()
/external/mesa3d/src/mesa/main/
Dtexstorage.c52 legal_texobj_target(const struct gl_context *ctx, GLuint dims, GLenum target) in legal_texobj_target()
308 GLuint dims, GLenum target, in tex_storage_error_check()
392 texture_storage(struct gl_context *ctx, GLuint dims, in texture_storage()
495 texture_storage_error(struct gl_context *ctx, GLuint dims, in texture_storage_error()
507 texture_storage_no_error(struct gl_context *ctx, GLuint dims, in texture_storage_no_error()
522 texstorage_error(GLuint dims, GLenum target, GLsizei levels, in texstorage_error()
563 texstorage_no_error(GLuint dims, GLenum target, GLsizei levels, in texstorage_no_error()
579 texturestorage_error(GLuint dims, GLuint texture, GLsizei levels, in texturestorage_error()
620 texturestorage_no_error(GLuint dims, GLuint texture, GLsizei levels, in texturestorage_no_error()
806 _mesa_texture_storage_memory(struct gl_context *ctx, GLuint dims, in _mesa_texture_storage_memory()
/external/androidplot/AndroidPlot-Core/src/main/java/com/androidplot/ui/
DResizable.java39 public void layout(final DisplayDimensions dims); in layout()
/external/tensorflow/tensorflow/core/kernels/neon/
Dtypes.h31 inline int Offset(const Dims<4>& dims, int i0, int i1, int i2, int i3) { in Offset()
62 inline int RequiredBufferSizeForDims(const Dims<4>& dims) { in RequiredBufferSizeForDims()

1234567891011