Home
last modified time | relevance | path

Searched defs:a_shape (Results 1 – 25 of 47) sorted by relevance

12

/aosp_15_r20/external/XNNPACK/models/
H A Dfp32-mobilenet-v3-large.cc2628 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 Dfp16-mobilenet-v3-large.cc2631 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 Dfp32-sparse-mobilenet-v3-large.cc2720 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 Dfp32-mobilenet-v3-small.cc2303 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 Dfp16-mobilenet-v3-small.cc2306 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 Dfp32-sparse-mobilenet-v3-small.cc2385 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 Dqu8-mobilenet-v2.cc1889 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 Dqs8-mobilenet-v2.cc1889 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 Dfp16-mobilenet-v2.cc1817 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 Dfp32-sparse-mobilenet-v2.cc1884 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 Dfp32-mobilenet-v2.cc1817 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 Dqc8-mobilenet-v2.cc1996 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 Dbatch_matmul_op_test.py76 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 Dsparse_csr_matrix_ops.cc235 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 Dsparse_csr_matrix_ops_test.cc192 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 Dsparse_ops.cc78 ShapeHandle a_shape; in __anone91b66370302() local
101 ShapeHandle a_shape; in __anone91b66370402() local
/aosp_15_r20/external/pytorch/torch/utils/
H A Dflop_counter.py56 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 Dsparse_tensor_dense_matmul_op_test.cc27 Node* a_shape, Node* b, bool adjoint_a, in SparseTensorDenseMatMulNode()
47 Tensor a_shape(DT_INT64, TensorShape({2})); in SparseTensorDenseMatmul() local
H A Dbetainc_op.cc47 const TensorShape& a_shape = a.shape(); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dbeta_op.cc37 const TensorShape& a_shape = ctx->InputShape(0); in Compile() local
H A Dmatmul_op.cc56 const TensorShape a_shape = ctx->InputShape(0); in Compile() local
/aosp_15_r20/external/pytorch/benchmarks/dynamo/microbenchmarks/
H A Dmatmul_relu.py50 a_shape = a_shapes[i] variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtracked_device_buffer_test.cc54 Shape a_shape = ShapeUtil::MakeShape(F32, {3, 101, 4}); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dqr.cc117 Shape a_shape = a_shape_or.ValueOrDie(); in QrExplicit() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dcusolver_rewriter.cc57 Shape a_shape = operand->shape(); in CreateCholesky() local

12