| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_contraction.cpp | 145 Eigen::array<DimPair, 2> dims2({{DimPair(0, 1), DimPair(1, 0)}}); in test_multidims() local
297 Eigen::array<DimPair, 2> dims2 = {{DimPair(0, 2), DimPair(2, 0)}}; in test_out_of_order_contraction() local
330 Eigen::array<DimPair, 2> dims2 = {{DimPair(4, 0), DimPair(0, 1)}}; in test_consistency() local
|
| H A D | cxx11_tensor_trace.cpp | 71 array<ptrdiff_t, 2> dims2 = { { 2, 3 } }; in test_simple_trace() local
|
| /aosp_15_r20/external/pytorch/torch/autograd/_functions/ |
| H A D | utils.py | 38 def check_onnx_broadcast(dims1, dims2): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | kernel_util.cc | 461 const int dims2 = NumDimensions(input2); in CalculateShapeForBroadcast() local
493 const int dims2 = NumDimensions(input2); in CalculateShapeForBroadcast() local
|
| H A D | kernel_util_test.cc | 344 TfLiteIntArray* dims2 = TfLiteIntArrayCreate(2); in TEST_F() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Linear.cpp | 769 Tensor tensordot(const Tensor& input1, const Tensor& input2, IntArrayRef dims1, IntArrayRef dims2) { in tensordot()
827 …_out(const Tensor& input1, const Tensor& input2, IntArrayRef dims1, IntArrayRef dims2, Tensor& res… in tensordot_out()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| H A D | trt_engine_resource_ops_test.cc | 99 nvinfer1::Dims dims2{1, {2}}; in NetworkWith2Inputs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | concatenate_dataset_op.cc | 246 auto dims2 = ts2.dim_sizes(); in MostSpecificCompatibleShape() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/ |
| H A D | kernel_test.cc | 498 TfLiteIntArray* dims2 = TfLiteIntArrayCreate(2); in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| H A D | directed_interleave_dataset_op.cc | 308 auto dims2 = ts2.dim_sizes(); in MostSpecificCompatibleShape() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/xpu/ |
| H A D | Blas.cpp | 395 IntArrayRef dims2, in tensordot_out()
|
| /aosp_15_r20/external/XNNPACK/test/ |
| H A D | workspace.cc | 256 std::array<size_t, 4> dims2 = dims1; in TEST() local
|
| /aosp_15_r20/external/pytorch/torch/fx/experimental/migrate_gradual_types/ |
| H A D | constraint_generator.py | 1065 def add_linear_constraints(dims1, dims2, in_features, out_features): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_utils.py | 801 def try_check_onnx_broadcast(dims1, dims2, expect_broadcast, expect_fail): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | c_api_test.cc | 2400 const int64_t dims2[] = {2, 2}; in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | legacy_optimized_ops.h | 130 inline bool AreSameDims(const Dims<4>& dims1, const Dims<4>& dims2) { in AreSameDims()
|
| /aosp_15_r20/prebuilts/devtools/tools/lib/ |
| HD | lombok-ast-0.2.3.jar | META-INF/
META-INF/MANIFEST.MF
META-INF/maven/
META- ... |