Home
last modified time | relevance | path

Searched defs:tensor2 (Results 1 – 25 of 44) sorted by relevance

12

/aosp_15_r20/external/armnn/delegate/test/
H A DTestUtils.cpp11 void CompareData(bool tensor1[], bool tensor2[], size_t tensorSize) in CompareData()
20 void CompareData(std::vector<bool>& tensor1, std::vector<bool>& tensor2, size_t tensorSize) in CompareData()
29 void CompareData(float tensor1[], float tensor2[], size_t tensorSize) in CompareData()
37 void CompareData(float tensor1[], float tensor2[], size_t tensorSize, float percentTolerance) in CompareData()
46 void CompareData(uint8_t tensor1[], uint8_t tensor2[], size_t tensorSize) in CompareData()
55 void CompareData(int16_t tensor1[], int16_t tensor2[], size_t tensorSize) in CompareData()
64 void CompareData(int32_t tensor1[], int32_t tensor2[], size_t tensorSize) in CompareData()
73 void CompareData(int8_t tensor1[], int8_t tensor2[], size_t tensorSize) in CompareData()
82 void CompareData(Half tensor1[], Half tensor2[], size_t tensorSize) in CompareData()
90 void CompareData(TfLiteFloat16 tensor1[], TfLiteFloat16 tensor2[], size_t tensorSize) in CompareData()
[all …]
/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_morphing.cpp22 Tensor<float, 3> tensor2(2,3,7); in test_simple_reshape() local
75 TensorMap<Tensor<float, 5>> tensor2(m2.data(), 3,5,7,11,13); in test_reshape_in_expr() local
162 TensorMap<Tensor<T, 2, DataLayout>> tensor2(m2.data(), 3, 3); in test_slice_in_expr() local
193 Tensor<T, 3, DataLayout> tensor2(2,2,7); in test_slice_as_lvalue() local
351 Tensor<T, 2, DataLayout> tensor2(7,11); in test_strided_slice() local
443 Tensor<T, 2, DataLayout> tensor(7,11),tensor2(7,11); in test_strided_slice_write() local
H A Dcxx11_tensor_layout_swap.cpp21 Tensor<float, 3, RowMajor> tensor2 = tensor.swap_layout(); in test_simple_swap() local
41 Tensor<float, 3, RowMajor> tensor2(7,3,2); in test_swap_as_lvalue() local
H A Dcxx11_tensor_layout_swap_sycl.cpp38 Tensor<DataType, 3, RowMajor, IndexType> tensor2(tensorRowRange); in test_simple_swap_sycl() local
81 Tensor<DataType, 3, RowMajor, IndexType> tensor2(tensorRowRange); in test_swap_as_lvalue_sycl() local
H A Dcxx11_tensor_trace.cpp39 Tensor<float, 5, DataLayout> tensor2(7, 7, 7, 7, 7); in test_all_dimensions_trace() local
69 Tensor<float, 4, DataLayout> tensor2(5, 5, 7, 7); in test_simple_trace() local
H A Dcxx11_tensor_morphing_sycl.cpp40 Tensor<DataType, 3, DataLayout, IndexType> tensor2(dim2); in test_simple_reshape() local
267 Tensor<DataType, 2, DataLayout, IndexType> tensor(tensorRange),tensor2(tensorRange); in test_strided_slice_write_sycl() local
H A Dcxx11_tensor_move.cpp32 Tensor<int,3,RowMajor> tensor2(2, 2, 2); in test_move() local
H A Dcxx11_tensor_io.cpp74 Tensor<int, 1, DataLayout> tensor2(5); in test_output_expr() local
H A Dcxx11_tensor_custom_op.cpp88 Tensor<float, 3> tensor2(3,7,5); in test_custom_binary_op() local
H A Dcxx11_tensor_image_op_sycl.cpp41 Tensor<DataType, 3,DataLayout, IndexType> tensor2(input_range); in test_image_op_sycl() local
/aosp_15_r20/external/executorch/extension/tensor/test/
H A Dtensor_ptr_maker_test.cpp189 auto tensor2 = empty({4, 5}, exec_aten::ScalarType::Int); in TEST_F() local
216 auto tensor2 = full({4, 5}, 3, exec_aten::ScalarType::Int); in TEST_F() local
246 auto tensor2 = scalar_tensor(5, exec_aten::ScalarType::Int); in TEST_F() local
269 auto tensor2 = ones({4, 5}, exec_aten::ScalarType::Int); in TEST_F() local
299 auto tensor2 = zeros({4, 5}, exec_aten::ScalarType::Int); in TEST_F() local
H A Dtensor_ptr_test.cpp97 auto tensor2 = tensor1; in TEST_F() local
308 auto tensor2 = make_tensor_ptr({2, 2}, {1.0f, 2.0f, 3.0f, 4.0f}); in TEST_F() local
323 auto tensor2 = tensor1; in TEST_F() local
336 auto tensor2 = tensor1; in TEST_F() local
704 auto tensor2 = tensor1; in TEST_F() local
/aosp_15_r20/external/pytorch/test/cpp/lazy/
H A Dtest_lazy_ops_util.cpp43 bool EqualValues(at::Tensor tensor1, at::Tensor tensor2) { in EqualValues()
67 bool EqualValuesNoElementTypeCheck(at::Tensor tensor1, at::Tensor tensor2) { in EqualValuesNoElementTypeCheck()
96 at::Tensor tensor2, in CloseValues()
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DCPUApplyUtils.h261 inline void CPU_tensor_apply2(Tensor tensor1, Tensor tensor2, const Op op) { in CPU_tensor_apply2()
284 Tensor tensor2, in CPU_tensor_apply3()
316 Tensor tensor2, in CPU_tensor_apply4()
/aosp_15_r20/external/pytorch/test/distributed/tensor/parallel/
H A Dtest_tp_random_state.py95 def tp_weights_assert(tensor1, tensor2): argument
117 def dp_weights_assert(tensor1, tensor2): argument
/aosp_15_r20/external/executorch/extension/module/test/
H A Dmodule_test.cpp227 auto tensor2 = make_tensor_ptr({2.f}); in TEST_F() local
381 auto tensor2 = make_tensor_ptr({5.f}); in TEST_F() local
396 auto tensor2 = make_tensor_ptr({3.f}); in TEST_F() local
/aosp_15_r20/external/pytorch/test/
H A Dtest_utils.py344 def run_fn(tensor1, tensor2): argument
415 def run_fn(tensor1, tensor2): argument
424 def run_fn2(tensor1, tensor2): argument
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dtensor_flatten.cpp14 auto tensor2 = at::tensor({4, 5}); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dthreadpool_device_test.cc36 bool Equal(const Tensor& tensor1, const Tensor& tensor2) { in Equal()
/aosp_15_r20/external/tensorflow/tensorflow/lite/swift/Tests/
H A DTensorTests.swift54 var tensor2 = Tensor( in testEquatable() variable
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DLinearAlgebra.cpp1925 static bool should_fold(const Tensor& tensor1, const Tensor& tensor2, bool has_out) { in should_fold()
2001 const Tensor& tensor2) { in _matmul_impl()
2178 Tensor matmul(const Tensor & tensor1, const Tensor & tensor2) { in matmul()
2186 Tensor& matmul_out(const Tensor & tensor1, const Tensor & tensor2, Tensor &result) { in matmul_out()
2194 Tensor linalg_matmul(const Tensor & tensor1, const Tensor & tensor2) { in linalg_matmul()
2198 Tensor& linalg_matmul_out(const Tensor & tensor1, const Tensor & tensor2, Tensor &result) { in linalg_matmul_out()
/aosp_15_r20/external/armnn/src/armnnSerializer/test/
H A DSerializerTestUtils.cpp122 void CompareConstTensor(const armnn::ConstTensor& tensor1, const armnn::ConstTensor& tensor2) in CompareConstTensor()
/aosp_15_r20/external/pytorch/test/quantization/eager/
H A Dtest_equalize_eager.py14 def checkChannelsEqualized(self, tensor1, tensor2, output_axis, input_axis): argument
/aosp_15_r20/external/tensorflow/tensorflow/c/kernels/
H A Dsummary_op.cc113 bool IsSameSize(TF_Tensor* tensor1, TF_Tensor* tensor2) { in IsSameSize()
/aosp_15_r20/external/ComputeLibrary/tests/
H A DSimpleTensor.h442 void swap(SimpleTensor<U> &tensor1, SimpleTensor<U> &tensor2) in swap()

12