Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dgpu_model_test_util.cc205 TensorFloat32 src1_tensor; in TestLinkingConvolution2InputMul2InputMul() local
206 src1_tensor.shape = input1->tensor.shape; in TestLinkingConvolution2InputMul2InputMul()
207 src1_tensor.data.resize(src1_tensor.shape.DimensionsProduct()); in TestLinkingConvolution2InputMul2InputMul()
208 for (int i = 0; i < src1_tensor.data.size(); ++i) { in TestLinkingConvolution2InputMul2InputMul()
209 src1_tensor.data[i] = std::sin(i * 0.12345f); in TestLinkingConvolution2InputMul2InputMul()
220 env->ExecuteGpuModel({src0_tensor, src1_tensor, src2_tensor}, in TestLinkingConvolution2InputMul2InputMul()
242 op_def_mul, OperationType::MUL, src1_tensor.shape); in TestLinkingConvolution2InputMul2InputMul()
245 {intermediate0, src1_tensor}, in TestLinkingConvolution2InputMul2InputMul()
344 TensorFloat32 src1_tensor; in TestLinkingConvolution2InputBroadcastMul2InputMul() local
345 src1_tensor.shape = input1->tensor.shape; in TestLinkingConvolution2InputBroadcastMul2InputMul()
[all …]