Home
last modified time | relevance | path

Searched defs:tensor1 (Results 1 – 25 of 53) sorted by relevance

123

/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/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()
95 at::Tensor tensor1, in CloseValues()
H A Dtest_backend_device.cpp131 auto tensor1 = torch::rand({0, 1, 3, 0}); in TEST() local
/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()
283 Tensor tensor1, in CPU_tensor_apply3()
315 Tensor tensor1, 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/eigen/unsupported/test/
H A Dcxx11_tensor_morphing.cpp19 Tensor<float, 5> tensor1(2,3,1,7,1); in test_simple_reshape() local
74 TensorMap<Tensor<float, 5>> tensor1(m1.data(), 2,3,5,7,11); in test_reshape_in_expr() local
161 TensorMap<Tensor<T, 2, DataLayout>> tensor1(m1.data(), 7, 7); in test_slice_in_expr() local
191 Tensor<T, 3, DataLayout> tensor1(2,2,7); in test_slice_as_lvalue() local
H A Dcxx11_tensor_layout_swap_sycl.cpp37 Tensor<DataType, 3, ColMajor, IndexType> tensor1(tensorColRange); in test_simple_swap_sycl() local
80 Tensor<DataType, 3, ColMajor, IndexType> tensor1(tensorColRange); in test_swap_as_lvalue_sycl() local
H A Dcxx11_tensor_trace.cpp29 Tensor<float, 3, DataLayout> tensor1(5, 5, 5); in test_all_dimensions_trace() local
54 Tensor<float, 3, DataLayout> tensor1(3, 5, 3); in test_simple_trace() local
H A Dcxx11_tensor_move.cpp31 Tensor<int,3> tensor1(2, 2, 2); in test_move() local
H A Dcxx11_tensor_io.cpp73 Tensor<int, 1, DataLayout> tensor1(5); in test_output_expr() local
H A Dcxx11_tensor_custom_op.cpp86 Tensor<float, 3> tensor1(2,3,5); in test_custom_binary_op() local
H A Dcxx11_tensor_image_op_sycl.cpp40 Tensor<DataType, 3,DataLayout, IndexType> tensor1(input_range); in test_image_op_sycl() local
/aosp_15_r20/external/executorch/extension/tensor/test/
H A Dtensor_ptr_test.cpp96 auto tensor1 = make_tensor_ptr({4, 5}, data); in TEST_F() local
322 auto tensor1 = make_tensor_ptr({2, 3}, std::move(data)); in TEST_F() local
335 auto tensor1 = make_tensor_ptr({3, 4}, std::move(data)); in TEST_F() local
703 auto tensor1 = make_tensor_ptr({10, 10}, data->data()); 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/tensorflow/tensorflow/lite/kernels/
H A Dconcatenation_test.cc213 auto tensor1 = {7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f}; in TEST() local
407 auto tensor1 = {7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f}; in TEST() local
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dtensor_flatten.cpp13 auto tensor1 = at::tensor({1, 2, 3}); in TEST() local
/aosp_15_r20/external/executorch/extension/module/test/
H A Dmodule_test.cpp380 auto tensor1 = make_tensor_ptr({4.f}); in TEST_F() local
395 auto tensor1 = make_tensor_ptr({2.f}); in TEST_F() 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.swift47 let tensor1 = 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()
2000 const Tensor& tensor1, 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()
/aosp_15_r20/system/media/audio_utils/tests/
H A Daudio_nnapi_tests.cpp237 uint32_t tensor1 = opIdx++; in CreateModel() local

123