Home
last modified time | relevance | path

Searched refs:src0_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.cc199 TensorFloat32 src0_tensor; in TestLinkingConvolution2InputMul2InputMul() local
200 src0_tensor.shape = input0->tensor.shape; in TestLinkingConvolution2InputMul2InputMul()
201 src0_tensor.data.resize(src0_tensor.shape.DimensionsProduct()); in TestLinkingConvolution2InputMul2InputMul()
202 for (int i = 0; i < src0_tensor.data.size(); ++i) { in TestLinkingConvolution2InputMul2InputMul()
203 src0_tensor.data[i] = std::sin(i * 0.12345f); in TestLinkingConvolution2InputMul2InputMul()
220 env->ExecuteGpuModel({src0_tensor, src1_tensor, src2_tensor}, in TestLinkingConvolution2InputMul2InputMul()
232 src0_tensor, std::make_unique<ConvGeneric>(std::move(conv_operation)), in TestLinkingConvolution2InputMul2InputMul()
338 TensorFloat32 src0_tensor; in TestLinkingConvolution2InputBroadcastMul2InputMul() local
339 src0_tensor.shape = input0->tensor.shape; in TestLinkingConvolution2InputBroadcastMul2InputMul()
340 src0_tensor.data.resize(src0_tensor.shape.DimensionsProduct()); in TestLinkingConvolution2InputBroadcastMul2InputMul()
[all …]