Home
last modified time | relevance | path

Searched defs:a_shape (Results 1 – 25 of 33) 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.cc2554 const size_t a_shape[] = { 1, 112, 112, 16 }; in FP32MobileNetV3Large() local
2628 const size_t a_shape[] = { 1, 56, 56, 24 }; in FP32MobileNetV3Large() local
2692 const size_t a_shape[] = { 1, 28, 28, 72 }; in FP32MobileNetV3Large() local
2766 const size_t a_shape[] = { 1, 28, 28, 120 }; in FP32MobileNetV3Large() local
2790 const size_t a_shape[] = { 1, 28, 28, 40 }; in FP32MobileNetV3Large() local
2854 const size_t a_shape[] = { 1, 28, 28, 120 }; in FP32MobileNetV3Large() local
2878 const size_t a_shape[] = { 1, 28, 28, 40 }; in FP32MobileNetV3Large() local
2992 const size_t a_shape[] = { 1, 14, 14, 80 }; in FP32MobileNetV3Large() local
3056 const size_t a_shape[] = { 1, 14, 14, 80 }; in FP32MobileNetV3Large() local
3120 const size_t a_shape[] = { 1, 14, 14, 80 }; in FP32MobileNetV3Large() local
[all …]
Dfp16-mobilenet-v3-large.cc2557 const size_t a_shape[] = { 1, 112, 112, 16 }; in FP16MobileNetV3Large() local
2631 const size_t a_shape[] = { 1, 56, 56, 24 }; in FP16MobileNetV3Large() local
2695 const size_t a_shape[] = { 1, 28, 28, 72 }; in FP16MobileNetV3Large() local
2769 const size_t a_shape[] = { 1, 28, 28, 120 }; in FP16MobileNetV3Large() local
2793 const size_t a_shape[] = { 1, 28, 28, 40 }; in FP16MobileNetV3Large() local
2857 const size_t a_shape[] = { 1, 28, 28, 120 }; in FP16MobileNetV3Large() local
2881 const size_t a_shape[] = { 1, 28, 28, 40 }; in FP16MobileNetV3Large() local
2995 const size_t a_shape[] = { 1, 14, 14, 80 }; in FP16MobileNetV3Large() local
3059 const size_t a_shape[] = { 1, 14, 14, 80 }; in FP16MobileNetV3Large() local
3123 const size_t a_shape[] = { 1, 14, 14, 80 }; in FP16MobileNetV3Large() local
[all …]
Dfp32-sparse-mobilenet-v3-large.cc2646 const size_t a_shape[] = { 1, 16, 112, 112 }; in FP32SparseMobileNetV3Large() local
2720 const size_t a_shape[] = { 1, 24, 56, 56 }; in FP32SparseMobileNetV3Large() local
2784 const size_t a_shape[] = { 1, 72, 28, 28 }; in FP32SparseMobileNetV3Large() local
2858 const size_t a_shape[] = { 1, 120, 28, 28 }; in FP32SparseMobileNetV3Large() local
2882 const size_t a_shape[] = { 1, 40, 28, 28 }; in FP32SparseMobileNetV3Large() local
2946 const size_t a_shape[] = { 1, 120, 28, 28 }; in FP32SparseMobileNetV3Large() local
2970 const size_t a_shape[] = { 1, 40, 28, 28 }; in FP32SparseMobileNetV3Large() local
3084 const size_t a_shape[] = { 1, 80, 14, 14 }; in FP32SparseMobileNetV3Large() local
3148 const size_t a_shape[] = { 1, 80, 14, 14 }; in FP32SparseMobileNetV3Large() local
3212 const size_t a_shape[] = { 1, 80, 14, 14 }; in FP32SparseMobileNetV3Large() local
[all …]
Dfp32-sparse-mobilenet-v3-small.cc2321 const size_t a_shape[] = { 1, 16, 56, 56 }; in FP32SparseMobileNetV3Small() local
2405 const size_t a_shape[] = { 1, 24, 28, 28 }; in FP32SparseMobileNetV3Small() local
2489 const size_t a_shape[] = { 1, 96, 14, 14 }; in FP32SparseMobileNetV3Small() local
2583 const size_t a_shape[] = { 1, 240, 14, 14 }; in FP32SparseMobileNetV3Small() local
2607 const size_t a_shape[] = { 1, 40, 14, 14 }; in FP32SparseMobileNetV3Small() local
2691 const size_t a_shape[] = { 1, 240, 14, 14 }; in FP32SparseMobileNetV3Small() local
2715 const size_t a_shape[] = { 1, 40, 14, 14 }; in FP32SparseMobileNetV3Small() local
2799 const size_t a_shape[] = { 1, 120, 14, 14 }; in FP32SparseMobileNetV3Small() local
2893 const size_t a_shape[] = { 1, 144, 14, 14 }; in FP32SparseMobileNetV3Small() local
2917 const size_t a_shape[] = { 1, 48, 14, 14 }; in FP32SparseMobileNetV3Small() local
[all …]
Dfp16-mobilenet-v3-small.cc2242 const size_t a_shape[] = { 1, 56, 56, 16 }; in FP16MobileNetV3Small() local
2326 const size_t a_shape[] = { 1, 28, 28, 24 }; in FP16MobileNetV3Small() local
2410 const size_t a_shape[] = { 1, 14, 14, 96 }; in FP16MobileNetV3Small() local
2504 const size_t a_shape[] = { 1, 14, 14, 240 }; in FP16MobileNetV3Small() local
2528 const size_t a_shape[] = { 1, 14, 14, 40 }; in FP16MobileNetV3Small() local
2612 const size_t a_shape[] = { 1, 14, 14, 240 }; in FP16MobileNetV3Small() local
2636 const size_t a_shape[] = { 1, 14, 14, 40 }; in FP16MobileNetV3Small() local
2720 const size_t a_shape[] = { 1, 14, 14, 120 }; in FP16MobileNetV3Small() local
2814 const size_t a_shape[] = { 1, 14, 14, 144 }; in FP16MobileNetV3Small() local
2838 const size_t a_shape[] = { 1, 14, 14, 48 }; in FP16MobileNetV3Small() local
[all …]
Dfp32-mobilenet-v3-small.cc2239 const size_t a_shape[] = { 1, 56, 56, 16 }; in FP32MobileNetV3Small() local
2323 const size_t a_shape[] = { 1, 28, 28, 24 }; in FP32MobileNetV3Small() local
2407 const size_t a_shape[] = { 1, 14, 14, 96 }; in FP32MobileNetV3Small() local
2501 const size_t a_shape[] = { 1, 14, 14, 240 }; in FP32MobileNetV3Small() local
2525 const size_t a_shape[] = { 1, 14, 14, 40 }; in FP32MobileNetV3Small() local
2609 const size_t a_shape[] = { 1, 14, 14, 240 }; in FP32MobileNetV3Small() local
2633 const size_t a_shape[] = { 1, 14, 14, 40 }; in FP32MobileNetV3Small() local
2717 const size_t a_shape[] = { 1, 14, 14, 120 }; in FP32MobileNetV3Small() local
2811 const size_t a_shape[] = { 1, 14, 14, 144 }; in FP32MobileNetV3Small() local
2835 const size_t a_shape[] = { 1, 14, 14, 48 }; in FP32MobileNetV3Small() local
[all …]
Dfp32-mobilenet-v2.cc1751 const size_t a_shape[] = { 1, 56, 56, 24 }; in FP32MobileNetV2() local
1825 const size_t a_shape[] = { 1, 28, 28, 32 }; in FP32MobileNetV2() local
1869 const size_t a_shape[] = { 1, 28, 28, 32 }; in FP32MobileNetV2() local
1943 const size_t a_shape[] = { 1, 14, 14, 64 }; in FP32MobileNetV2() local
1987 const size_t a_shape[] = { 1, 14, 14, 64 }; in FP32MobileNetV2() local
2031 const size_t a_shape[] = { 1, 14, 14, 64 }; in FP32MobileNetV2() local
2105 const size_t a_shape[] = { 1, 14, 14, 96 }; in FP32MobileNetV2() local
2149 const size_t a_shape[] = { 1, 14, 14, 96 }; in FP32MobileNetV2() local
2223 const size_t a_shape[] = { 1, 7, 7, 160 }; in FP32MobileNetV2() local
2267 const size_t a_shape[] = { 1, 7, 7, 160 }; in FP32MobileNetV2() local
Dqs8-mobilenet-v2.cc1826 const size_t a_shape[] = { 1, 56, 56, 24 }; in QS8MobileNetV2() local
1900 const size_t a_shape[] = { 1, 28, 28, 32 }; in QS8MobileNetV2() local
1944 const size_t a_shape[] = { 1, 28, 28, 32 }; in QS8MobileNetV2() local
2018 const size_t a_shape[] = { 1, 14, 14, 64 }; in QS8MobileNetV2() local
2062 const size_t a_shape[] = { 1, 14, 14, 64 }; in QS8MobileNetV2() local
2106 const size_t a_shape[] = { 1, 14, 14, 64 }; in QS8MobileNetV2() local
2180 const size_t a_shape[] = { 1, 14, 14, 96 }; in QS8MobileNetV2() local
2224 const size_t a_shape[] = { 1, 14, 14, 96 }; in QS8MobileNetV2() local
2298 const size_t a_shape[] = { 1, 7, 7, 160 }; in QS8MobileNetV2() local
2342 const size_t a_shape[] = { 1, 7, 7, 160 }; in QS8MobileNetV2() local
Dfp16-mobilenet-v2.cc1754 const size_t a_shape[] = { 1, 56, 56, 24 }; in FP16MobileNetV2() local
1828 const size_t a_shape[] = { 1, 28, 28, 32 }; in FP16MobileNetV2() local
1872 const size_t a_shape[] = { 1, 28, 28, 32 }; in FP16MobileNetV2() local
1946 const size_t a_shape[] = { 1, 14, 14, 64 }; in FP16MobileNetV2() local
1990 const size_t a_shape[] = { 1, 14, 14, 64 }; in FP16MobileNetV2() local
2034 const size_t a_shape[] = { 1, 14, 14, 64 }; in FP16MobileNetV2() local
2108 const size_t a_shape[] = { 1, 14, 14, 96 }; in FP16MobileNetV2() local
2152 const size_t a_shape[] = { 1, 14, 14, 96 }; in FP16MobileNetV2() local
2226 const size_t a_shape[] = { 1, 7, 7, 160 }; in FP16MobileNetV2() local
2270 const size_t a_shape[] = { 1, 7, 7, 160 }; in FP16MobileNetV2() local
Dfp32-sparse-mobilenet-v2.cc1821 const size_t a_shape[] = { 1, 24, 56, 56 }; in FP32SparseMobileNetV2() local
1895 const size_t a_shape[] = { 1, 32, 28, 28 }; in FP32SparseMobileNetV2() local
1939 const size_t a_shape[] = { 1, 32, 28, 28 }; in FP32SparseMobileNetV2() local
2013 const size_t a_shape[] = { 1, 64, 14, 14 }; in FP32SparseMobileNetV2() local
2057 const size_t a_shape[] = { 1, 64, 14, 14 }; in FP32SparseMobileNetV2() local
2101 const size_t a_shape[] = { 1, 64, 14, 14 }; in FP32SparseMobileNetV2() local
2175 const size_t a_shape[] = { 1, 96, 14, 14 }; in FP32SparseMobileNetV2() local
2219 const size_t a_shape[] = { 1, 96, 14, 14 }; in FP32SparseMobileNetV2() local
2293 const size_t a_shape[] = { 1, 160, 7, 7 }; in FP32SparseMobileNetV2() local
2337 const size_t a_shape[] = { 1, 160, 7, 7 }; in FP32SparseMobileNetV2() local
/external/tensorflow/tensorflow/core/kernels/
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
Dbetainc_op.cc47 const TensorShape& a_shape = a.shape(); in Compute() local
Dsparse_tensor_dense_add_op.cc36 const Tensor *a_shape, const Tensor *b) { in ValidateInputs()
Dsparse_add_op.cc32 const Tensor *a_indices, *b_indices, *a_values_t, *b_values_t, *a_shape, in Compute() local
Dsparse_sparse_binary_op_shared.cc174 const auto a_shape = a_shape_t->flat<int64>(); in Compute() local
Dsparse_tensor_dense_matmul_op.cc45 const Tensor* a_shape; in Compute() 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.cc235 ShapeHandle a_shape = sparse_matrix_shape_and_type.shape; in __anon59a9d6c30702() local
306 ShapeHandle a_shape = sparse_matrix_shape_and_type.shape; in __anon59a9d6c30802() local
359 ShapeHandle a_shape = sparse_matrix_shape_and_type.shape; in __anon59a9d6c30902() local
393 ShapeHandle a_shape = sparse_matrix_shape_and_type.shape; in __anon59a9d6c30a02() local
Dsparse_ops.cc76 ShapeHandle a_shape; in __anon36a8fc880302() local
99 ShapeHandle a_shape; in __anon36a8fc880402() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbeta_op.cc37 const TensorShape& a_shape = ctx->InputShape(0); in Compile() local
Dmatmul_op.cc56 const TensorShape a_shape = ctx->InputShape(0); in Compile() local
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtracked_device_buffer_test.cc54 Shape a_shape = ShapeUtil::MakeShape(F32, {3, 101, 4}); in TEST() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcusolver_rewriter.cc55 Shape a_shape = operand->shape(); in CreateCholesky() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_types.cc38 bool GetCastCompatibleShape(llvm::ArrayRef<int64_t> a_shape, in GetCastCompatibleShape()

12