Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dinference_context.cc67 const OperationDef op_def = node.cl_operation.GetDefinition(); in GetCLNodeTensors()
84 return dst->cl_operation.AddOperation(&src->cl_operation); in MergeCLNodes()
186 node.cl_operation.MoveObjectRefsFromCLToGeneric(); in InitFromGraph()
187 node.cl_operation.SyncScalarValues(); in InitFromGraph()
196 node.cl_operation.MoveObjectRefsFromGenericToCL(); in InitFromGraph()
222 RETURN_IF_ERROR(node.cl_operation.CompileDeserialized(creation_context)); in RestoreDeserialized()
376 cl_node.cl_operation.Init(std::move(gpu_op.operation)); in ConvertOperations()
431 if (!linkable_node.cl_operation.GetGpuOperation().IsLinkable() || in Merge()
437 node.cl_operation.GetDefinition().dst_tensors[0]; in Merge()
439 linkable_node.cl_operation.GetDefinition().dst_tensors[0]; in Merge()
[all …]
DBUILD240 name = "cl_operation",
241 srcs = ["cl_operation.cc"],
242 hdrs = ["cl_operation.h"],
393 ":cl_operation",
Dinference_context.h46 ClOperation cl_operation; member
Dserialization.cc918 auto op_fb = Encode(node.cl_operation.GetGpuOperation(), builder); in Encode()
941 node->cl_operation.Init(absl::make_unique<GPUOperation>(std::move(op))); in Decode()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
DBUILD34 "//tensorflow/lite/delegates/gpu/cl:cl_operation",