Home
last modified time | relevance | path

Searched refs:tensor1 (Results 1 – 21 of 21) sorted by relevance

/external/eigen/unsupported/test/
Dcxx11_tensor_morphing.cpp19 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 …]
Dcxx11_tensor_custom_op.cpp86 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()
Dcxx11_tensor_io.cpp73 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()
Dcxx11_tensor_assign.cpp233 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/
Dconcatenation_test.cc213 auto tensor1 = {7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f}; in TEST() local
218 m0.SetInput(1, tensor1); in TEST()
226 m0_negative.SetInput(1, tensor1); in TEST()
234 m1.SetInput(1, tensor1); in TEST()
242 m1_negative.SetInput(1, tensor1); in TEST()
407 auto tensor1 = {7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f}; in TEST() local
414 m0.SetInput<uint8_t>(1, tensor1); in TEST()
424 m0_negative.SetInput<uint8_t>(1, tensor1); in TEST()
434 m1.SetInput<uint8_t>(1, tensor1); in TEST()
444 m1_negative.SetInput<uint8_t>(1, tensor1); in TEST()
/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()
117 for (int d = 0; d < TF_NumDims(tensor1); d++) { in IsSameSize()
118 if (TF_Dim(tensor1, d) != TF_Dim(tensor2, d)) { in IsSameSize()
/external/tensorflow/tensorflow/python/eager/
Dcontext_test.py56 tensor1 = constant_op.constant([[3.]])
59 tensor2 = tensor1 * tensor1
71 del tensor1
/external/tensorflow/tensorflow/lite/swift/Tests/
DTensorTests.swift47 let tensor1 = Tensor( in testEquatable() variable
61 XCTAssertEqual(tensor1, tensor2) in testEquatable()
70 XCTAssertNotEqual(tensor1, tensor2) in testEquatable()
/external/libaom/libaom/av1/encoder/
Dcnn.c836 TENSOR tensor1[CNN_MAX_BRANCHES] = { 0 }; in av1_cnn_predict_c() local
850 init_tensor(&tensor1[b]); in av1_cnn_predict_c()
863 assign_tensor(&tensor1[branch], (float **)input, in av1_cnn_predict_c()
867 swap_tensor(&tensor1[branch], &tensor2[branch]); in av1_cnn_predict_c()
869 i_width = tensor1[branch].width; in av1_cnn_predict_c()
870 i_height = tensor1[branch].height; in av1_cnn_predict_c()
893 copy_active_tensor_to_branches(&tensor1[branch], layer_config, branch, in av1_cnn_predict_c()
897 assert(tensor1[branch].channels == layer_config->in_channels); in av1_cnn_predict_c()
903 convolve_layer_mt((const float **)tensor1[branch].buf, in av1_cnn_predict_c()
904 tensor1[branch].width, tensor1[branch].height, in av1_cnn_predict_c()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dthreadpool_device_test.cc36 bool Equal(const Tensor& tensor1, const Tensor& tensor2) { in Equal() argument
37 auto eigen_tensor1 = tensor1.tensor<float, kDimSize>(); in Equal()
/external/tensorflow/tensorflow/lite/micro/
Dmicro_allocator_test.cc669 TfLiteTensor* tensor1 = allocator->AllocatePersistentTfLiteTensor( in TF_LITE_MICRO_TEST() local
671 TF_LITE_MICRO_EXPECT_NE(tensor1, nullptr); in TF_LITE_MICRO_TEST()
672 TF_LITE_MICRO_EXPECT_NE(tensor1->quantization.params, nullptr); in TF_LITE_MICRO_TEST()
673 TF_LITE_MICRO_EXPECT_FALSE(tensor1->is_variable); in TF_LITE_MICRO_TEST()
683 TF_LITE_MICRO_EXPECT_GT(tensor1, tensor2); in TF_LITE_MICRO_TEST()
694 TfLiteTensor* tensor1 = allocator->AllocateTempTfLiteTensor( in TF_LITE_MICRO_TEST() local
696 TF_LITE_MICRO_EXPECT_NE(tensor1, nullptr); in TF_LITE_MICRO_TEST()
707 TfLiteTensor* tensor1 = allocator->AllocateTempTfLiteTensor( in TF_LITE_MICRO_TEST() local
709 TF_LITE_MICRO_EXPECT_NE(tensor1, nullptr); in TF_LITE_MICRO_TEST()
717 TF_LITE_MICRO_EXPECT_GT(tensor2, tensor1); in TF_LITE_MICRO_TEST()
[all …]
/external/llvm-project/llvm/test/DebugInfo/X86/
Dvla-multi.ll5 ; 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/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
134 m0.SetInput<integer_type>(1, tensor1); in TwoInputsNegativeAxisImpl()
/external/tensorflow/tensorflow/python/kernel_tests/
Ddeterminant_op_test.py147 tensor1 = constant_op.constant([1., 2.])
149 linalg_ops.matrix_determinant(tensor1)
/external/tensorflow/tensorflow/lite/
Dinterpreter_test.cc622 TfLiteTensor* tensor1; in GetPassthroughOpRegistration() local
623 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &tensor1)); in GetPassthroughOpRegistration()
626 TF_LITE_ENSURE_STATUS(context->ResizeTensor(context, tensor1, newSize)); in GetPassthroughOpRegistration()
857 TfLiteTensor* tensor1; in TEST() local
858 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &tensor1)); in TEST()
860 return context->ResizeTensor(context, tensor1, newSize); in TEST()
1207 TfLiteTensor* tensor1; in CopyOpRegistration() local
1208 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &tensor1)); in CopyOpRegistration()
1210 return context->ResizeTensor(context, tensor1, newSize); in CopyOpRegistration()
/external/tensorflow/tensorflow/core/grappler/
Dutils.cc102 TensorId tensor1 = ParseTensorName(name1); in IsSameInput() local
104 return tensor1 == tensor2; in IsSameInput()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder_test.cc874 auto* tensor1 = interpreter_.tensor(1); in InterpreterMultiNode() local
877 tensor1->allocation_type = kTfLiteMmapRo; in InterpreterMultiNode()
/external/tensorflow/tensorflow/core/kernels/
Dlookup_table_op.cc777 bool IsEqualKey(typename TTypes<K>::Matrix tensor1, int64 index1, MT2 tensor2, in IsEqualKey() argument
780 if (tensor1(index1, i) != tensor2(index2, i)) { in IsEqualKey()
/external/tensorflow/tensorflow/c/
Dc_api_test.cc2364 const char tensor1[] = {5, 7}; in TEST_F() local
2374 Int8Tensor(dims1, ndims1, tensor1), in TEST_F()
2391 const char* tensor_data[] = {&tensor1[0], &tensor2[0]}; in TEST_F()
2392 const size_t tensor_size[] = {TF_ARRAYSIZE(tensor1), TF_ARRAYSIZE(tensor2)}; in TEST_F()
/external/tensorflow/tensorflow/compiler/xrt/tests/
Draw_api_test.cc485 Tensor tensor1; in TEST() local
486 TF_ASSERT_OK(LiteralToHostTensor(literal1, DT_FLOAT, &tensor1)); in TEST()
492 auto handle = ops::XRTAllocateFromTensor(root, {tensor0, tensor1}, in TEST()
493 {tensor0.shape(), tensor1.shape()}, in TEST()
/external/tflite-support/tensorflow_lite_support/cc/task/
DREADME.md66 Output tensor1: optional