Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dcl_test.cc51 CreationContext creation_context; in ExecuteGPUOperation() local
52 creation_context.device = env_.GetDevicePtr(); in ExecuteGPUOperation()
53 creation_context.context = &env_.context(); in ExecuteGPUOperation()
54 creation_context.queue = env_.queue(); in ExecuteGPUOperation()
55 creation_context.cache = env_.program_cache(); in ExecuteGPUOperation()
65 RETURN_IF_ERROR(CreateTensor(*creation_context.context, src_shape, in ExecuteGPUOperation()
67 RETURN_IF_ERROR(src[i].WriteData(creation_context.queue, src_cpu[i])); in ExecuteGPUOperation()
78 RETURN_IF_ERROR(CreateTensor(*creation_context.context, dst_shape, in ExecuteGPUOperation()
86 RETURN_IF_ERROR(cl_op.Compile(creation_context)); in ExecuteGPUOperation()
89 RETURN_IF_ERROR(cl_op.AddToQueue(creation_context.queue)); in ExecuteGPUOperation()
[all …]
Dcl_test.h84 const CreationContext& creation_context,
89 const CreationContext& creation_context,
94 const CreationContext& creation_context,
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dcl_operation.cc183 absl::Status ClOperation::Compile(const CreationContext& creation_context) { in Compile() argument
184 operation_->AssembleCode(creation_context.GetGpuInfo()); in Compile()
189 creation_context.GetGpuInfo(), in Compile()
191 creation_context.context, &operation_->args_, &operation_->code_)); in Compile()
192 RETURN_IF_ERROR(creation_context.cache->GetOrCreateCLKernel( in Compile()
194 *creation_context.context, *creation_context.device, &kernel_)); in Compile()
195 return operation_->PostCompileCheck(creation_context.GetGpuInfo(), in Compile()
200 const CreationContext& creation_context) { in CompileDeserialized() argument
201 RETURN_IF_ERROR(cl_args_.Init(creation_context.GetGpuInfo(), in CompileDeserialized()
202 &operation_->args_, creation_context.context)); in CompileDeserialized()
[all …]
Dinference_context.cc141 CreationContext creation_context; in InitFromGraph() local
142 creation_context.device = env->GetDevicePtr(); in InitFromGraph()
143 creation_context.context = &env->context(); in InitFromGraph()
144 creation_context.queue = env->queue(); in InitFromGraph()
145 creation_context.cache = env->program_cache(); in InitFromGraph()
147 ReserveGraphTensors(create_info, creation_context.GetGpuInfo(), graph); in InitFromGraph()
161 RETURN_IF_ERROR(ConvertOperations(creation_context.GetGpuInfo(), graph, in InitFromGraph()
164 RETURN_IF_ERROR(AllocateMemory(creation_context.context)); in InitFromGraph()
166 RETURN_IF_ERROR(Compile(creation_context)); in InitFromGraph()
213 CreationContext creation_context; in RestoreDeserialized() local
[all …]
Dcl_operation.h77 absl::Status Compile(const CreationContext& creation_context);
79 absl::Status CompileDeserialized(const CreationContext& creation_context);
Dinference_context.h137 absl::Status Compile(const CreationContext& creation_context);