/aosp_15_r20/external/XNNPACK/models/ |
H A D | fp32-mobilenet-v3-large.cc | 2628 const size_t a_shape[] = { 1, 112, 112, 16 }; in FP32MobileNetV3Large() local 2702 const size_t a_shape[] = { 1, 56, 56, 24 }; in FP32MobileNetV3Large() local 2766 const size_t a_shape[] = { 1, 28, 28, 72 }; in FP32MobileNetV3Large() local 2840 const size_t a_shape[] = { 1, 28, 28, 120 }; in FP32MobileNetV3Large() local 2864 const size_t a_shape[] = { 1, 28, 28, 40 }; in FP32MobileNetV3Large() local 2928 const size_t a_shape[] = { 1, 28, 28, 120 }; in FP32MobileNetV3Large() local 2952 const size_t a_shape[] = { 1, 28, 28, 40 }; in FP32MobileNetV3Large() local 3066 const size_t a_shape[] = { 1, 14, 14, 80 }; in FP32MobileNetV3Large() local 3130 const size_t a_shape[] = { 1, 14, 14, 80 }; in FP32MobileNetV3Large() local 3194 const size_t a_shape[] = { 1, 14, 14, 80 }; in FP32MobileNetV3Large() local [all …]
|
H A D | fp16-mobilenet-v3-large.cc | 2631 const size_t a_shape[] = { 1, 112, 112, 16 }; in FP16MobileNetV3Large() local 2705 const size_t a_shape[] = { 1, 56, 56, 24 }; in FP16MobileNetV3Large() local 2769 const size_t a_shape[] = { 1, 28, 28, 72 }; in FP16MobileNetV3Large() local 2843 const size_t a_shape[] = { 1, 28, 28, 120 }; in FP16MobileNetV3Large() local 2867 const size_t a_shape[] = { 1, 28, 28, 40 }; in FP16MobileNetV3Large() local 2931 const size_t a_shape[] = { 1, 28, 28, 120 }; in FP16MobileNetV3Large() local 2955 const size_t a_shape[] = { 1, 28, 28, 40 }; in FP16MobileNetV3Large() local 3069 const size_t a_shape[] = { 1, 14, 14, 80 }; in FP16MobileNetV3Large() local 3133 const size_t a_shape[] = { 1, 14, 14, 80 }; in FP16MobileNetV3Large() local 3197 const size_t a_shape[] = { 1, 14, 14, 80 }; in FP16MobileNetV3Large() local [all …]
|
H A D | fp32-sparse-mobilenet-v3-large.cc | 2720 const size_t a_shape[] = { 1, 16, 112, 112 }; in FP32SparseMobileNetV3Large() local 2794 const size_t a_shape[] = { 1, 24, 56, 56 }; in FP32SparseMobileNetV3Large() local 2858 const size_t a_shape[] = { 1, 72, 28, 28 }; in FP32SparseMobileNetV3Large() local 2932 const size_t a_shape[] = { 1, 120, 28, 28 }; in FP32SparseMobileNetV3Large() local 2956 const size_t a_shape[] = { 1, 40, 28, 28 }; in FP32SparseMobileNetV3Large() local 3020 const size_t a_shape[] = { 1, 120, 28, 28 }; in FP32SparseMobileNetV3Large() local 3044 const size_t a_shape[] = { 1, 40, 28, 28 }; in FP32SparseMobileNetV3Large() local 3158 const size_t a_shape[] = { 1, 80, 14, 14 }; in FP32SparseMobileNetV3Large() local 3222 const size_t a_shape[] = { 1, 80, 14, 14 }; in FP32SparseMobileNetV3Large() local 3286 const size_t a_shape[] = { 1, 80, 14, 14 }; in FP32SparseMobileNetV3Large() local [all …]
|
H A D | fp32-mobilenet-v3-small.cc | 2303 const size_t a_shape[] = { 1, 56, 56, 16 }; in FP32MobileNetV3Small() local 2387 const size_t a_shape[] = { 1, 28, 28, 24 }; in FP32MobileNetV3Small() local 2471 const size_t a_shape[] = { 1, 14, 14, 96 }; in FP32MobileNetV3Small() local 2565 const size_t a_shape[] = { 1, 14, 14, 240 }; in FP32MobileNetV3Small() local 2589 const size_t a_shape[] = { 1, 14, 14, 40 }; in FP32MobileNetV3Small() local 2673 const size_t a_shape[] = { 1, 14, 14, 240 }; in FP32MobileNetV3Small() local 2697 const size_t a_shape[] = { 1, 14, 14, 40 }; in FP32MobileNetV3Small() local 2781 const size_t a_shape[] = { 1, 14, 14, 120 }; in FP32MobileNetV3Small() local 2875 const size_t a_shape[] = { 1, 14, 14, 144 }; in FP32MobileNetV3Small() local 2899 const size_t a_shape[] = { 1, 14, 14, 48 }; in FP32MobileNetV3Small() local [all …]
|
H A D | fp16-mobilenet-v3-small.cc | 2306 const size_t a_shape[] = { 1, 56, 56, 16 }; in FP16MobileNetV3Small() local 2390 const size_t a_shape[] = { 1, 28, 28, 24 }; in FP16MobileNetV3Small() local 2474 const size_t a_shape[] = { 1, 14, 14, 96 }; in FP16MobileNetV3Small() local 2568 const size_t a_shape[] = { 1, 14, 14, 240 }; in FP16MobileNetV3Small() local 2592 const size_t a_shape[] = { 1, 14, 14, 40 }; in FP16MobileNetV3Small() local 2676 const size_t a_shape[] = { 1, 14, 14, 240 }; in FP16MobileNetV3Small() local 2700 const size_t a_shape[] = { 1, 14, 14, 40 }; in FP16MobileNetV3Small() local 2784 const size_t a_shape[] = { 1, 14, 14, 120 }; in FP16MobileNetV3Small() local 2878 const size_t a_shape[] = { 1, 14, 14, 144 }; in FP16MobileNetV3Small() local 2902 const size_t a_shape[] = { 1, 14, 14, 48 }; in FP16MobileNetV3Small() local [all …]
|
H A D | fp32-sparse-mobilenet-v3-small.cc | 2385 const size_t a_shape[] = { 1, 16, 56, 56 }; in FP32SparseMobileNetV3Small() local 2469 const size_t a_shape[] = { 1, 24, 28, 28 }; in FP32SparseMobileNetV3Small() local 2553 const size_t a_shape[] = { 1, 96, 14, 14 }; in FP32SparseMobileNetV3Small() local 2647 const size_t a_shape[] = { 1, 240, 14, 14 }; in FP32SparseMobileNetV3Small() local 2671 const size_t a_shape[] = { 1, 40, 14, 14 }; in FP32SparseMobileNetV3Small() local 2755 const size_t a_shape[] = { 1, 240, 14, 14 }; in FP32SparseMobileNetV3Small() local 2779 const size_t a_shape[] = { 1, 40, 14, 14 }; in FP32SparseMobileNetV3Small() local 2863 const size_t a_shape[] = { 1, 120, 14, 14 }; in FP32SparseMobileNetV3Small() local 2957 const size_t a_shape[] = { 1, 144, 14, 14 }; in FP32SparseMobileNetV3Small() local 2981 const size_t a_shape[] = { 1, 48, 14, 14 }; in FP32SparseMobileNetV3Small() local [all …]
|
H A D | qu8-mobilenet-v2.cc | 1889 const size_t a_shape[] = { 1, 56, 56, 24 }; in QU8MobileNetV2() local 1963 const size_t a_shape[] = { 1, 28, 28, 32 }; in QU8MobileNetV2() local 2007 const size_t a_shape[] = { 1, 28, 28, 32 }; in QU8MobileNetV2() local 2081 const size_t a_shape[] = { 1, 14, 14, 64 }; in QU8MobileNetV2() local 2125 const size_t a_shape[] = { 1, 14, 14, 64 }; in QU8MobileNetV2() local 2169 const size_t a_shape[] = { 1, 14, 14, 64 }; in QU8MobileNetV2() local 2243 const size_t a_shape[] = { 1, 14, 14, 96 }; in QU8MobileNetV2() local 2287 const size_t a_shape[] = { 1, 14, 14, 96 }; in QU8MobileNetV2() local 2361 const size_t a_shape[] = { 1, 7, 7, 160 }; in QU8MobileNetV2() local 2405 const size_t a_shape[] = { 1, 7, 7, 160 }; in QU8MobileNetV2() local
|
H A D | qs8-mobilenet-v2.cc | 1889 const size_t a_shape[] = { 1, 56, 56, 24 }; in QS8MobileNetV2() local 1963 const size_t a_shape[] = { 1, 28, 28, 32 }; in QS8MobileNetV2() local 2007 const size_t a_shape[] = { 1, 28, 28, 32 }; in QS8MobileNetV2() local 2081 const size_t a_shape[] = { 1, 14, 14, 64 }; in QS8MobileNetV2() local 2125 const size_t a_shape[] = { 1, 14, 14, 64 }; in QS8MobileNetV2() local 2169 const size_t a_shape[] = { 1, 14, 14, 64 }; in QS8MobileNetV2() local 2243 const size_t a_shape[] = { 1, 14, 14, 96 }; in QS8MobileNetV2() local 2287 const size_t a_shape[] = { 1, 14, 14, 96 }; in QS8MobileNetV2() local 2361 const size_t a_shape[] = { 1, 7, 7, 160 }; in QS8MobileNetV2() local 2405 const size_t a_shape[] = { 1, 7, 7, 160 }; in QS8MobileNetV2() local
|
H A D | fp16-mobilenet-v2.cc | 1817 const size_t a_shape[] = { 1, 56, 56, 24 }; in FP16MobileNetV2() local 1891 const size_t a_shape[] = { 1, 28, 28, 32 }; in FP16MobileNetV2() local 1935 const size_t a_shape[] = { 1, 28, 28, 32 }; in FP16MobileNetV2() local 2009 const size_t a_shape[] = { 1, 14, 14, 64 }; in FP16MobileNetV2() local 2053 const size_t a_shape[] = { 1, 14, 14, 64 }; in FP16MobileNetV2() local 2097 const size_t a_shape[] = { 1, 14, 14, 64 }; in FP16MobileNetV2() local 2171 const size_t a_shape[] = { 1, 14, 14, 96 }; in FP16MobileNetV2() local 2215 const size_t a_shape[] = { 1, 14, 14, 96 }; in FP16MobileNetV2() local 2289 const size_t a_shape[] = { 1, 7, 7, 160 }; in FP16MobileNetV2() local 2333 const size_t a_shape[] = { 1, 7, 7, 160 }; in FP16MobileNetV2() local
|
H A D | fp32-sparse-mobilenet-v2.cc | 1884 const size_t a_shape[] = { 1, 24, 56, 56 }; in FP32SparseMobileNetV2() local 1958 const size_t a_shape[] = { 1, 32, 28, 28 }; in FP32SparseMobileNetV2() local 2002 const size_t a_shape[] = { 1, 32, 28, 28 }; in FP32SparseMobileNetV2() local 2076 const size_t a_shape[] = { 1, 64, 14, 14 }; in FP32SparseMobileNetV2() local 2120 const size_t a_shape[] = { 1, 64, 14, 14 }; in FP32SparseMobileNetV2() local 2164 const size_t a_shape[] = { 1, 64, 14, 14 }; in FP32SparseMobileNetV2() local 2238 const size_t a_shape[] = { 1, 96, 14, 14 }; in FP32SparseMobileNetV2() local 2282 const size_t a_shape[] = { 1, 96, 14, 14 }; in FP32SparseMobileNetV2() local 2356 const size_t a_shape[] = { 1, 160, 7, 7 }; in FP32SparseMobileNetV2() local 2400 const size_t a_shape[] = { 1, 160, 7, 7 }; in FP32SparseMobileNetV2() local
|
H A D | fp32-mobilenet-v2.cc | 1817 const size_t a_shape[] = { 1, 56, 56, 24 }; in FP32MobileNetV2() local 1891 const size_t a_shape[] = { 1, 28, 28, 32 }; in FP32MobileNetV2() local 1935 const size_t a_shape[] = { 1, 28, 28, 32 }; in FP32MobileNetV2() local 2009 const size_t a_shape[] = { 1, 14, 14, 64 }; in FP32MobileNetV2() local 2053 const size_t a_shape[] = { 1, 14, 14, 64 }; in FP32MobileNetV2() local 2097 const size_t a_shape[] = { 1, 14, 14, 64 }; in FP32MobileNetV2() local 2171 const size_t a_shape[] = { 1, 14, 14, 96 }; in FP32MobileNetV2() local 2215 const size_t a_shape[] = { 1, 14, 14, 96 }; in FP32MobileNetV2() local 2289 const size_t a_shape[] = { 1, 7, 7, 160 }; in FP32MobileNetV2() local 2333 const size_t a_shape[] = { 1, 7, 7, 160 }; in FP32MobileNetV2() local
|
H A D | qc8-mobilenet-v2.cc | 1996 const size_t a_shape[] = { 1, 56, 56, 24 }; in QC8MobileNetV2() local 2070 const size_t a_shape[] = { 1, 28, 28, 32 }; in QC8MobileNetV2() local 2114 const size_t a_shape[] = { 1, 28, 28, 32 }; in QC8MobileNetV2() local 2188 const size_t a_shape[] = { 1, 14, 14, 64 }; in QC8MobileNetV2() local 2232 const size_t a_shape[] = { 1, 14, 14, 64 }; in QC8MobileNetV2() local 2276 const size_t a_shape[] = { 1, 14, 14, 64 }; in QC8MobileNetV2() local 2350 const size_t a_shape[] = { 1, 14, 14, 96 }; in QC8MobileNetV2() local 2394 const size_t a_shape[] = { 1, 14, 14, 96 }; in QC8MobileNetV2() local 2468 const size_t a_shape[] = { 1, 7, 7, 160 }; in QC8MobileNetV2() local 2512 const size_t a_shape[] = { 1, 7, 7, 160 }; in QC8MobileNetV2() local
|
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/math_ops/ |
H A D | batch_matmul_op_test.py | 76 def CompareNonEmpty(self, a_shape, b_shape): argument 96 def CompareNonEmpty(self, a_shape, b_shape): argument 114 def CompareEmpty(self, a_shape, b_shape): argument 174 def _compare(self, a_shape, b_shape, dtype, adjoint_a, adjoint_b): argument 184 def CheckGradients(self, a_shape, b_shape): argument 196 def CheckGradients(self, a_shape, b_shape): argument
|
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
H A D | sparse_csr_matrix_ops.cc | 235 ShapeHandle a_shape = sparse_matrix_shape_and_type.shape; in __anona88575120702() local 306 ShapeHandle a_shape = sparse_matrix_shape_and_type.shape; in __anona88575120802() local 359 ShapeHandle a_shape = sparse_matrix_shape_and_type.shape; in __anona88575120902() local 393 ShapeHandle a_shape = sparse_matrix_shape_and_type.shape; in __anona88575120a02() local
|
H A D | sparse_csr_matrix_ops_test.cc | 192 const string& a_shape, const string& b_shape) { in TEST() 228 const string& a_shape, const string& b_shape) { in TEST() 328 const string& a_shape, const string& b_shape) { in TEST()
|
H A D | sparse_ops.cc | 78 ShapeHandle a_shape; in __anone91b66370302() local 101 ShapeHandle a_shape; in __anone91b66370402() local
|
/aosp_15_r20/external/pytorch/torch/utils/ |
H A D | flop_counter.py | 56 def mm_flop(a_shape, b_shape, *args, out_shape=None, **kwargs) -> int: argument 67 def addmm_flop(self_shape, a_shape, b_shape, out_shape=None, **kwargs) -> int: argument 72 def bmm_flop(a_shape, b_shape, out_shape=None, **kwargs) -> int: argument 85 def baddbmm_flop(self_shape, a_shape, b_shape, out_shape=None, **kwargs) -> int: argument
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | sparse_tensor_dense_matmul_op_test.cc | 27 Node* a_shape, Node* b, bool adjoint_a, in SparseTensorDenseMatMulNode() 47 Tensor a_shape(DT_INT64, TensorShape({2})); in SparseTensorDenseMatmul() local
|
H A D | betainc_op.cc | 47 const TensorShape& a_shape = a.shape(); in Compute() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
H A D | beta_op.cc | 37 const TensorShape& a_shape = ctx->InputShape(0); in Compile() local
|
H A D | matmul_op.cc | 56 const TensorShape a_shape = ctx->InputShape(0); in Compile() local
|
/aosp_15_r20/external/pytorch/benchmarks/dynamo/microbenchmarks/ |
H A D | matmul_relu.py | 50 a_shape = a_shapes[i] variable
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
H A D | tracked_device_buffer_test.cc | 54 Shape a_shape = ShapeUtil::MakeShape(F32, {3, 101, 4}); in TEST() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
H A D | qr.cc | 117 Shape a_shape = a_shape_or.ValueOrDie(); in QrExplicit() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
H A D | cusolver_rewriter.cc | 57 Shape a_shape = operand->shape(); in CreateCholesky() local
|