Searched refs:src2_tensor (Results 1 – 1 of 1) sorted by relevance
211 TensorFloat32 src2_tensor; in TestLinkingConvolution2InputMul2InputMul() local212 src2_tensor.shape = input2->tensor.shape; in TestLinkingConvolution2InputMul2InputMul()213 src2_tensor.data.resize(src2_tensor.shape.DimensionsProduct()); in TestLinkingConvolution2InputMul2InputMul()214 for (int i = 0; i < src2_tensor.data.size(); ++i) { in TestLinkingConvolution2InputMul2InputMul()215 src2_tensor.data[i] = std::sin(i * 0.12345f); in TestLinkingConvolution2InputMul2InputMul()220 env->ExecuteGpuModel({src0_tensor, src1_tensor, src2_tensor}, in TestLinkingConvolution2InputMul2InputMul()250 op_def_mul, OperationType::MUL, src2_tensor.shape); in TestLinkingConvolution2InputMul2InputMul()253 {intermediate1, src2_tensor}, in TestLinkingConvolution2InputMul2InputMul()350 TensorFloat32 src2_tensor; in TestLinkingConvolution2InputBroadcastMul2InputMul() local351 src2_tensor.shape = input2->tensor.shape; in TestLinkingConvolution2InputBroadcastMul2InputMul()[all …]