| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | GEMMTranspose1xWFixture.h | 84 …TensorType compute_target(const TensorShape &shape_a, const TensorShape &shape_b, DataType data_ty… in compute_target()
115 SimpleTensor<T> compute_reference(const TensorShape &shape_a, DataType data_type) in compute_reference()
|
| H A D | GEMMInterleave4x4Fixture.h | 83 …TensorType compute_target(const TensorShape &shape_a, const TensorShape &shape_b, DataType data_ty… in compute_target()
113 …SimpleTensor<T> compute_reference(const TensorShape &shape_a, const TensorShape &shape_b, DataType… in compute_reference()
|
| H A D | GEMMLowpFixture.h | 235 …void setup(TensorShape shape_a, TensorShape shape_b, TensorShape shape_output, int32_t a_offset, i… in setup()
242 …TensorType compute_target(const TensorShape &shape_a, const TensorShape &shape_b, const TensorShap… in compute_target()
248 …SimpleTensor<int32_t> compute_reference(const TensorShape &shape_a, const TensorShape &shape_b, co… in compute_reference()
262 …void setup(TensorShape shape_a, TensorShape shape_b, TensorShape shape_output, int32_t a_offset, i… in setup()
301 …SimpleTensor<TI> compute_reference(const TensorShape &shape_a, const TensorShape &shape_b, const T… in compute_reference()
336 …void setup(TensorShape shape_a, TensorShape shape_b, TensorShape shape_output, int32_t a_offset, i… in setup()
|
| H A D | GEMMFixture.h | 56 …void setup(TensorShape shape_a, TensorShape shape_b, TensorShape shape_c, TensorShape output_shape… in setup()
86 …TensorType compute_target(const TensorShape &shape_a, const TensorShape &shape_b, const TensorShap… in compute_target()
151 …SimpleTensor<T> compute_reference(const TensorShape &shape_a, const TensorShape &shape_b, const Te… in compute_reference()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | horizontal_input_fusion.cc | 62 bool CompareShapeDimsFromLeftToRight(const Shape& shape_a, in CompareShapeDimsFromLeftToRight()
96 Shape shape_a = GetInputShapeForMultiOutputFusion(*a); in FindAndSortFusionCandidates() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/resource/ |
| H A D | resource_variable_test.cc | 99 std::vector<int> shape_a = {1}; in TEST() local
140 std::vector<int> shape_a = {1}; in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | partially_decluster_pass_test.cc | 301 Output shape_a = ops::Placeholder(s.WithOpName("shape_a"), DT_INT32, in TEST() local
353 Output shape_a = ops::Placeholder(s.WithOpName("shape_a"), DT_INT32, in TEST() local
378 Output shape_a = ops::Placeholder(s.WithOpName("shape_a"), DT_INT32, in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | process_broadcast_shapes.h | 92 const RuntimeShape* shape_a = in ProcessBroadcastShapes() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/NEON/ |
| H A D | GEMMLowp.cpp | 57 shape_a, shape_b, shape_c, a_offset, b_offset) in DATA_TEST_CASE() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/experimental/saved_model/internal/ |
| H A D | saved_model_api_test.cc | 227 const TF_Shape* shape_a = TF_TensorSpecShape(tensor_spec_a); in TEST_P() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | graph_properties_test.cc | 1978 const auto shape_a = properties.GetOutputProperties("a").at(0).shape(); in TEST_F() local
2062 const auto shape_a = properties.GetOutputProperties("a").at(0).shape(); in TEST_F() local
2170 const auto shape_a = properties.GetOutputProperties("a").at(0).shape(); in TEST_F() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A D | constant_fold.cpp | 412 auto shape_a = inputTensorValues[1].accessor<int64_t, 1>(); in runTorchBackendForOnnx() local
|
| /aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/lib/ |
| H A D | test_shape_base_.py | 793 def test_kron_shape(self, shape_a, shape_b): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
| H A D | spmd_expander_common.cc | 194 mlir::ArrayRef<int64_t> shape_a, mlir::ArrayRef<int64_t> shape_b, in GetBroadcastLayoutForElementWise()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_sparse.py | 3607 def grad_with_custom_sparsity_pattern_test_helper(sparse_dims, nnz, shape_a, shape_b): argument
3638 def test_sparse_matmul(sparse_dims, nnz, shape_a, shape_b): argument
3671 def fn(sparse_dims, nnz, shape_a, shape_b): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | shape_util_test.cc | 204 Shape shape_a = ShapeUtil::MakeShape(F32, {4, 3}, {true, false}); in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | dynamic_ragged_shape.py | 1718 def _row_partitions_identical(shape_a, shape_b): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes_test.cc | 3065 std::vector<int> shape_a; member
3411 std::vector<int> shape_a; in TEST_P() member
|