| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_morphing.cpp | 19 Tensor<float, 5> tensor1(2,3,1,7,1); in test_simple_reshape() local
20 tensor1.setRandom(); in test_simple_reshape()
27 tensor2 = tensor1.reshape(dim1); in test_simple_reshape()
29 tensor3 = tensor1.reshape(dim2); in test_simple_reshape()
31 tensor4 = tensor1.reshape(dim1).reshape(dim3); in test_simple_reshape()
36 VERIFY_IS_EQUAL(tensor1(i,j,0,k,0), tensor2(i,j,k)); in test_simple_reshape()
37 VERIFY_IS_EQUAL(tensor1(i,j,0,k,0), tensor3(i+2*j,k)); in test_simple_reshape()
38 VERIFY_IS_EQUAL(tensor1(i,j,0,k,0), tensor4(i,j+3*k)); in test_simple_reshape()
52 TensorMap<Tensor<float, 5>> tensor1(m1.data(), 2,3,5,7,11); in test_reshape_in_expr() local
59 tensor3 = tensor1.reshape(newDims1).contract(tensor2.reshape(newDims2), contract_along); in test_reshape_in_expr()
[all …]
|
| D | cxx11_tensor_custom_op.cpp | 86 Tensor<float, 3> tensor1(2,3,5); in test_custom_binary_op() local
87 tensor1.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()
|
| D | cxx11_tensor_io.cpp | 73 Tensor<int, 1, DataLayout> tensor1(5); in test_output_expr() local
76 tensor1(i) = i; in test_output_expr()
81 os << tensor1 + tensor2; in test_output_expr()
|
| D | cxx11_tensor_assign.cpp | 233 Tensor<int, 1> tensor1; in test_auto_resize() local
241 tensor1 = tensor2 = tensor3 = tensor4 = new_tensor; in test_auto_resize()
243 VERIFY_IS_EQUAL(tensor1.dimension(0), new_tensor.dimension(0)); in test_auto_resize()
248 VERIFY_IS_EQUAL(tensor1(i), new_tensor(i)); in test_auto_resize()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | concatenation_test.cc | 123 auto tensor1 = {7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f}; in TEST() local
128 m0.SetInput(1, tensor1); in TEST()
136 m0_negative.SetInput(1, tensor1); in TEST()
144 m1.SetInput(1, tensor1); in TEST()
152 m1_negative.SetInput(1, tensor1); in TEST()
|
| /external/swiftshader/third_party/llvm-7.0/llvm/test/DebugInfo/X86/ |
| D | vla-multi.ll | 5 ; int tensor1[i][j][k][r];
7 ; use(tensor1, tensor2);
121 !25 = !DILocalVariable(name: "tensor1", scope: !8, file: !1, line: 2, type: !26)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | determinant_op_test.py | 147 tensor1 = constant_op.constant([1., 2.])
149 linalg_ops.matrix_determinant(tensor1)
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils.cc | 167 TensorId tensor1 = ParseTensorName(name1); in IsSameInput() local
169 return tensor1 == tensor2; in IsSameInput()
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter_test.cc | 545 TfLiteTensor* tensor1 = &context->tensors[node->outputs->data[0]]; in TEST() local
548 TF_LITE_ENSURE_STATUS(context->ResizeTensor(context, tensor1, newSize)); in TEST()
694 TfLiteTensor* tensor1 = &context->tensors[node->outputs->data[0]]; in TEST() local
696 return context->ResizeTensor(context, tensor1, newSize); in TEST()
941 TfLiteTensor* tensor1 = &context->tensors[node->outputs->data[0]]; in CopyOpRegistration() local
943 return context->ResizeTensor(context, tensor1, newSize); in CopyOpRegistration()
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 280 Tensor tensor1; in TEST() local
281 TF_ASSERT_OK(LiteralToHostTensor(literal1, DT_FLOAT, &tensor1)); in TEST()
287 auto handle = ops::XRTAllocateFromTensor(root, {tensor0, tensor1}, in TEST()
288 {tensor0.shape(), tensor1.shape()}, in TEST()
|
| /external/tensorflow/tensorflow/contrib/gan/python/ |
| D | train_test.py | 747 self, tensor1, tensor2, pool_size): argument
752 t1, t2 = sess.run([tensor1, tensor2])
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lookup_table_op.cc | 755 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/ |
| D | c_api_test.cc | 2317 const char tensor1[] = {5, 7}; in TEST_F() local
2327 Int8Tensor(dims1, ndims1, tensor1), 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()
|