Home
last modified time | relevance | path

Searched full:tensor2 (Results 1 – 25 of 108) sorted by relevance

12345

/aosp_15_r20/external/armnn/delegate/test/
H A DTestUtils.cpp11 void CompareData(bool tensor1[], bool tensor2[], size_t tensorSize) in CompareData() argument
16 CHECK(compareBool(tensor1[i], tensor2[i])); in CompareData()
20 void CompareData(std::vector<bool>& tensor1, std::vector<bool>& tensor2, size_t tensorSize) in CompareData() argument
25 CHECK(compareBool(tensor1[i], tensor2[i])); in CompareData()
29 void CompareData(float tensor1[], float tensor2[], size_t tensorSize) in CompareData() argument
33 CHECK(tensor1[i] == doctest::Approx( tensor2[i] )); in CompareData()
37 void CompareData(float tensor1[], float tensor2[], size_t tensorSize, float percentTolerance) in CompareData() argument
41 CHECK(std::max(tensor1[i], tensor2[i]) - std::min(tensor1[i], tensor2[i]) <= in CompareData()
46 void CompareData(uint8_t tensor1[], uint8_t tensor2[], size_t tensorSize) in CompareData() argument
51 CHECK(std::max(tensor1[i], tensor2[i]) - std::min(tensor1[i], tensor2[i]) <= tolerance); in CompareData()
[all …]
H A DTestUtils.hpp25 void CompareData(std::vector<bool>& tensor1, std::vector<bool>& tensor2, size_t tensorSize);
26 void CompareData(bool tensor1[], bool tensor2[], size_t tensorSize);
29 void CompareData(float tensor1[], float tensor2[], size_t tensorSize);
32 void CompareData(float tensor1[], float tensor2[], size_t tensorSize, float percentTolerance);
35 void CompareData(int8_t tensor1[], int8_t tensor2[], size_t tensorSize);
38 void CompareData(uint8_t tensor1[], uint8_t tensor2[], size_t tensorSize);
41 void CompareData(int16_t tensor1[], int16_t tensor2[], size_t tensorSize);
44 void CompareData(int32_t tensor1[], int32_t tensor2[], size_t tensorSize);
47 void CompareData(Half tensor1[], Half tensor2[], size_t tensorSize);
50 void CompareData(TfLiteFloat16 tensor1[], TfLiteFloat16 tensor2[], size_t tensorSize);
[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() argument
45 tensor2 = ToCpuTensor(tensor2); in EqualValues()
47 EXPECT_TRUE(EqualValues(torch::isnan(tensor1), torch::isnan(tensor2))); in EqualValues()
49 tensor2.nan_to_num_(); in EqualValues()
51 if (tensor1.sizes() != tensor2.sizes() || in EqualValues()
52 tensor1.dtype() != tensor2.dtype()) { in EqualValues()
55 << tensor2.dtype() << " " << tensor2.sizes() << "\n"; in EqualValues()
59 at::ScalarType type2 = tensor2.scalar_type(); in EqualValues()
63 bool equal = tensor1.equal(tensor2); in EqualValues()
67 bool EqualValuesNoElementTypeCheck(at::Tensor tensor1, at::Tensor tensor2) { in EqualValuesNoElementTypeCheck() argument
[all …]
/aosp_15_r20/external/libaom/av1/encoder/
H A Dcnn.c906 TENSOR tensor2[CNN_MAX_BRANCHES] = { { 0 } }; in av1_cnn_predict_c() local
920 init_tensor(&tensor2[b]); in av1_cnn_predict_c()
935 // Swap tensor1 and tensor2 in av1_cnn_predict_c()
936 swap_tensor(&tensor1[branch], &tensor2[branch]); in av1_cnn_predict_c()
947 if (!realloc_tensor(&tensor2[branch], layer_config->out_channels, o_width, in av1_cnn_predict_c()
952 free_tensor(&tensor2[branch]); in av1_cnn_predict_c()
953 assign_tensor(&tensor2[branch], output[output_num], in av1_cnn_predict_c()
965 branch, tensor2)) { in av1_cnn_predict_c()
971 assert(tensor2[branch].channels == layer_config->out_channels); in av1_cnn_predict_c()
979 tensor2[branch].buf, tensor2[branch].stride); in av1_cnn_predict_c()
[all …]
/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_layout_swap_sycl.cpp38 Tensor<DataType, 3, RowMajor, IndexType> tensor2(tensorRowRange); in test_simple_swap_sycl() local
42 …DataType* gpu_data2 = static_cast<DataType*>(sycl_device.allocate(tensor2.size()*sizeof(DataType)… in test_simple_swap_sycl()
48 sycl_device.memcpyDeviceToHost(tensor2.data(), gpu_data2,(tensor2.size())*sizeof(DataType)); in test_simple_swap_sycl()
54 // Tensor<float, 3, RowMajor> tensor2 = tensor.swap_layout(); in test_simple_swap_sycl()
55 VERIFY_IS_EQUAL(tensor1.dimension(0), tensor2.dimension(2)); in test_simple_swap_sycl()
56 VERIFY_IS_EQUAL(tensor1.dimension(1), tensor2.dimension(1)); in test_simple_swap_sycl()
57 VERIFY_IS_EQUAL(tensor1.dimension(2), tensor2.dimension(0)); in test_simple_swap_sycl()
62 VERIFY_IS_EQUAL(tensor1(i,j,k), tensor2(k,j,i)); in test_simple_swap_sycl()
81 Tensor<DataType, 3, RowMajor, IndexType> tensor2(tensorRowRange); in test_swap_as_lvalue_sycl() local
85 …DataType* gpu_data2 = static_cast<DataType*>(sycl_device.allocate(tensor2.size()*sizeof(DataType)… in test_swap_as_lvalue_sycl()
[all …]
H A Dcxx11_tensor_layout_swap.cpp21 Tensor<float, 3, RowMajor> tensor2 = tensor.swap_layout(); in test_simple_swap() local
22 VERIFY_IS_EQUAL(tensor.dimension(0), tensor2.dimension(2)); in test_simple_swap()
23 VERIFY_IS_EQUAL(tensor.dimension(1), tensor2.dimension(1)); in test_simple_swap()
24 VERIFY_IS_EQUAL(tensor.dimension(2), tensor2.dimension(0)); in test_simple_swap()
29 VERIFY_IS_EQUAL(tensor(i,j,k), tensor2(k,j,i)); in test_simple_swap()
41 Tensor<float, 3, RowMajor> tensor2(7,3,2); in test_swap_as_lvalue() local
42 tensor2.swap_layout() = tensor; in test_swap_as_lvalue()
43 VERIFY_IS_EQUAL(tensor.dimension(0), tensor2.dimension(2)); in test_swap_as_lvalue()
44 VERIFY_IS_EQUAL(tensor.dimension(1), tensor2.dimension(1)); in test_swap_as_lvalue()
45 VERIFY_IS_EQUAL(tensor.dimension(2), tensor2.dimension(0)); in test_swap_as_lvalue()
[all …]
H A Dcxx11_tensor_morphing.cpp22 Tensor<float, 3> tensor2(2,3,7); in test_simple_reshape() local
27 tensor2 = tensor1.reshape(dim1); in test_simple_reshape()
36 VERIFY_IS_EQUAL(tensor1(i,j,0,k,0), tensor2(i,j,k)); in test_simple_reshape()
75 TensorMap<Tensor<float, 5>> tensor2(m2.data(), 3,5,7,11,13); in test_reshape_in_expr() local
81 tensor3 = tensor1.reshape(newDims1).contract(tensor2.reshape(newDims2), contract_along); in test_reshape_in_expr()
162 TensorMap<Tensor<T, 2, DataLayout>> tensor2(m2.data(), 3, 3); in test_slice_in_expr() local
171 …tensor3 = tensor1.slice(indices1, sizes1).contract(tensor2.slice(indices2, sizes2), contract_along… in test_slice_in_expr()
193 Tensor<T, 3, DataLayout> tensor2(2,2,7); in test_slice_as_lvalue() local
194 tensor2.setRandom(); in test_slice_as_lvalue()
207 result.slice(second_slice, sizes12).device(Eigen::DefaultDevice()) = tensor2; in test_slice_as_lvalue()
[all …]
H A Dcxx11_tensor_trace.cpp39 Tensor<float, 5, DataLayout> tensor2(7, 7, 7, 7, 7); in test_all_dimensions_trace() local
40 tensor2.setRandom(); in test_all_dimensions_trace()
42 Tensor<float, 0, DataLayout> result2 = tensor2.trace(dims); in test_all_dimensions_trace()
46 sum += tensor2(i, i, i, i, i); in test_all_dimensions_trace()
69 Tensor<float, 4, DataLayout> tensor2(5, 5, 7, 7); in test_simple_trace() local
70 tensor2.setRandom(); in test_simple_trace()
72 Tensor<float, 2, DataLayout> result2 = tensor2.trace(dims2); in test_simple_trace()
80 sum += tensor2(i, j, k, k); in test_simple_trace()
87 Tensor<float, 2, DataLayout> result3 = tensor2.trace(dims3); in test_simple_trace()
95 sum += tensor2(k, k, i, j); in test_simple_trace()
H A Dcxx11_tensor_image_op_sycl.cpp41 Tensor<DataType, 3,DataLayout, IndexType> tensor2(input_range); in test_image_op_sycl() local
58 tensor2.setRandom(); in test_image_op_sycl()
62 …DataType* gpu_data2 = static_cast<DataType*>(sycl_device.allocate(tensor2.size()*sizeof(DataType)… in test_image_op_sycl()
70 sycl_device.memcpyHostToDevice(gpu_data2, tensor2.data(),(tensor2.size())*sizeof(DataType)); in test_image_op_sycl()
74 …tensor3_cpu = tensor1.stridedSlice(indicesStart1,indicesStop1,strides1) - tensor2.stridedSlice(ind… in test_image_op_sycl()
H A Dcxx11_tensor_morphing_sycl.cpp40 Tensor<DataType, 3, DataLayout, IndexType> tensor2(dim2); in test_simple_reshape() local
47 …DataType* gpu_data2 = static_cast<DataType*>(sycl_device.allocate(tensor2.size()*sizeof(DataType)… in test_simple_reshape()
59 sycl_device.memcpyDeviceToHost(tensor2.data(), gpu_data2,(tensor1.size())*sizeof(DataType)); in test_simple_reshape()
69 VERIFY_IS_EQUAL(tensor1(i,j,0,k,0), tensor2(i,j,k)); ///ColMajor in test_simple_reshape()
75 //VERIFY_IS_EQUAL(tensor1(i,j,0,k,0), tensor2(i,j,k)); /// RowMajor in test_simple_reshape()
267 Tensor<DataType, 2, DataLayout, IndexType> tensor(tensorRange),tensor2(tensorRange); in test_strided_slice_write_sycl() local
278 …DataType* gpu_data2 = static_cast<DataType*>(sycl_device.allocate(tensor2.size()*sizeof(DataType)… in test_strided_slice_write_sycl()
296 sycl_device.memcpyDeviceToHost(tensor2.data(), gpu_data2,(tensor2.size())*sizeof(DataType)); in test_strided_slice_write_sycl()
300 VERIFY_IS_EQUAL(tensor(i,j), tensor2(i,j)); in test_strided_slice_write_sycl()
/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
190 EXPECT_EQ(tensor2->dim(), 2); in TEST_F()
191 EXPECT_EQ(tensor2->size(0), 4); in TEST_F()
192 EXPECT_EQ(tensor2->size(1), 5); in TEST_F()
193 EXPECT_EQ(tensor2->scalar_type(), exec_aten::ScalarType::Int); in TEST_F()
216 auto tensor2 = full({4, 5}, 3, exec_aten::ScalarType::Int); in TEST_F() local
217 EXPECT_EQ(tensor2->dim(), 2); in TEST_F()
218 EXPECT_EQ(tensor2->size(0), 4); in TEST_F()
219 EXPECT_EQ(tensor2->size(1), 5); in TEST_F()
220 EXPECT_EQ(tensor2->scalar_type(), exec_aten::ScalarType::Int); in TEST_F()
[all …]
H A Dtensor_ptr_test.cpp97 auto tensor2 = tensor1; in TEST_F() local
98 EXPECT_EQ(tensor1.get(), tensor2.get()); in TEST_F()
99 EXPECT_EQ(tensor1->unsafeGetTensorImpl(), tensor2->unsafeGetTensorImpl()); in TEST_F()
308 auto tensor2 = make_tensor_ptr({2, 2}, {1.0f, 2.0f, 3.0f, 4.0f}); in TEST_F() local
310 EXPECT_EQ(tensor2->dim(), 2); in TEST_F()
311 EXPECT_EQ(tensor2->size(0), 2); in TEST_F()
312 EXPECT_EQ(tensor2->size(1), 2); in TEST_F()
313 EXPECT_EQ(tensor2->strides()[0], 2); in TEST_F()
314 EXPECT_EQ(tensor2->strides()[1], 1); in TEST_F()
315 EXPECT_EQ(tensor2->const_data_ptr<float>()[0], 1.0f); in TEST_F()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Dmoving_averages_test.py179 # Note that tensor2 is not a Variable but just a plain Tensor resulting
181 tensor2 = var0 + var1
184 update = ema.apply([var0, var1, tensor2])
187 avg2 = ema.average(tensor2)
205 self.assertAllClose(_Repeat(10.0 + 30.0, dim), self.evaluate(tensor2))
392 tensor2 = v0 + v1
398 self.assertEqual("add/foo", ema.average_name(tensor2))
399 ema.apply([v0, v1, tensor2])
409 ema.average_name(tensor2), v2.op.name
416 ema.average_name(tensor2) + "/biased",
[all …]
/aosp_15_r20/external/executorch/runtime/core/exec_aten/testing_util/
H A Dtensor_util.h41 * Note that gtest users can write `EXPECT_THAT(tensor1, IsCloseTo(tensor2))` or
42 * `EXPECT_THAT(tensor1, Not(IsCloseTo(tensor2)))`, or use the helper macros
45 * For exact equality, use `EXPECT_THAT(tensor1, IsEqualTo(tensor2))` or
46 * `EXPECT_THAT(tensor1, Not(IsEqualTo(tensor2)))`, or the helper macros
81 * IsDataCloseTo(tensor2))` or `EXPECT_THAT(tensor1,
82 * Not(IsDataCloseTo(tensor2)))`, or use the helper macros
85 * For exact equality, use `EXPECT_THAT(tensor1, IsDataEqualTo(tensor2))` or
86 * `EXPECT_THAT(tensor1, Not(IsDataEqualTo(tensor2)))`, or the helper macros
121 * Lets gtest users write `EXPECT_THAT(tensor1, IsCloseTo(tensor2))` or
122 * `EXPECT_THAT(tensor1, Not(IsCloseTo(tensor2)))`.
[all …]
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DPointwiseOps.cpp20 const Tensor& tensor2, in TORCH_META_FUNC()
22 build_ternary_op(maybe_get_output(), self, tensor1, tensor2); in TORCH_META_FUNC()
28 const Tensor& tensor2, in TORCH_META_FUNC()
31 isIntegralType(tensor2.scalar_type(), /*includeBool=*/true)) { in TORCH_META_FUNC()
35 "release addcdiv will perform a true division of tensor1 and tensor2. ", in TORCH_META_FUNC()
37 "(input + value * torch.trunc(tensor1 / tensor2)).to(input.dtype) ", in TORCH_META_FUNC()
39 "(input + value * tensor1 / tensor2) for float inputs. ", in TORCH_META_FUNC()
41 "(input + value * tensor1 / tensor2), for all dtypes."); in TORCH_META_FUNC()
43 build_ternary_op(maybe_get_output(), self, tensor1, tensor2); in TORCH_META_FUNC()
52 const Tensor& tensor2, in TORCH_IMPL_FUNC()
[all …]
H A DLinearAlgebra.cpp1925 static bool should_fold(const Tensor& tensor1, const Tensor& tensor2, bool has_out) { in should_fold() argument
1928 const auto tensor1_larger = tensor1.dim() >= tensor2.dim(); in should_fold()
1931 // We can always transpose tensor2 as the dimensions are always >= 1 (precondition from matmul) in should_fold()
1932 // and tensor1_larger iff tensor2.dim() > tensor1.dim(9 in should_fold()
1934 : MaybeOwned<Tensor>::owned(tensor2.mT()); in should_fold()
1936 const auto dim_t2 = tensor1_larger ? tensor2.dim() in should_fold()
1953 bool t2_requires_grad = tensor1_larger ? tensor2.requires_grad() : tensor1.requires_grad(); in should_fold()
2001 const Tensor& tensor2) { in _matmul_impl() argument
2004 const auto dim_tensor2 = tensor2.dim(); in _matmul_impl()
2019 …TORCH_CHECK(!(tensor1.requires_grad() || tensor2.requires_grad() || out.requires_grad()) || !at::G… in _matmul_impl()
[all …]
/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() argument
262 if (!_apply_preamble({tensor1, tensor2})) in CPU_tensor_apply2()
264 if (_max_dim_tensors({tensor1, tensor2}) <= 8) { in CPU_tensor_apply2()
270 strided_tensor_iter_fixed<scalar2, 8>(tensor2)); in CPU_tensor_apply2()
277 strided_tensor_iter<scalar2>(tensor2)); in CPU_tensor_apply2()
284 Tensor tensor2, in CPU_tensor_apply3() argument
287 if (!_apply_preamble({tensor1, tensor2, tensor3})) in CPU_tensor_apply3()
289 if (_max_dim_tensors({tensor1, tensor2, tensor3}) <= 8) { in CPU_tensor_apply3()
295 strided_tensor_iter_fixed<scalar2, 8>(tensor2), in CPU_tensor_apply3()
303 strided_tensor_iter<scalar2>(tensor2), in CPU_tensor_apply3()
[all …]
/aosp_15_r20/external/pytorch/tools/autograd/
H A Ddeprecated.yaml31 - name: addcdiv(Tensor self, Scalar value, Tensor tensor1, Tensor tensor2) -> Tensor
32 aten: addcdiv(self, tensor1, tensor2, value)
34 - name: addcdiv_(Tensor(a!) self, Scalar value, Tensor tensor1, Tensor tensor2) -> Tensor(a!)
35 aten: addcdiv_(self, tensor1, tensor2, value)
37 - name: addcdiv(Tensor self, Scalar value, Tensor tensor1, Tensor tensor2, *, Tensor(a!) out) -> Te…
38 aten: addcdiv_out(out, self, tensor1, tensor2, value)
40 - name: addcmul(Tensor self, Scalar value, Tensor tensor1, Tensor tensor2) -> Tensor
41 aten: addcmul(self, tensor1, tensor2, value)
43 - name: addcmul_(Tensor(a!) self, Scalar value, Tensor tensor1, Tensor tensor2) -> Tensor(a!)
44 aten: addcmul_(self, tensor1, tensor2, value)
[all …]
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/
H A DPointwiseOps.mm18 const Tensor& tensor2,
47 string key = op_name + getTensorsStringKey({self, tensor1, tensor2});
51 …c10::promoteTypes(self.scalar_type(), c10::promoteTypes(tensor1.scalar_type(), tensor2.scalar_type…
54 newCachedGraph->secondTensor = mpsGraphRankedPlaceHolder(mpsGraph, tensor2);
83 Placeholder tensor2Placeholder = Placeholder(cachedGraph->secondTensor, tensor2);
107 (const Tensor& self, const Tensor& tensor1, const Tensor& tensor2, const Scalar& value, const Tenso…
108 mps::addc_mul_div_out_mps(self, tensor1, tensor2, value, output, false, "addcmul_out_mps");
112 (const Tensor& self, const Tensor& tensor1, const Tensor& tensor2, const Scalar& value, const Tenso…
113 mps::addc_mul_div_out_mps(self, tensor1, tensor2, value, output, true, "addcdiv_out_mps");
/aosp_15_r20/external/armnn/src/armnnSerializer/test/
H A DSerializerTestUtils.cpp122 void CompareConstTensor(const armnn::ConstTensor& tensor1, const armnn::ConstTensor& tensor2) in CompareConstTensor() argument
124 CHECK(tensor1.GetShape() == tensor2.GetShape()); in CompareConstTensor()
125 CHECK(GetDataTypeName(tensor1.GetDataType()) == GetDataTypeName(tensor2.GetDataType())); in CompareConstTensor()
131 tensor1.GetMemoryArea(), tensor2.GetMemoryArea(), tensor1.GetNumElements()); in CompareConstTensor()
136 tensor1.GetMemoryArea(), tensor2.GetMemoryArea(), tensor1.GetNumElements()); in CompareConstTensor()
140 tensor1.GetMemoryArea(), tensor2.GetMemoryArea(), tensor1.GetNumElements()); in CompareConstTensor()
144 tensor1.GetMemoryArea(), tensor2.GetMemoryArea(), tensor1.GetNumElements()); in CompareConstTensor()
/aosp_15_r20/external/ComputeLibrary/tests/
H A DSimpleTensor.h207 * @param[in, out] tensor2 Tensor to be swapped.
210 friend void swap(SimpleTensor<U> &tensor1, SimpleTensor<U> &tensor2);
442 void swap(SimpleTensor<U> &tensor1, SimpleTensor<U> &tensor2) in swap() argument
447 swap(tensor1._shape, tensor2._shape); in swap()
448 swap(tensor1._format, tensor2._format); in swap()
449 swap(tensor1._data_type, tensor2._data_type); in swap()
450 swap(tensor1._num_channels, tensor2._num_channels); in swap()
451 swap(tensor1._quantization_info, tensor2._quantization_info); in swap()
452 swap(tensor1._buffer, tensor2._buffer); in swap()
/aosp_15_r20/system/media/audio_utils/tests/
H A Daudio_nnapi_tests.cpp61 * tensor2 ---+ |
133 * (tensor0 + tensor1) * (tensor2 + tensor3).
139 * tensor2 ---+ |
245 uint32_t tensor2 = opIdx++; in CreateModel() local
248 "ANeuralNetworksModel_addOperand failed for operand (%d)", tensor2); in CreateModel()
311 tensor2, in CreateModel()
338 // Inputs: {tensor0, tensor1, tensor2, tensor3} in CreateModel()
343 tensor2, in CreateModel()
452 * @param inputValue2 value to fill tensor2
509 // Set all the elements of the third input tensor (tensor2) to the same value in Compute()
/aosp_15_r20/external/pytorch/test/distributed/tensor/parallel/
H A Dtest_tp_random_state.py95 def tp_weights_assert(tensor1, tensor2): argument
98 self.assertNotEqual(tensor1, tensor2)
102 self.assertEqual(tensor1, tensor2)
117 def dp_weights_assert(tensor1, tensor2): argument
120 self.assertEqual(tensor1, tensor2)
125 self.assertNotEqual(tensor1, tensor2)
/aosp_15_r20/external/tensorflow/tensorflow/lite/swift/Tests/
H A DTensorTests.swift54 var tensor2 = Tensor( in testEquatable() variable
61 XCTAssertEqual(tensor1, tensor2) in testEquatable()
63 tensor2 = Tensor( in testEquatable()
64 name: "Tensor2", in testEquatable()
70 XCTAssertNotEqual(tensor1, tensor2) in testEquatable()
/aosp_15_r20/external/pytorch/test/
H A Dtest_foreach.py674 tensor2 = torch.ones(10, 10, device="cuda:1", dtype=dtype)
678 foreach_op([tensor1], [tensor2])
684 foreach_op_([tensor1], [tensor2])
689 foreach_op_([tensor1], [tensor2])
700 tensor2 = make_tensor((1,), device=device, dtype=dtype).expand_as(tensor1)
701 inputs = ([tensor1], [tensor2])
725 tensor2 = torch.ones(10, 10, device=device, dtype=dtype)
726 inputs = ([tensor1], [tensor2.t()])
752 tensor2 = make_tensor(
756 self.assertFalse(tensor2.is_contiguous())
[all …]

12345