Home
last modified time | relevance | path

Searched refs:tensor2 (Results 1 – 25 of 75) sorted by relevance

123

/external/armnn/delegate/test/
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 …]
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 …]
/external/pytorch/test/cpp/lazy/
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 …]
Dtest_lazy_ops_util.h32 bool EqualValues(at::Tensor tensor1, at::Tensor tensor2);
34 bool EqualValuesNoElementTypeCheck(at::Tensor tensor1, at::Tensor tensor2);
38 at::Tensor tensor2,
/external/eigen/unsupported/test/
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 …]
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()
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()
91 sycl_device.memcpyDeviceToHost(tensor2.data(), gpu_data2,(tensor2.size())*sizeof(DataType)); in test_swap_as_lvalue_sycl()
[all …]
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()
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 …]
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()
Dcxx11_tensor_move.cpp32 Tensor<int,3,RowMajor> tensor2(2, 2, 2); in test_move() local
38 tensor2(x,y,z) = 2 * i; in test_move()
43 Tensor<int,3,RowMajor> moved_tensor2 = std::move(tensor2); in test_move()
46 VERIFY_IS_EQUAL(tensor2.size(), 0); in test_move()
Dcxx11_tensor_custom_op.cpp88 Tensor<float, 3> tensor2(3,7,5); in test_custom_binary_op() local
89 tensor2.setRandom(); in test_custom_binary_op()
91 Tensor<float, 3> result = tensor1.customOp(tensor2, BatchMatMul()); in test_custom_binary_op()
96 Tensor<float, 2> reference = tensor1.chip<2>(i).contract(tensor2.chip<2>(i), dims); in test_custom_binary_op()
Dcxx11_tensor_io.cpp74 Tensor<int, 1, DataLayout> tensor2(5); in test_output_expr() local
77 tensor2(i) = 7; in test_output_expr()
81 os << tensor1 + tensor2; in test_output_expr()
/external/pytorch/aten/src/ATen/native/
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()
43 build_ternary_op(maybe_get_output(), self, tensor1, tensor2); in TORCH_META_FUNC()
52 const Tensor& tensor2, in TORCH_IMPL_FUNC()
61 const Tensor& tensor2, in TORCH_IMPL_FUNC()
/external/executorch/extension/tensor/test/
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 …]
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 …]
/external/libaom/av1/encoder/
Dcnn.c906 TENSOR tensor2[CNN_MAX_BRANCHES] = { { 0 } }; in av1_cnn_predict_c() local
920 init_tensor(&tensor2[b]); 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()
984 tensor2[branch].buf, tensor2[branch].stride, 0, 1); in av1_cnn_predict_c()
[all …]
/external/pytorch/aten/src/ATen/
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 …]
/external/tensorflow/tensorflow/python/training/
Dmoving_averages_test.py181 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",
417 ema.average_name(tensor2) + "/local_step"
[all …]
/external/pytorch/aten/src/ATen/native/mps/operations/
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");
/external/pytorch/test/distributed/tensor/parallel/
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)
/external/armnn/src/armnnSerializer/test/
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()
/external/tensorflow/tensorflow/lite/swift/Tests/
DTensorTests.swift54 var tensor2 = Tensor( in testEquatable() variable
61 XCTAssertEqual(tensor1, tensor2) in testEquatable()
63 tensor2 = Tensor( in testEquatable()
70 XCTAssertNotEqual(tensor1, tensor2) in testEquatable()
/external/ComputeLibrary/tests/
DSimpleTensor.h210 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()
/external/tensorflow/tensorflow/c/kernels/
Dsummary_op.cc65 bool IsSameSize(TF_Tensor* tensor1, TF_Tensor* tensor2);
113 bool IsSameSize(TF_Tensor* tensor1, TF_Tensor* tensor2) { in IsSameSize() argument
114 if (TF_NumDims(tensor1) != TF_NumDims(tensor2)) { in IsSameSize()
118 if (TF_Dim(tensor1, d) != TF_Dim(tensor2, d)) { in IsSameSize()
/external/pytorch/test/cpp/api/
Dtensor_flatten.cpp14 auto tensor2 = at::tensor({4, 5}); in TEST() local
16 std::vector<at::Tensor>({tensor1, emptyTensor1, emptyTensor2, tensor2}); in TEST()

123