Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/kernel_tests/
Dbatch_matmul_op_test.py80 def CompareNonEmpty(self, a_shape, b_shape): argument
100 def CompareNonEmpty(self, a_shape, b_shape): argument
118 def CompareEmpty(self, a_shape, b_shape): argument
178 def _compare(self, a_shape, b_shape, dtype, adjoint_a, adjoint_b): argument
188 def CheckGradients(self, a_shape, b_shape): argument
200 def CheckGradients(self, a_shape, b_shape): argument
/external/XNNPACK/models/
Dfp32-mobilenet-v3-large.cc2555 const size_t b_shape[] = { 1, 112, 112, 16 }; in FP32MobileNetV3Large() local
2629 const size_t b_shape[] = { 1, 56, 56, 24 }; in FP32MobileNetV3Large() local
2693 const size_t b_shape[] = { 1, 1, 1, 72 }; in FP32MobileNetV3Large() local
2767 const size_t b_shape[] = { 1, 1, 1, 120 }; in FP32MobileNetV3Large() local
2791 const size_t b_shape[] = { 1, 28, 28, 40 }; in FP32MobileNetV3Large() local
2855 const size_t b_shape[] = { 1, 1, 1, 120 }; in FP32MobileNetV3Large() local
2879 const size_t b_shape[] = { 1, 28, 28, 40 }; in FP32MobileNetV3Large() local
2993 const size_t b_shape[] = { 1, 14, 14, 80 }; in FP32MobileNetV3Large() local
3057 const size_t b_shape[] = { 1, 14, 14, 80 }; in FP32MobileNetV3Large() local
3121 const size_t b_shape[] = { 1, 14, 14, 80 }; in FP32MobileNetV3Large() local
[all …]
Dfp16-mobilenet-v3-large.cc2558 const size_t b_shape[] = { 1, 112, 112, 16 }; in FP16MobileNetV3Large() local
2632 const size_t b_shape[] = { 1, 56, 56, 24 }; in FP16MobileNetV3Large() local
2696 const size_t b_shape[] = { 1, 1, 1, 72 }; in FP16MobileNetV3Large() local
2770 const size_t b_shape[] = { 1, 1, 1, 120 }; in FP16MobileNetV3Large() local
2794 const size_t b_shape[] = { 1, 28, 28, 40 }; in FP16MobileNetV3Large() local
2858 const size_t b_shape[] = { 1, 1, 1, 120 }; in FP16MobileNetV3Large() local
2882 const size_t b_shape[] = { 1, 28, 28, 40 }; in FP16MobileNetV3Large() local
2996 const size_t b_shape[] = { 1, 14, 14, 80 }; in FP16MobileNetV3Large() local
3060 const size_t b_shape[] = { 1, 14, 14, 80 }; in FP16MobileNetV3Large() local
3124 const size_t b_shape[] = { 1, 14, 14, 80 }; in FP16MobileNetV3Large() local
[all …]
Dfp32-sparse-mobilenet-v3-large.cc2647 const size_t b_shape[] = { 1, 16, 112, 112 }; in FP32SparseMobileNetV3Large() local
2721 const size_t b_shape[] = { 1, 24, 56, 56 }; in FP32SparseMobileNetV3Large() local
2785 const size_t b_shape[] = { 1, 72, 1, 1 }; in FP32SparseMobileNetV3Large() local
2859 const size_t b_shape[] = { 1, 120, 1, 1 }; in FP32SparseMobileNetV3Large() local
2883 const size_t b_shape[] = { 1, 40, 28, 28 }; in FP32SparseMobileNetV3Large() local
2947 const size_t b_shape[] = { 1, 120, 1, 1 }; in FP32SparseMobileNetV3Large() local
2971 const size_t b_shape[] = { 1, 40, 28, 28 }; in FP32SparseMobileNetV3Large() local
3085 const size_t b_shape[] = { 1, 80, 14, 14 }; in FP32SparseMobileNetV3Large() local
3149 const size_t b_shape[] = { 1, 80, 14, 14 }; in FP32SparseMobileNetV3Large() local
3213 const size_t b_shape[] = { 1, 80, 14, 14 }; in FP32SparseMobileNetV3Large() local
[all …]
Dfp32-sparse-mobilenet-v3-small.cc2322 const size_t b_shape[] = { 1, 16, 1, 1 }; in FP32SparseMobileNetV3Small() local
2406 const size_t b_shape[] = { 1, 24, 28, 28 }; in FP32SparseMobileNetV3Small() local
2490 const size_t b_shape[] = { 1, 96, 1, 1 }; in FP32SparseMobileNetV3Small() local
2584 const size_t b_shape[] = { 1, 240, 1, 1 }; in FP32SparseMobileNetV3Small() local
2608 const size_t b_shape[] = { 1, 40, 14, 14 }; in FP32SparseMobileNetV3Small() local
2692 const size_t b_shape[] = { 1, 240, 1, 1 }; in FP32SparseMobileNetV3Small() local
2716 const size_t b_shape[] = { 1, 40, 14, 14 }; in FP32SparseMobileNetV3Small() local
2800 const size_t b_shape[] = { 1, 120, 1, 1 }; in FP32SparseMobileNetV3Small() local
2894 const size_t b_shape[] = { 1, 144, 1, 1 }; in FP32SparseMobileNetV3Small() local
2918 const size_t b_shape[] = { 1, 48, 14, 14 }; in FP32SparseMobileNetV3Small() local
[all …]
Dfp16-mobilenet-v3-small.cc2243 const size_t b_shape[] = { 1, 1, 1, 16 }; in FP16MobileNetV3Small() local
2327 const size_t b_shape[] = { 1, 28, 28, 24 }; in FP16MobileNetV3Small() local
2411 const size_t b_shape[] = { 1, 1, 1, 96 }; in FP16MobileNetV3Small() local
2505 const size_t b_shape[] = { 1, 1, 1, 240 }; in FP16MobileNetV3Small() local
2529 const size_t b_shape[] = { 1, 14, 14, 40 }; in FP16MobileNetV3Small() local
2613 const size_t b_shape[] = { 1, 1, 1, 240 }; in FP16MobileNetV3Small() local
2637 const size_t b_shape[] = { 1, 14, 14, 40 }; in FP16MobileNetV3Small() local
2721 const size_t b_shape[] = { 1, 1, 1, 120 }; in FP16MobileNetV3Small() local
2815 const size_t b_shape[] = { 1, 1, 1, 144 }; in FP16MobileNetV3Small() local
2839 const size_t b_shape[] = { 1, 14, 14, 48 }; in FP16MobileNetV3Small() local
[all …]
Dfp32-mobilenet-v3-small.cc2240 const size_t b_shape[] = { 1, 1, 1, 16 }; in FP32MobileNetV3Small() local
2324 const size_t b_shape[] = { 1, 28, 28, 24 }; in FP32MobileNetV3Small() local
2408 const size_t b_shape[] = { 1, 1, 1, 96 }; in FP32MobileNetV3Small() local
2502 const size_t b_shape[] = { 1, 1, 1, 240 }; in FP32MobileNetV3Small() local
2526 const size_t b_shape[] = { 1, 14, 14, 40 }; in FP32MobileNetV3Small() local
2610 const size_t b_shape[] = { 1, 1, 1, 240 }; in FP32MobileNetV3Small() local
2634 const size_t b_shape[] = { 1, 14, 14, 40 }; in FP32MobileNetV3Small() local
2718 const size_t b_shape[] = { 1, 1, 1, 120 }; in FP32MobileNetV3Small() local
2812 const size_t b_shape[] = { 1, 1, 1, 144 }; in FP32MobileNetV3Small() local
2836 const size_t b_shape[] = { 1, 14, 14, 48 }; in FP32MobileNetV3Small() local
[all …]
Dfp32-mobilenet-v2.cc1752 const size_t b_shape[] = { 1, 56, 56, 24 }; in FP32MobileNetV2() local
1826 const size_t b_shape[] = { 1, 28, 28, 32 }; in FP32MobileNetV2() local
1870 const size_t b_shape[] = { 1, 28, 28, 32 }; in FP32MobileNetV2() local
1944 const size_t b_shape[] = { 1, 14, 14, 64 }; in FP32MobileNetV2() local
1988 const size_t b_shape[] = { 1, 14, 14, 64 }; in FP32MobileNetV2() local
2032 const size_t b_shape[] = { 1, 14, 14, 64 }; in FP32MobileNetV2() local
2106 const size_t b_shape[] = { 1, 14, 14, 96 }; in FP32MobileNetV2() local
2150 const size_t b_shape[] = { 1, 14, 14, 96 }; in FP32MobileNetV2() local
2224 const size_t b_shape[] = { 1, 7, 7, 160 }; in FP32MobileNetV2() local
2268 const size_t b_shape[] = { 1, 7, 7, 160 }; in FP32MobileNetV2() local
Dqs8-mobilenet-v2.cc1827 const size_t b_shape[] = { 1, 56, 56, 24 }; in QS8MobileNetV2() local
1901 const size_t b_shape[] = { 1, 28, 28, 32 }; in QS8MobileNetV2() local
1945 const size_t b_shape[] = { 1, 28, 28, 32 }; in QS8MobileNetV2() local
2019 const size_t b_shape[] = { 1, 14, 14, 64 }; in QS8MobileNetV2() local
2063 const size_t b_shape[] = { 1, 14, 14, 64 }; in QS8MobileNetV2() local
2107 const size_t b_shape[] = { 1, 14, 14, 64 }; in QS8MobileNetV2() local
2181 const size_t b_shape[] = { 1, 14, 14, 96 }; in QS8MobileNetV2() local
2225 const size_t b_shape[] = { 1, 14, 14, 96 }; in QS8MobileNetV2() local
2299 const size_t b_shape[] = { 1, 7, 7, 160 }; in QS8MobileNetV2() local
2343 const size_t b_shape[] = { 1, 7, 7, 160 }; in QS8MobileNetV2() local
Dfp16-mobilenet-v2.cc1755 const size_t b_shape[] = { 1, 56, 56, 24 }; in FP16MobileNetV2() local
1829 const size_t b_shape[] = { 1, 28, 28, 32 }; in FP16MobileNetV2() local
1873 const size_t b_shape[] = { 1, 28, 28, 32 }; in FP16MobileNetV2() local
1947 const size_t b_shape[] = { 1, 14, 14, 64 }; in FP16MobileNetV2() local
1991 const size_t b_shape[] = { 1, 14, 14, 64 }; in FP16MobileNetV2() local
2035 const size_t b_shape[] = { 1, 14, 14, 64 }; in FP16MobileNetV2() local
2109 const size_t b_shape[] = { 1, 14, 14, 96 }; in FP16MobileNetV2() local
2153 const size_t b_shape[] = { 1, 14, 14, 96 }; in FP16MobileNetV2() local
2227 const size_t b_shape[] = { 1, 7, 7, 160 }; in FP16MobileNetV2() local
2271 const size_t b_shape[] = { 1, 7, 7, 160 }; in FP16MobileNetV2() local
Dfp32-sparse-mobilenet-v2.cc1822 const size_t b_shape[] = { 1, 24, 56, 56 }; in FP32SparseMobileNetV2() local
1896 const size_t b_shape[] = { 1, 32, 28, 28 }; in FP32SparseMobileNetV2() local
1940 const size_t b_shape[] = { 1, 32, 28, 28 }; in FP32SparseMobileNetV2() local
2014 const size_t b_shape[] = { 1, 64, 14, 14 }; in FP32SparseMobileNetV2() local
2058 const size_t b_shape[] = { 1, 64, 14, 14 }; in FP32SparseMobileNetV2() local
2102 const size_t b_shape[] = { 1, 64, 14, 14 }; in FP32SparseMobileNetV2() local
2176 const size_t b_shape[] = { 1, 96, 14, 14 }; in FP32SparseMobileNetV2() local
2220 const size_t b_shape[] = { 1, 96, 14, 14 }; in FP32SparseMobileNetV2() local
2294 const size_t b_shape[] = { 1, 160, 7, 7 }; in FP32SparseMobileNetV2() local
2338 const size_t b_shape[] = { 1, 160, 7, 7 }; in FP32SparseMobileNetV2() local
/external/tensorflow/tensorflow/core/ops/
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()
Dsparse_csr_matrix_ops.cc241 ShapeHandle b_shape; in __anon59a9d6c30702() local
311 ShapeHandle b_shape; in __anon59a9d6c30802() local
367 ShapeHandle b_shape = sparse_matrix_shape_and_type.shape; in __anon59a9d6c30902() local
401 ShapeHandle b_shape = sparse_matrix_shape_and_type.shape; in __anon59a9d6c30a02() local
Dmath_ops.cc1586 ShapeHandle b_shape; in __anon725c8f640c02() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbeta_op.cc38 const TensorShape& b_shape = ctx->InputShape(1); in Compile() local
Dmatmul_op.cc57 const TensorShape b_shape = ctx->InputShape(1); in Compile() local
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtracked_device_buffer_test.cc55 Shape b_shape = ShapeUtil::MakeShape(S8, {77}); in TEST() local
/external/tensorflow/tensorflow/core/kernels/
Dbetainc_op.cc48 const TensorShape& b_shape = b.shape(); in Compute() local
Dsparse_add_op.cc33 *b_shape, *thresh_t; in Compute() local
Dsparse_sparse_binary_op_shared.cc175 const auto b_shape = b_shape_t->flat<int64>(); in Compute() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_types.cc39 llvm::ArrayRef<int64_t> b_shape, in GetCastCompatibleShape()
/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_mat_mul_op.cc129 TensorShape b_shape; in Compute() local
/external/tensorflow/tensorflow/core/kernels/rnn/
Dlstm_ops.cc66 Eigen::array<Eigen::DenseIndex, 2> b_shape({1, b.dimensions()[0]}); in LSTMBlockCellFpropWithEigen() local
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc311 ShapeHandle b_shape; in BatchMatMulV2Shape() local
348 ShapeHandle b_shape; in BatchMatMulShape() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.cc720 const Shape& b_shape = hlo->operand(1)->shape(); in HandleTriangularSolve() local

12