Searched refs:GraphToGpuModel (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| H A D | gpu_model_test_util.cc | 81 GraphToGpuModel(graph, create_info, env->GetGpuInfo(), &gpu_model)); in TestLinkingConvolutionAndCosOp()
193 GraphToGpuModel(graph, create_info, env->GetGpuInfo(), &gpu_model)); in TestLinkingConvolution2InputMul2InputMul()
332 GraphToGpuModel(graph, create_info, env->GetGpuInfo(), &gpu_model)); in TestLinkingConvolution2InputBroadcastMul2InputMul()
471 GraphToGpuModel(graph, create_info, env->GetGpuInfo(), &gpu_model)); in TestLinkingConvolution2InputMul2InputBroadcastMul()
617 GraphToGpuModel(graph, create_info, env->GetGpuInfo(), &gpu_model)); in TestLinkingConvolution2InputMul2InputMulCos()
757 GraphToGpuModel(graph, create_info, env->GetGpuInfo(), &gpu_model)); in TestLinkingConvolutionFirstTanh2InputDiff()
875 GraphToGpuModel(graph, create_info, env->GetGpuInfo(), &gpu_model)); in TestLinkingConvolutionSecondTanh2InputDiff()
1010 GraphToGpuModel(graph, create_info, env->GetGpuInfo(), &gpu_model)); in TestLinkingConvolutionFirstTanhSecondCos2InputDiff()
1201 GraphToGpuModel(graph, create_info, env->GetGpuInfo(), &gpu_model)); in TestLinkingComplex0()
|
| H A D | gpu_model.h | 107 absl::Status GraphToGpuModel(const GraphFloat32& graph,
|
| H A D | gpu_model.cc | 771 absl::Status GraphToGpuModel(const GraphFloat32& graph, in GraphToGpuModel() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/testing/ |
| H A D | memory_sharing_sample.cc | 130 GraphToGpuModel(graph_mv1, create_info_mv1, gpu_info, &gpu_model_mv1)); in RunSample()
138 GraphToGpuModel(graph_mv2, create_info_mv2, gpu_info, &gpu_model_mv2)); in RunSample()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| H A D | inference_context.cc | 147 GraphToGpuModel(graph, create_info, metal_device.GetInfo(), &gpu_model)); in InitFromGraph()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| H A D | inference_context.cc | 240 RETURN_IF_ERROR(GraphToGpuModel(graph, create_info, in InitFromGraph()
|