Home
last modified time | relevance | path

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

/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_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()
53 TensorMap<Tensor<float, 5>> tensor2(m2.data(), 3,5,7,11,13); in test_reshape_in_expr() local
59 tensor3 = tensor1.reshape(newDims1).contract(tensor2.reshape(newDims2), contract_along); in test_reshape_in_expr()
140 TensorMap<Tensor<float, 2, DataLayout>> tensor2(m2.data(), 3, 3); in test_slice_in_expr() local
149 …tensor3 = tensor1.slice(indices1, sizes1).contract(tensor2.slice(indices2, sizes2), contract_along… in test_slice_in_expr()
171 Tensor<float, 3, DataLayout> tensor2(2,2,7); in test_slice_as_lvalue() local
172 tensor2.setRandom(); in test_slice_as_lvalue()
185 result.slice(second_slice, sizes12).device(Eigen::DefaultDevice()) = tensor2; in test_slice_as_lvalue()
[all …]
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()
Dcxx11_tensor_assign.cpp234 Tensor<int, 1> tensor2(3); in test_auto_resize() local
241 tensor1 = tensor2 = tensor3 = tensor4 = new_tensor; in test_auto_resize()
244 VERIFY_IS_EQUAL(tensor2.dimension(0), new_tensor.dimension(0)); in test_auto_resize()
249 VERIFY_IS_EQUAL(tensor2(i), new_tensor(i)); in test_auto_resize()
/external/tensorflow/tensorflow/python/training/
Dmoving_averages_test.py183 tensor2 = var0 + var1
184 update = ema.apply([var0, var1, tensor2])
187 avg2 = ema.average(tensor2)
203 self.assertAllClose(_Repeat(10.0 + 30.0, dim), self.evaluate(tensor2))
340 tensor2 = v0 + v1
346 self.assertEqual("add/foo", ema.average_name(tensor2))
347 ema.apply([v0, v1, tensor2])
355 ema.average_name(v0), ema.average_name(v1), ema.average_name(tensor2),
363 ema.average_name(tensor2) + "/biased",
364 ema.average_name(tensor2) + "/local_step"
[all …]
Dbasic_session_run_hooks_test.py1136 tensor2 = tensor * 2
1138 self.summary_op2 = summary_lib.scalar('my_summary2', tensor2)
/external/swiftshader/third_party/llvm-7.0/llvm/test/DebugInfo/X86/
Dvla-multi.ll6 ; int tensor2[i][j][k][r];
7 ; use(tensor1, tensor2);
109 !13 = !DILocalVariable(name: "tensor2", scope: !8, file: !1, line: 3, type: !14)
/external/tensorflow/tensorflow/core/grappler/
Dutils.cc168 TensorId tensor2 = ParseTensorName(name2); in IsSameInput() local
169 return tensor1 == tensor2; in IsSameInput()
/external/tensorflow/tensorflow/contrib/gan/python/
Dtrain_test.py747 self, tensor1, tensor2, pool_size): argument
752 t1, t2 = sess.run([tensor1, tensor2])
/external/tensorflow/tensorflow/core/kernels/
Dlookup_table_op.cc755 bool IsEqualKey(typename TTypes<K>::Matrix tensor1, int64 index1, MT2 tensor2, in IsEqualKey() argument
758 if (tensor1(index1, i) != tensor2(index2, i)) { in IsEqualKey()
/external/tensorflow/tensorflow/c/
Dc_api_test.cc2321 const char tensor2[] = {2, 4, 6, 8}; in TEST_F() local
2328 Int8Tensor(dims2, ndims2, tensor2), in TEST_F()
2344 const char* tensor_data[] = {&tensor1[0], &tensor2[0]}; in TEST_F()
2345 const size_t tensor_size[] = {TF_ARRAYSIZE(tensor1), TF_ARRAYSIZE(tensor2)}; in TEST_F()