Home
last modified time | relevance | path

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

123

/external/armnn/delegate/test/
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 …]
/external/pytorch/test/distributed/tensor/parallel/
Dtest_tp_random_state.py95 def tp_weights_assert(tensor1, tensor2): argument
117 def dp_weights_assert(tensor1, tensor2): argument
/external/pytorch/test/cpp/lazy/
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()
Dtest_backend_device.cpp131 auto tensor1 = torch::rand({0, 1, 3, 0}); in TEST() local
/external/eigen/unsupported/test/
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
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
Dcxx11_tensor_move.cpp31 Tensor<int,3> tensor1(2, 2, 2); in test_move() local
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
Dcxx11_tensor_io.cpp73 Tensor<int, 1, DataLayout> tensor1(5); in test_output_expr() local
Dcxx11_tensor_custom_op.cpp86 Tensor<float, 3> tensor1(2,3,5); in test_custom_binary_op() local
Dcxx11_tensor_image_op_sycl.cpp40 Tensor<DataType, 3,DataLayout, IndexType> tensor1(input_range); in test_image_op_sycl() local
Dcxx11_tensor_assign.cpp233 Tensor<int, 1> tensor1; in test_auto_resize() local
/external/pytorch/aten/src/ATen/
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()
/external/pytorch/test/cpp/api/
Dtensor_flatten.cpp13 auto tensor1 = at::tensor({1, 2, 3}); in TEST() local
/external/tensorflow/tensorflow/core/common_runtime/
Dthreadpool_device_test.cc36 bool Equal(const Tensor& tensor1, const Tensor& tensor2) { in Equal()
/external/tensorflow/tensorflow/lite/swift/Tests/
DTensorTests.swift47 let tensor1 = Tensor( in testEquatable() variable
/external/armnn/src/armnnSerializer/test/
DSerializerTestUtils.cpp122 void CompareConstTensor(const armnn::ConstTensor& tensor1, const armnn::ConstTensor& tensor2) in CompareConstTensor()
/external/pytorch/test/quantization/eager/
Dtest_equalize_eager.py14 def checkChannelsEqualized(self, tensor1, tensor2, output_axis, input_axis): argument
/external/tensorflow/tensorflow/c/kernels/
Dsummary_op.cc113 bool IsSameSize(TF_Tensor* tensor1, TF_Tensor* tensor2) { in IsSameSize()
/external/ComputeLibrary/tests/
DSimpleTensor.h442 void swap(SimpleTensor<U> &tensor1, SimpleTensor<U> &tensor2) in swap()
/external/executorch/extension/tensor/test/
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
/external/pytorch/test/
Dtest_utils.py344 def run_fn(tensor1, tensor2): argument
415 def run_fn(tensor1, tensor2): argument
424 def run_fn2(tensor1, tensor2): argument
/external/pytorch/torch/_dynamo/polyfills/
D__init__.py165 def addcmul_inplace(self, tensor1, tensor2, value): argument
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dconcat_test.cc118 auto tensor1 = {7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f}; in TwoInputsNegativeAxisImpl() local
/external/tensorflow/tensorflow/lite/kernels/
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

123