Home
last modified time | relevance | path

Searched refs:tensor2 (Results 1 – 18 of 18) 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.py185 tensor2 = var0 + var1
186 update = ema.apply([var0, var1, tensor2])
189 avg2 = ema.average(tensor2)
205 self.assertAllClose(_Repeat(10.0 + 30.0, dim), self.evaluate(tensor2))
330 tensor2 = v0 + v1
336 self.assertEqual("add/foo", ema.average_name(tensor2))
337 ema.apply([v0, v1, tensor2])
347 ema.average_name(tensor2), v2.op.name
354 ema.average_name(tensor2) + "/biased",
355 ema.average_name(tensor2) + "/local_step"
[all …]
Dbasic_session_run_hooks_test.py1176 tensor2 = tensor * 2
1178 self.summary_op2 = summary_lib.scalar('my_summary2', tensor2)
/external/libaom/libaom/av1/encoder/
Dcnn.c837 TENSOR tensor2[CNN_MAX_BRANCHES] = { 0 }; in av1_cnn_predict_c() local
851 init_tensor(&tensor2[b]); in av1_cnn_predict_c()
867 swap_tensor(&tensor1[branch], &tensor2[branch]); in av1_cnn_predict_c()
878 realloc_tensor(&tensor2[branch], layer_config->out_channels, o_width, in av1_cnn_predict_c()
881 free_tensor(&tensor2[branch]); in av1_cnn_predict_c()
882 assign_tensor(&tensor2[branch], output[output_num], in av1_cnn_predict_c()
894 tensor2); in av1_cnn_predict_c()
898 assert(tensor2[branch].channels == layer_config->out_channels); in av1_cnn_predict_c()
906 tensor2[branch].buf, tensor2[branch].stride); in av1_cnn_predict_c()
911 tensor2[branch].buf, tensor2[branch].stride, 0, 1); in av1_cnn_predict_c()
[all …]
/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/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/tensorflow/tensorflow/core/common_runtime/
Dthreadpool_device_test.cc36 bool Equal(const Tensor& tensor1, const Tensor& tensor2) { in Equal() argument
38 auto eigen_tensor2 = tensor2.tensor<float, kDimSize>(); in Equal()
/external/tensorflow/tensorflow/lite/micro/
Dmicro_allocator_test.cc675 TfLiteTensor* tensor2 = allocator->AllocatePersistentTfLiteTensor( in TF_LITE_MICRO_TEST() local
677 TF_LITE_MICRO_EXPECT_NE(tensor2, nullptr); in TF_LITE_MICRO_TEST()
678 TF_LITE_MICRO_EXPECT_NE(tensor2->quantization.params, nullptr); in TF_LITE_MICRO_TEST()
679 TF_LITE_MICRO_EXPECT_FALSE(tensor2->is_variable); in TF_LITE_MICRO_TEST()
683 TF_LITE_MICRO_EXPECT_GT(tensor1, tensor2); in TF_LITE_MICRO_TEST()
711 TfLiteTensor* tensor2 = allocator->AllocateTempTfLiteTensor( in TF_LITE_MICRO_TEST() local
713 TF_LITE_MICRO_EXPECT_NE(tensor2, nullptr); in TF_LITE_MICRO_TEST()
717 TF_LITE_MICRO_EXPECT_GT(tensor2, tensor1); in TF_LITE_MICRO_TEST()
734 TfLiteTensor* tensor2 = allocator->AllocateTempTfLiteTensor( in TF_LITE_MICRO_TEST() local
736 TF_LITE_MICRO_EXPECT(tensor2 != nullptr); in TF_LITE_MICRO_TEST()
[all …]
/external/tensorflow/tensorflow/python/eager/
Dcontext_test.py59 tensor2 = tensor1 * tensor1
75 del tensor2
/external/llvm-project/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/lite/delegates/gpu/common/
Dmodel_builder_test.cc253 auto* tensor2 = interpreter_.tensor(2); in InterpreterFp16() local
255 tensor2->allocation_type = kTfLiteMmapRo; in InterpreterFp16()
875 auto* tensor2 = interpreter_.tensor(2); in InterpreterMultiNode() local
878 tensor2->allocation_type = kTfLiteMmapRo; in InterpreterMultiNode()
/external/tensorflow/tensorflow/core/grappler/
Dutils.cc103 TensorId tensor2 = ParseTensorName(name2); in IsSameInput() local
104 return tensor1 == tensor2; in IsSameInput()
/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.cc2368 const char tensor2[] = {2, 4, 6, 8}; in TEST_F() local
2375 Int8Tensor(dims2, ndims2, tensor2), 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()