Home
last modified time | relevance | path

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

12345678910>>...47

/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_block_access.cpp38 static size_t RandomTargetSize(const DSizes<Index, NumDims>& dims) { in RandomTargetSize()
44 array<Index, NumDims> dims; in RandomDims() local
61 static void Debug(DSizes<Index, NumDims> dims) { in Debug()
127 DSizes<Index, NumDims> dims = RandomDims<NumDims>(); in test_block_mapper_maps_every_element() local
236 DSizes<Index, 5> dims(11, 5, 6, 17, 7); in test_uniform_block_shape() local
250 DSizes<Index, 5> dims(11, 5, 6, 17, 7); in test_uniform_block_shape() local
261 DSizes<Index, 5> dims(11, 5, 6, 17, 7); in test_uniform_block_shape() local
276 DSizes<Index, 5> dims(11, 5, 6, 17, 7); in test_uniform_block_shape() local
287 DSizes<Index, 5> dims(11, 5, 6, 17, 7); in test_uniform_block_shape() local
302 DSizes<Index, 5> dims(7, 5, 6, 17, 7); in test_uniform_block_shape() local
[all …]
H A Dcxx11_tensor_block_eval.cpp21 DSizes<Index, NumDims> dims; in RandomDims() local
38 static TensorBlockParams<NumDims> RandomBlock(DSizes<Index, NumDims> dims, in RandomBlock()
63 DSizes<Index, NumDims> dims) { in SkewedInnerBlock()
100 static TensorBlockParams<NumDims> FixedSizeBlock(DSizes<Index, NumDims> dims) { in FixedSizeBlock()
209 DSizes<Index, NumDims> dims = RandomDims<NumDims>(10, 20); in test_eval_tensor_block() local
220 DSizes<Index, NumDims> dims = RandomDims<NumDims>(10, 20); in test_eval_tensor_unary_expr_block() local
230 DSizes<Index, NumDims> dims = RandomDims<NumDims>(10, 20); in test_eval_tensor_binary_expr_block() local
241 DSizes<Index, NumDims> dims = RandomDims<NumDims>(10, 20); in test_eval_tensor_binary_with_unary_expr_block() local
253 DSizes<Index, NumDims> dims = RandomDims<NumDims>(1, 10); in test_eval_tensor_broadcast() local
283 DSizes<Index, NumDims> dims = RandomDims<NumDims>(1, 10); in test_eval_tensor_reshape() local
[all …]
H A Dcxx11_tensor_executor.cpp50 array<Index, NumDims> dims; in RandomDims() local
65 auto dims = RandomDims<NumDims>(50 / NumDims, 100 / NumDims); in test_execute_unary_expr() local
93 auto dims = RandomDims<NumDims>(50 / NumDims, 100 / NumDims); in test_execute_binary_expr() local
122 auto dims = RandomDims<NumDims>(1, 10); in test_execute_broadcasting() local
152 auto dims = RandomDims<NumDims>(1, 10); in test_execute_chipping_rvalue() local
191 auto dims = RandomDims<NumDims>(1, 10); in test_execute_chipping_lvalue() local
245 auto dims = RandomDims<NumDims>(1, 10); in test_execute_shuffle_rvalue() local
283 auto dims = RandomDims<NumDims>(5, 10); in test_execute_shuffle_lvalue() local
322 auto dims = RandomDims<NumDims>(5, 10); in test_execute_reshape() local
360 auto dims = RandomDims<NumDims>(5, 10); in test_execute_slice_rvalue() local
[all …]
H A 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 …]
H A Dcxx11_tensor_thread_pool.cpp121 Eigen::array<DimPair, 2> dims({{DimPair(2, 0), DimPair(3, 1)}}); in test_multithread_contraction() local
162 Eigen::array<DimPair, 1> dims{{DimPair(0, 0)}}; in test_contraction_corner_cases() local
257 Eigen::array<DimPair, 1> dims({{DimPair(1, 2)}}); in test_multithread_contraction_agrees_with_singlethread() local
320 Eigen::array<DimPair, 2> dims({{DimPair(2, 0), DimPair(3, 1)}}); in test_multithread_contraction_with_output_kernel() local
354 Eigen::array<DimPair, 1> dims({{DimPair(1, 2)}}); in test_async_multithread_contraction_agrees_with_singlethread() local
408 Eigen::array<DimPair, 1> dims({{DimPair(1, 0)}}); in test_sharded_by_inner_dim_contraction() local
448 Eigen::array<DimPair, 1> dims({{DimPair(1, 0)}}); in test_sharded_by_inner_dim_contraction_with_output_kernel() local
488 Eigen::array<DimPair, 1> dims({{DimPair(1, 0)}}); in test_async_sharded_by_inner_dim_contraction() local
532 Eigen::array<DimPair, 1> dims({{DimPair(1, 0)}}); in test_async_sharded_by_inner_dim_contraction_with_output_kernel() local
563 Eigen::array<DimPair, 2> dims({{DimPair(0, 0), DimPair(1, 1)}}); in test_full_contraction() local
/aosp_15_r20/external/executorch/kernels/test/
H A Dop_expand_copy_test.cpp42 const std::vector<int64_t> dims{2, 2}; in TEST_F() local
54 const std::vector<int64_t> dims{3, 3, 3, 2, 2}; in TEST_F() local
67 const std::vector<int64_t> dims{2, 92}; in TEST_F() local
79 const std::vector<int64_t> dims{-1, -1, -1}; in TEST_F() local
91 const std::vector<int64_t> dims{-1, -1, -1}; in TEST_F() local
103 const std::vector<int64_t> dims{-1, 14, -1}; in TEST_F() local
115 const std::vector<int64_t> dims{-1, -1, 12}; in TEST_F() local
127 const std::vector<int64_t> dims{2}; in TEST_F() local
138 const std::vector<int64_t> dims{-1, -1, -1, -1, 2, 14, 1}; in TEST_F() local
149 const std::vector<int64_t> dims{2, 6}; in TEST_F() local
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Analysis/
H A Dshape_component_analysis.cc254 auto &dims = insert(ShapeOrValueInfo::getShapeInfoOf(op)); in forwardAssumingShape() local
279 auto &dims = insert(ShapeOrValueInfo::getValueInfoOf(op)); in forwardBroadcast() local
328 auto &dims = insert(ShapeOrValueInfo::getShapeInfoOf(op)); in forwardDynamicBroadcastInDimShape() local
340 auto &dims = insert(ShapeOrValueInfo::getShapeInfoOf(op)); in forwardDynamicReshapeShape() local
354 auto &dims = insert(ShapeOrValueInfo::getShapeInfoOf(op)); in forwardReduceShape() local
366 auto &dims = insert(ShapeOrValueInfo::getShapeInfoOf(op)); in forwardTransposeShape() local
376 auto &dims = insert(ShapeOrValueInfo::getShapeInfoOf(op)); in forwardSelectShape() local
386 auto &dims = insert(ShapeOrValueInfo::getShapeInfoOf(v)); in forwardSameOperandsShape() local
411 auto &dims = insert(ShapeOrValueInfo::getShapeInfoOf(argument)); in backwardBlockArgumentShape() local
440 auto &dims = insert(ShapeOrValueInfo::getShapeInfoOf(v)); in forwardUnknownShape() local
[all …]
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dtensor.cpp101 const std::vector<ExprHandle>& dims, in Compute()
111 const std::vector<ExprHandle>& dims, in Compute()
118 const std::vector<ExprHandle>& dims, in Compute()
132 const std::vector<ExprHandle>& dims, in Compute()
139 const std::vector<ExprHandle>& dims, in Compute()
153 const std::vector<ExprHandle>& dims, in Compute()
161 const std::vector<ExprHandle>& dims, in Compute()
176 const std::vector<ExprHandle>& dims, in Compute()
185 const std::vector<ExprHandle>& dims, in Compute()
202 const std::vector<ExprHandle>& dims, in Compute()
[all …]
H A Dtensor.h132 const std::vector<ExprHandle>& dims) { in create_index_vars()
146 const std::vector<ExprHandle>& dims, in Reduce()
197 const std::vector<ExprHandle>& dims, in Reduce()
215 const std::vector<ExprHandle>& dims, in Reduce()
234 const std::vector<ExprHandle>& dims, in Reduce()
246 const std::vector<ExprHandle>& dims, in Reduce()
256 const std::vector<ExprHandle>& dims, in Reduce()
H A Dexpr.h236 dims, in Buf() argument
260 std::vector<ExprPtr> dims() const { in dims() function
263 void set_dims(std::vector<ExprPtr> dims) { in set_dims()
338 const std::vector<ExprHandle>& dims, in BufHandle()
344 const std::vector<ExprHandle>& dims, in BufHandle()
349 BufHandle(const std::vector<ExprHandle>& dims, Dtype dtype) in BufHandle()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Drandomized_tests.cc147 std::vector<int64_t> dims; member
178 std::vector<int64_t> dims) { in RandomInput()
189 std::vector<int64_t> dims) { in RandomUniqueInput()
752 bool OpTest::TensorSizeIsOk(absl::Span<const int64_t> dims) { in TensorSizeIsOk()
766 std::vector<int64_t> dims(rank); in RandomDims() local
963 std::vector<int64_t> OpTest::BroadcastableToDims(std::vector<int64_t> dims) { in BroadcastableToDims()
984 std::vector<int64_t> dims) { in BroadcastableDims()
1069 std::vector<int64_t> dims = RandomDims(1, 4, 0, 64); in ChooseConcatArguments() local
1093 std::vector<int64> dims; in ChooseEinsumArguments() local
1111 std::vector<int64> dims = RandomDims(1, 1); in ChooseEinsumArguments() local
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dtensor.h284 int dims() const { return shape().dims(); } in dims() function
798 Eigen::array<Eigen::DenseIndex, NDIMS> dims; in reinterpret_last_dimension() local
815 Eigen::array<Eigen::DenseIndex, NDIMS> dims; in reinterpret_last_dimension() local
825 Eigen::array<Eigen::DenseIndex, NDIMS>* dims) const { in FillDimsAndValidateCompatibleShape()
838 Eigen::array<Eigen::DenseIndex, NDIMS>* dims) const { in FillDimsAndValidateCompatibleShape()
864 Eigen::array<Eigen::DenseIndex, 1> dims; in flat() local
873 Eigen::array<Eigen::DenseIndex, 1> dims; in flat() local
882 Eigen::array<Eigen::DenseIndex, NDIMS> dims; in shaped() local
891 Eigen::array<Eigen::DenseIndex, NDIMS> dims; in bit_casted_shaped() local
900 Eigen::array<Eigen::DenseIndex, NDIMS> dims; in unaligned_shaped() local
[all …]
H A Dshape_inference.cc326 std::vector<DimensionHandle> dims; in WithRank() local
449 std::vector<DimensionHandle> dims; in MergePrefix() local
498 std::vector<DimensionHandle> dims(rank); in Relax() local
560 std::vector<DimensionHandle> dims(rank, nullptr); in Merge() local
638 std::vector<DimensionHandle> dims; in Subshape() local
653 std::vector<DimensionHandle> dims; in Concatenate() local
675 std::vector<DimensionHandle> dims(s->dims_); in ReplaceDim() local
681 const std::vector<DimensionHandle>& dims) { in MakeShape()
686 std::initializer_list<DimensionOrConstant> dims) { in MakeShape()
706 std::vector<DimensionHandle> dims(rank); in UnknownShapeOfRank() local
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/tools/android/inference_interface/java/org/tensorflow/contrib/android/
H A 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 …]
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_conv_ops_test.cc73 static Tensor GetRandomInputTensor(const Conv2DDimensions& dims) { in GetRandomInputTensor()
79 static Tensor GetRandomFilterTensor(const Conv2DDimensions& dims) { in GetRandomFilterTensor()
85 static Tensor GetRandomOutputTensor(const Conv2DDimensions& dims) { in GetRandomOutputTensor()
91 static Tensor GetInputSizesTensor(const Conv2DDimensions& dims) { in GetInputSizesTensor()
97 static Tensor GetFilterSizesTensor(const Conv2DDimensions& dims) { in GetFilterSizesTensor()
102 static Graph* DefaultConv2D(const Conv2DDimensions& dims) { in DefaultConv2D()
123 static Graph* MklConv2D(const Conv2DDimensions& dims) { in MklConv2D()
150 static Graph* DefaultConv2DBwdInput(const Conv2DDimensions& dims) { in DefaultConv2DBwdInput()
177 static Graph* MklConv2DBwdInput(const Conv2DDimensions& dims) { in MklConv2DBwdInput()
211 static Graph* DefaultConv2DBwdFilter(const Conv2DDimensions& dims) { in DefaultConv2DBwdFilter()
[all …]
/aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorBase.h608 contract(const OtherDerived& other, const Dimensions& dims) const { in contract()
614 …contract(const OtherDerived& other, const Dimensions& dims, const OutputKernel& output_kernel) con… in contract()
621 convolve(const KernelDerived& kernel, const Dimensions& dims) const { in convolve()
628 fft(const FFT& dims) const { in fft()
657 sum(const Dims& dims) const { in sum()
669 mean(const Dims& dims) const { in mean()
681 prod(const Dims& dims) const { in prod()
693 maximum(const Dims& dims) const { in maximum()
706 minimum(const Dims& dims) const { in minimum()
719 all(const Dims& dims) const { in all()
[all …]
/aosp_15_r20/external/XNNPACK/test/
H A Dworkspace.cc19 void DefineGraphWithoutInternalTensors(xnn_subgraph_t* subgraph, std::array<size_t, 4> dims) in DefineGraphWithoutInternalTensors()
40 void DefineGraph(xnn_subgraph_t* subgraph, std::array<size_t, 4> dims) in DefineGraph()
65 void DefineGraphWithStaticData(xnn_subgraph_t* subgraph, std::array<size_t, 4> dims, const std::vec… in DefineGraphWithStaticData()
123 std::array<size_t, 4> dims = {2, 20, 20, 3}; in TEST() local
187 std::array<size_t, 4> dims = {2, 20, 20, 3}; in TEST() local
305 const std::array<size_t, 4> dims = {2, 20, 20, 3}; in TEST() local
345 std::array<size_t, 4> dims = {2, 20, 20, 3}; in TEST() local
386 std::array<size_t, 4> dims = {2, 20, 20, 3}; in TEST() local
442 std::array<size_t, 4> dims = {2, 20, 20, 3}; in TEST() local
/aosp_15_r20/external/executorch/runtime/core/exec_aten/util/
H A Ddim_order_util.h22 bool validate_dim_order(const DimOrderType* dim_order, const size_t dims) { in validate_dim_order()
42 const size_t dims) { in is_contiguous_dim_order()
61 const size_t dims) { in is_channels_last_dim_order()
114 const size_t dims, in dim_order_to_stride_nocheck()
140 const size_t dims, in dim_order_to_stride()
231 const size_t dims, in stride_to_dim_order()
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dc_api_test_util.cc84 int64_t dims[] = {2, 2}; in DoubleTestMatrixTensorHandle() local
98 int64_t dims[] = {2, 2}; in TestMatrixTensorHandle() local
112 float data[], int64_t dims[], in TestMatrixTensorHandleWithInput()
126 int64_t dims[], int num_dims) { in TestTensorHandleWithDimsFloat()
139 int64_t dims[], int num_dims) { in TestTensorHandleWithDimsInt()
152 constexpr int64_t dims[] = {100, 100}; in TestMatrixTensorHandle100x100() local
171 int64_t dims[] = {3, 2}; in DoubleTestMatrixTensorHandle3X2() local
185 int64_t dims[] = {3, 2}; in TestMatrixTensorHandle3X2() local
307 int64_t dims[] = {1}; in TestAxisTensorHandle() local
/aosp_15_r20/external/google-java-format/core/src/main/java/com/google/googlejavaformat/java/
H A DDimensionHelpers.java42 final ImmutableList<List<AnnotationTree>> dims; field in DimensionHelpers.TypeWithDims
44 public TypeWithDims(Tree node, ImmutableList<List<AnnotationTree>> dims) { in TypeWithDims()
59 Iterable<List<AnnotationTree>> dims; in extractDims() local
81 Deque<List<AnnotationTree>> dims) { in reorderBySourcePosition()
108 private static Tree extractDims(Deque<List<AnnotationTree>> dims, Tree node) { in extractDims()
/aosp_15_r20/external/mesa3d/src/mesa/main/
H A Dtexstorage.c55 GLuint dims, GLenum target) in _mesa_is_legal_tex_storage_target()
356 GLuint dims, GLenum target, in tex_storage_error_check()
435 _mesa_sparse_texture_error_check(struct gl_context *ctx, GLuint dims, in _mesa_sparse_texture_error_check()
518 texture_storage(struct gl_context *ctx, GLuint dims, in texture_storage()
641 texture_storage_error(struct gl_context *ctx, GLuint dims, in texture_storage_error()
654 texture_storage_no_error(struct gl_context *ctx, GLuint dims, in texture_storage_no_error()
670 texstorage_error(GLuint dims, GLenum target, GLsizei levels, in texstorage_error()
711 texstorage_no_error(GLuint dims, GLenum target, GLsizei levels, in texstorage_no_error()
727 texturestorage_error(GLuint dims, GLuint texture, GLsizei levels, in texturestorage_error()
768 texturestorage_no_error(GLuint dims, GLuint texture, GLsizei levels, in texturestorage_no_error()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
H A Dtrt_tensor_proxy.h43 SimpleITensor(const nvinfer1::Dims& dims) in SimpleITensor()
46 SimpleITensor(const std::vector<int>& dims) { in SimpleITensor()
137 explicit ITensorProxy(const nvinfer1::Dims& dims) in ITensorProxy()
141 explicit ITensorProxy(const std::vector<int>& dims) in ITensorProxy()
420 ITensorProxyPtr(const nvinfer1::Dims& dims) : p_(new ITensorProxy(dims)) {} in ITensorProxyPtr()
421 ITensorProxyPtr(const std::vector<int>& dims) : p_(new ITensorProxy(dims)) {} in ITensorProxyPtr()
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dinterpreter.cc212 const std::vector<int>& dims) { in ResizeInputTensor()
217 int tensor_index, const std::vector<int>& dims) { in ResizeInputTensorStrict()
251 const std::vector<int>& dims, TfLiteQuantization quantization, in SetTensorParametersReadOnly()
260 const std::vector<int>& dims, TfLiteQuantization quantization, in SetTensorParametersReadWrite()
269 const int* dims, TfLiteQuantizationParams quantization, const char* buffer, in SetTensorParametersReadOnly()
279 const int* dims, TfLiteQuantizationParams quantization, bool is_variable, in SetTensorParametersReadWrite()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dquantize_and_dequantize_op_test.cc100 std::vector<T> ScalePerSliceAlongAxis(std::vector<int64_t> dims, int axis, in ScalePerSliceAlongAxis()
135 const std::vector<int64_t> dims = {2, 3, 4, 5}; in TEST_P() local
186 const std::vector<int64_t> dims = {5, 7, 11, 13}; in TEST_P() local
240 const std::vector<int64_t> dims = {2, 3, 4, 5}; in TEST_P() local
323 const std::vector<int64_t> dims = {2, 3, 4, 5}; in TEST_P() local
376 const std::vector<int64_t> dims = {2, 3, 4, 5}; in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dtf_tensor.cc71 const int64_t* dims, int num_dims, size_t len) { in CreateTensor()
90 TF_Tensor* TF_AllocateTensor(TF_DataType dtype, const int64_t* dims, in TF_AllocateTensor()
100 TF_Tensor* TF_NewTensor(TF_DataType dtype, const int64_t* dims, int num_dims, in TF_NewTensor()
151 void TF_SetShape(TF_Tensor* t, const int64_t* dims, int num_dims) { in TF_SetShape()
238 void TensorInterface::SetShape(const int64_t* dims, int num_dims) { in SetShape()
276 std::vector<int64_t> dims; in EmptyTensor() local

12345678910>>...47