Home
last modified time | relevance | path

Searched defs:gpu_model (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
H A Dinference_context.cc86 TensorType GetTensorType(const GpuModel& gpu_model, in GetTensorType()
140 const GpuModel& gpu_model, const CreateGpuModelInfo* create_info, in GetBufferAsignment()
239 GpuModel gpu_model; in InitFromGraph() local
246 const CreateGpuModelInfo& create_info, GpuModel* gpu_model, in InitFromGpuModel()
345 GpuModel gpu_model; in RestoreDeserialized() local
404 void InferenceContext::InitFromGpuModel(GpuModel* gpu_model) { in InitFromGpuModel()
437 const GpuModel& gpu_model, const GpuInfo& gpu_info, in AllocateMemory()
448 absl::Status InferenceContext::AllocateConstTensors(const GpuModel& gpu_model, in AllocateConstTensors()
458 const GpuModel& gpu_model, CLContext* context) { in AllocateVariableTensors()
481 const GpuModel& gpu_model, const GpuInfo& gpu_info, in AllocateBufferBasedTensors()
[all …]
H A Dserialization_generated.h109 const tflite::gpu::data::GpuModel *gpu_model() const { in gpu_model() function
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dgpu_model.cc266 GpuModel* gpu_model) { in ConvertOperations()
385 GpuModel* gpu_model) { in MergeElementwiseNodes()
637 absl::Status MergeNodes(const GpuInfo& gpu_info, GpuModel* gpu_model) { in MergeNodes()
677 void CopyExternals(const GraphFloat32& graph, GpuModel* gpu_model) { in CopyExternals()
695 void RemoveUnusedTensors(GpuModel* gpu_model) { in RemoveUnusedTensors()
724 absl::Status ResolvePolymorphicArgs(GpuModel* gpu_model) { in ResolvePolymorphicArgs()
773 const GpuInfo& gpu_info, GpuModel* gpu_model) { in GraphToGpuModel()
795 const GpuModel& gpu_model, flatbuffers::FlatBufferBuilder* builder) { in Encode()
858 absl::Status Decode(const data::GpuModel* fb_gpu_model, GpuModel* gpu_model) { in Decode()
H A Dgpu_model_test_util.cc79 GpuModel gpu_model; in TestLinkingConvolutionAndCosOp() local
191 GpuModel gpu_model; in TestLinkingConvolution2InputMul2InputMul() local
330 GpuModel gpu_model; in TestLinkingConvolution2InputBroadcastMul2InputMul() local
469 GpuModel gpu_model; in TestLinkingConvolution2InputMul2InputBroadcastMul() local
615 GpuModel gpu_model; in TestLinkingConvolution2InputMul2InputMulCos() local
755 GpuModel gpu_model; in TestLinkingConvolutionFirstTanh2InputDiff() local
873 GpuModel gpu_model; in TestLinkingConvolutionSecondTanh2InputDiff() local
1008 GpuModel gpu_model; in TestLinkingConvolutionFirstTanhSecondCos2InputDiff() local
1199 GpuModel gpu_model; in TestLinkingComplex0() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
H A Dinference_context.cc122 void InferenceContext::CopyFromGpuModel(GpuModel* gpu_model) { in CopyFromGpuModel()
145 GpuModel gpu_model; in InitFromGraph() local
291 GpuModel gpu_model; in Decode() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
H A Dtesting_util.cc131 const std::vector<TensorFloat32*>& dst_cpu, GpuModel* gpu_model) { in ExecuteGpuModel()
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/convert/
H A Dxplane_to_op_stats.cc107 absl::string_view gpu_model = in SetRunEnvironment() local