Home
last modified time | relevance | path

Searched defs:b_shape (Results 1 – 25 of 42) sorted by relevance

12

/aosp_15_r20/external/XNNPACK/models/
H A Dfp32-mobilenet-v3-large.cc2629 const size_t b_shape[] = { 1, 112, 112, 16 }; in FP32MobileNetV3Large() local
2703 const size_t b_shape[] = { 1, 56, 56, 24 }; in FP32MobileNetV3Large() local
2767 const size_t b_shape[] = { 1, 1, 1, 72 }; in FP32MobileNetV3Large() local
2841 const size_t b_shape[] = { 1, 1, 1, 120 }; in FP32MobileNetV3Large() local
2865 const size_t b_shape[] = { 1, 28, 28, 40 }; in FP32MobileNetV3Large() local
2929 const size_t b_shape[] = { 1, 1, 1, 120 }; in FP32MobileNetV3Large() local
2953 const size_t b_shape[] = { 1, 28, 28, 40 }; in FP32MobileNetV3Large() local
3067 const size_t b_shape[] = { 1, 14, 14, 80 }; in FP32MobileNetV3Large() local
3131 const size_t b_shape[] = { 1, 14, 14, 80 }; in FP32MobileNetV3Large() local
3195 const size_t b_shape[] = { 1, 14, 14, 80 }; in FP32MobileNetV3Large() local
[all …]
H A Dfp16-mobilenet-v3-large.cc2632 const size_t b_shape[] = { 1, 112, 112, 16 }; in FP16MobileNetV3Large() local
2706 const size_t b_shape[] = { 1, 56, 56, 24 }; in FP16MobileNetV3Large() local
2770 const size_t b_shape[] = { 1, 1, 1, 72 }; in FP16MobileNetV3Large() local
2844 const size_t b_shape[] = { 1, 1, 1, 120 }; in FP16MobileNetV3Large() local
2868 const size_t b_shape[] = { 1, 28, 28, 40 }; in FP16MobileNetV3Large() local
2932 const size_t b_shape[] = { 1, 1, 1, 120 }; in FP16MobileNetV3Large() local
2956 const size_t b_shape[] = { 1, 28, 28, 40 }; in FP16MobileNetV3Large() local
3070 const size_t b_shape[] = { 1, 14, 14, 80 }; in FP16MobileNetV3Large() local
3134 const size_t b_shape[] = { 1, 14, 14, 80 }; in FP16MobileNetV3Large() local
3198 const size_t b_shape[] = { 1, 14, 14, 80 }; in FP16MobileNetV3Large() local
[all …]
H A Dfp32-sparse-mobilenet-v3-large.cc2721 const size_t b_shape[] = { 1, 16, 112, 112 }; in FP32SparseMobileNetV3Large() local
2795 const size_t b_shape[] = { 1, 24, 56, 56 }; in FP32SparseMobileNetV3Large() local
2859 const size_t b_shape[] = { 1, 72, 1, 1 }; in FP32SparseMobileNetV3Large() local
2933 const size_t b_shape[] = { 1, 120, 1, 1 }; in FP32SparseMobileNetV3Large() local
2957 const size_t b_shape[] = { 1, 40, 28, 28 }; in FP32SparseMobileNetV3Large() local
3021 const size_t b_shape[] = { 1, 120, 1, 1 }; in FP32SparseMobileNetV3Large() local
3045 const size_t b_shape[] = { 1, 40, 28, 28 }; in FP32SparseMobileNetV3Large() local
3159 const size_t b_shape[] = { 1, 80, 14, 14 }; in FP32SparseMobileNetV3Large() local
3223 const size_t b_shape[] = { 1, 80, 14, 14 }; in FP32SparseMobileNetV3Large() local
3287 const size_t b_shape[] = { 1, 80, 14, 14 }; in FP32SparseMobileNetV3Large() local
[all …]
H A Dfp32-mobilenet-v3-small.cc2304 const size_t b_shape[] = { 1, 1, 1, 16 }; in FP32MobileNetV3Small() local
2388 const size_t b_shape[] = { 1, 28, 28, 24 }; in FP32MobileNetV3Small() local
2472 const size_t b_shape[] = { 1, 1, 1, 96 }; in FP32MobileNetV3Small() local
2566 const size_t b_shape[] = { 1, 1, 1, 240 }; in FP32MobileNetV3Small() local
2590 const size_t b_shape[] = { 1, 14, 14, 40 }; in FP32MobileNetV3Small() local
2674 const size_t b_shape[] = { 1, 1, 1, 240 }; in FP32MobileNetV3Small() local
2698 const size_t b_shape[] = { 1, 14, 14, 40 }; in FP32MobileNetV3Small() local
2782 const size_t b_shape[] = { 1, 1, 1, 120 }; in FP32MobileNetV3Small() local
2876 const size_t b_shape[] = { 1, 1, 1, 144 }; in FP32MobileNetV3Small() local
2900 const size_t b_shape[] = { 1, 14, 14, 48 }; in FP32MobileNetV3Small() local
[all …]
H A Dfp16-mobilenet-v3-small.cc2307 const size_t b_shape[] = { 1, 1, 1, 16 }; in FP16MobileNetV3Small() local
2391 const size_t b_shape[] = { 1, 28, 28, 24 }; in FP16MobileNetV3Small() local
2475 const size_t b_shape[] = { 1, 1, 1, 96 }; in FP16MobileNetV3Small() local
2569 const size_t b_shape[] = { 1, 1, 1, 240 }; in FP16MobileNetV3Small() local
2593 const size_t b_shape[] = { 1, 14, 14, 40 }; in FP16MobileNetV3Small() local
2677 const size_t b_shape[] = { 1, 1, 1, 240 }; in FP16MobileNetV3Small() local
2701 const size_t b_shape[] = { 1, 14, 14, 40 }; in FP16MobileNetV3Small() local
2785 const size_t b_shape[] = { 1, 1, 1, 120 }; in FP16MobileNetV3Small() local
2879 const size_t b_shape[] = { 1, 1, 1, 144 }; in FP16MobileNetV3Small() local
2903 const size_t b_shape[] = { 1, 14, 14, 48 }; in FP16MobileNetV3Small() local
[all …]
H A Dfp32-sparse-mobilenet-v3-small.cc2386 const size_t b_shape[] = { 1, 16, 1, 1 }; in FP32SparseMobileNetV3Small() local
2470 const size_t b_shape[] = { 1, 24, 28, 28 }; in FP32SparseMobileNetV3Small() local
2554 const size_t b_shape[] = { 1, 96, 1, 1 }; in FP32SparseMobileNetV3Small() local
2648 const size_t b_shape[] = { 1, 240, 1, 1 }; in FP32SparseMobileNetV3Small() local
2672 const size_t b_shape[] = { 1, 40, 14, 14 }; in FP32SparseMobileNetV3Small() local
2756 const size_t b_shape[] = { 1, 240, 1, 1 }; in FP32SparseMobileNetV3Small() local
2780 const size_t b_shape[] = { 1, 40, 14, 14 }; in FP32SparseMobileNetV3Small() local
2864 const size_t b_shape[] = { 1, 120, 1, 1 }; in FP32SparseMobileNetV3Small() local
2958 const size_t b_shape[] = { 1, 144, 1, 1 }; in FP32SparseMobileNetV3Small() local
2982 const size_t b_shape[] = { 1, 48, 14, 14 }; in FP32SparseMobileNetV3Small() local
[all …]
H A Dqu8-mobilenet-v2.cc1890 const size_t b_shape[] = { 1, 56, 56, 24 }; in QU8MobileNetV2() local
1964 const size_t b_shape[] = { 1, 28, 28, 32 }; in QU8MobileNetV2() local
2008 const size_t b_shape[] = { 1, 28, 28, 32 }; in QU8MobileNetV2() local
2082 const size_t b_shape[] = { 1, 14, 14, 64 }; in QU8MobileNetV2() local
2126 const size_t b_shape[] = { 1, 14, 14, 64 }; in QU8MobileNetV2() local
2170 const size_t b_shape[] = { 1, 14, 14, 64 }; in QU8MobileNetV2() local
2244 const size_t b_shape[] = { 1, 14, 14, 96 }; in QU8MobileNetV2() local
2288 const size_t b_shape[] = { 1, 14, 14, 96 }; in QU8MobileNetV2() local
2362 const size_t b_shape[] = { 1, 7, 7, 160 }; in QU8MobileNetV2() local
2406 const size_t b_shape[] = { 1, 7, 7, 160 }; in QU8MobileNetV2() local
H A Dqs8-mobilenet-v2.cc1890 const size_t b_shape[] = { 1, 56, 56, 24 }; in QS8MobileNetV2() local
1964 const size_t b_shape[] = { 1, 28, 28, 32 }; in QS8MobileNetV2() local
2008 const size_t b_shape[] = { 1, 28, 28, 32 }; in QS8MobileNetV2() local
2082 const size_t b_shape[] = { 1, 14, 14, 64 }; in QS8MobileNetV2() local
2126 const size_t b_shape[] = { 1, 14, 14, 64 }; in QS8MobileNetV2() local
2170 const size_t b_shape[] = { 1, 14, 14, 64 }; in QS8MobileNetV2() local
2244 const size_t b_shape[] = { 1, 14, 14, 96 }; in QS8MobileNetV2() local
2288 const size_t b_shape[] = { 1, 14, 14, 96 }; in QS8MobileNetV2() local
2362 const size_t b_shape[] = { 1, 7, 7, 160 }; in QS8MobileNetV2() local
2406 const size_t b_shape[] = { 1, 7, 7, 160 }; in QS8MobileNetV2() local
H A Dfp16-mobilenet-v2.cc1818 const size_t b_shape[] = { 1, 56, 56, 24 }; in FP16MobileNetV2() local
1892 const size_t b_shape[] = { 1, 28, 28, 32 }; in FP16MobileNetV2() local
1936 const size_t b_shape[] = { 1, 28, 28, 32 }; in FP16MobileNetV2() local
2010 const size_t b_shape[] = { 1, 14, 14, 64 }; in FP16MobileNetV2() local
2054 const size_t b_shape[] = { 1, 14, 14, 64 }; in FP16MobileNetV2() local
2098 const size_t b_shape[] = { 1, 14, 14, 64 }; in FP16MobileNetV2() local
2172 const size_t b_shape[] = { 1, 14, 14, 96 }; in FP16MobileNetV2() local
2216 const size_t b_shape[] = { 1, 14, 14, 96 }; in FP16MobileNetV2() local
2290 const size_t b_shape[] = { 1, 7, 7, 160 }; in FP16MobileNetV2() local
2334 const size_t b_shape[] = { 1, 7, 7, 160 }; in FP16MobileNetV2() local
H A Dfp32-sparse-mobilenet-v2.cc1885 const size_t b_shape[] = { 1, 24, 56, 56 }; in FP32SparseMobileNetV2() local
1959 const size_t b_shape[] = { 1, 32, 28, 28 }; in FP32SparseMobileNetV2() local
2003 const size_t b_shape[] = { 1, 32, 28, 28 }; in FP32SparseMobileNetV2() local
2077 const size_t b_shape[] = { 1, 64, 14, 14 }; in FP32SparseMobileNetV2() local
2121 const size_t b_shape[] = { 1, 64, 14, 14 }; in FP32SparseMobileNetV2() local
2165 const size_t b_shape[] = { 1, 64, 14, 14 }; in FP32SparseMobileNetV2() local
2239 const size_t b_shape[] = { 1, 96, 14, 14 }; in FP32SparseMobileNetV2() local
2283 const size_t b_shape[] = { 1, 96, 14, 14 }; in FP32SparseMobileNetV2() local
2357 const size_t b_shape[] = { 1, 160, 7, 7 }; in FP32SparseMobileNetV2() local
2401 const size_t b_shape[] = { 1, 160, 7, 7 }; in FP32SparseMobileNetV2() local
H A Dfp32-mobilenet-v2.cc1818 const size_t b_shape[] = { 1, 56, 56, 24 }; in FP32MobileNetV2() local
1892 const size_t b_shape[] = { 1, 28, 28, 32 }; in FP32MobileNetV2() local
1936 const size_t b_shape[] = { 1, 28, 28, 32 }; in FP32MobileNetV2() local
2010 const size_t b_shape[] = { 1, 14, 14, 64 }; in FP32MobileNetV2() local
2054 const size_t b_shape[] = { 1, 14, 14, 64 }; in FP32MobileNetV2() local
2098 const size_t b_shape[] = { 1, 14, 14, 64 }; in FP32MobileNetV2() local
2172 const size_t b_shape[] = { 1, 14, 14, 96 }; in FP32MobileNetV2() local
2216 const size_t b_shape[] = { 1, 14, 14, 96 }; in FP32MobileNetV2() local
2290 const size_t b_shape[] = { 1, 7, 7, 160 }; in FP32MobileNetV2() local
2334 const size_t b_shape[] = { 1, 7, 7, 160 }; in FP32MobileNetV2() local
H A Dqc8-mobilenet-v2.cc1997 const size_t b_shape[] = { 1, 56, 56, 24 }; in QC8MobileNetV2() local
2071 const size_t b_shape[] = { 1, 28, 28, 32 }; in QC8MobileNetV2() local
2115 const size_t b_shape[] = { 1, 28, 28, 32 }; in QC8MobileNetV2() local
2189 const size_t b_shape[] = { 1, 14, 14, 64 }; in QC8MobileNetV2() local
2233 const size_t b_shape[] = { 1, 14, 14, 64 }; in QC8MobileNetV2() local
2277 const size_t b_shape[] = { 1, 14, 14, 64 }; in QC8MobileNetV2() local
2351 const size_t b_shape[] = { 1, 14, 14, 96 }; in QC8MobileNetV2() local
2395 const size_t b_shape[] = { 1, 14, 14, 96 }; in QC8MobileNetV2() local
2469 const size_t b_shape[] = { 1, 7, 7, 160 }; in QC8MobileNetV2() local
2513 const size_t b_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/ComputeLibrary/tests/validation/fixtures/
H A DBatchNormalizationLayerFusionFixture.h49 …void setup(TensorShape src_shape, TensorShape w_shape, TensorShape b_shape, TensorShape dst_shape,… in setup()
83 …TensorType compute_target(TensorShape src_shape, TensorShape w_shape, TensorShape b_shape, TensorS… in compute_target()
159 …r<T> compute_reference(TensorShape src_shape, TensorShape w_shape, TensorShape b_shape, TensorShap… in compute_reference()
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dsparse_csr_matrix_ops.cc241 ShapeHandle b_shape; in __anona88575120702() local
311 ShapeHandle b_shape; in __anona88575120802() local
367 ShapeHandle b_shape = sparse_matrix_shape_and_type.shape; in __anona88575120902() local
401 ShapeHandle b_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()
/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/compiler/tf2xla/kernels/
H A Dbeta_op.cc38 const TensorShape& b_shape = ctx->InputShape(1); in Compile() local
H A Dmatmul_op.cc57 const TensorShape b_shape = ctx->InputShape(1); in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dtriangular_solve_rewriter.cc40 const Shape& b_shape = instr->operand(1)->shape(); in Run() local
/aosp_15_r20/external/pytorch/benchmarks/dynamo/microbenchmarks/
H A Dmatmul_relu.py51 b_shape = b_shapes[i] variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtracked_device_buffer_test.cc55 Shape b_shape = ShapeUtil::MakeShape(S8, {77}); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dbetainc_op.cc48 const TensorShape& b_shape = b.shape(); in Compute() local
H A Dsparse_add_op.cc34 *b_shape, *thresh_t; in Compute() local
H A Dsparse_sparse_binary_op_shared.cc184 const auto b_shape = b_shape_t->flat<int64_t>(); in Compute() local

12