| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_block_access.cpp | 38 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 …]
|
| D | cxx11_tensor_block_eval.cpp | 21 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 …]
|
| D | cxx11_tensor_contraction.cpp | 90 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 …]
|
| D | cxx11_tensor_executor.cpp | 50 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 …]
|
| D | cxx11_tensor_convolution.cpp | 55 Eigen::array<ptrdiff_t, 2> dims; in test_expr() local
81 Eigen::array<ptrdiff_t, 1> dims; in test_modes() local
123 Eigen::array<ptrdiff_t, 1> dims; in test_strides() local
|
| D | cxx11_tensor_thread_pool.cpp | 121 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
|
| D | cxx11_tensor_argmax.cpp | 65 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
|
| /external/google-java-format/core/src/main/java/com/google/googlejavaformat/java/ |
| D | DimensionHelpers.java | 42 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()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Analysis/ |
| D | shape_component_analysis.cc | 254 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 …]
|
| /external/tensorflow/tensorflow/tools/android/inference_interface/java/org/tensorflow/contrib/android/ |
| D | TensorFlowInferenceInterface.java | 301 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/core/framework/ |
| D | tensor.h | 284 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 …]
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops_test.cc | 73 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 …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | fill.cc | 37 TfLiteStatus ResizeOutputImpl(TfLiteContext* context, const TfLiteTensor* dims, in ResizeOutputImpl()
52 TfLiteStatus ResizeOutput(TfLiteContext* context, const TfLiteTensor* dims, in ResizeOutput()
75 const TfLiteTensor* dims; in Prepare() local
134 const TfLiteTensor* dims; in Eval() local
|
| /external/XNNPACK/test/ |
| D | workspace.cc | 19 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
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 147 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 …]
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorBase.h | 608 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 …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reverse_sequence_op_gpu.cu.cc | 27 #define DEFINE_GPU_SPEC(T, Tlen, dims) \ argument
31 #define DEFINE_GPU_SPEC_LEN(T, dims) \ argument
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test_util.cc | 84 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
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_tensor_proxy.h | 43 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()
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter.cc | 212 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()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | quantization_util_test.cc | 59 int ElementCount(const TfLiteIntArray& dims) { in ElementCount()
85 TfLiteTensor CreateQuantizedTensor(const int8_t* data, TfLiteIntArray* dims, in CreateQuantizedTensor()
104 TfLiteTensor CreateQuantizedTensor(const uint8_t* data, TfLiteIntArray* dims, in CreateQuantizedTensor()
123 TfLiteTensor CreateTensor(TfLiteIntArray* dims, const char* name, in CreateTensor()
138 TfLiteTensor CreateFloatTensor(const float* data, TfLiteIntArray* dims, in CreateFloatTensor()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | util_test.cc | 156 std::vector<int> dims = {1}; in TEST() local
176 std::vector<int> dims = {1}; in TEST() local
207 TfLiteIntArray* dims = TfLiteIntArrayCreate(2); in TEST() local
245 TfLiteIntArray* dims = TfLiteIntArrayCreate(2); in TEST() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | weights.cc | 34 nvinfer1::DataType type, DimsAdapter dims, Tensor tensor) { in CreateWithTensor()
49 Status TRT_ShapedWeights::SetShape(DimsAdapter dims) { in SetShape()
190 nvinfer1::DataType trt_dtype, const DimsAdapter& dims) { in GetTempWeights()
|
| /external/mesa3d/src/mesa/main/ |
| D | texstorage.c | 52 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()
496 texture_storage_error(struct gl_context *ctx, GLuint dims, in texture_storage_error()
508 texture_storage_no_error(struct gl_context *ctx, GLuint dims, in texture_storage_no_error()
523 texstorage_error(GLuint dims, GLenum target, GLsizei levels, in texstorage_error()
564 texstorage_no_error(GLuint dims, GLenum target, GLsizei levels, in texstorage_no_error()
580 texturestorage_error(GLuint dims, GLuint texture, GLsizei levels, in texturestorage_error()
621 texturestorage_no_error(GLuint dims, GLuint texture, GLsizei levels, in texturestorage_no_error()
793 _mesa_texture_storage_memory(struct gl_context *ctx, GLuint dims, in _mesa_texture_storage_memory()
|
| /external/tensorflow/tensorflow/c/ |
| D | tf_tensor.cc | 71 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
|