Home
last modified time | relevance | path

Searched refs:gpu_op (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Doperation_selector.cc158 std::unique_ptr<GPUOperation>* gpu_op = in GPUOperationFromNode() local
168 *gpu_op = absl::make_unique<GPUOperation>(std::move(operation)); in GPUOperationFromNode()
176 SelectAdd(op_def, channels, output->tensor.shape.c, gpu_op); in GPUOperationFromNode()
183 *gpu_op = absl::make_unique<GPUOperation>(std::move(operation)); in GPUOperationFromNode()
269 return SelectConcat(attr, channels, op_def, gpu_info, gpu_op); in GPUOperationFromNode()
282 gpu_op = InitSingleOpSubgraph(inputs, outputs, gpu_subgraph); in GPUOperationFromNode()
283 *gpu_op = in GPUOperationFromNode()
333 *gpu_op = SelectConvolutionTransposed(attr, gpu_info, op_def); in GPUOperationFromNode()
401 *gpu_op = SelectDWConvolution(attr, gpu_info, op_def); in GPUOperationFromNode()
408 *gpu_op = SelectDWConvolutionDynamicWeights(attr, gpu_info, op_def); in GPUOperationFromNode()
[all …]
Dspecial_selector.cc85 std::unique_ptr<GPUOperation>* gpu_op = in TryDepthwiseConvPlus1x1Conv() local
88 *gpu_op = absl::make_unique<GPUOperation>(std::move(operation)); in TryDepthwiseConvPlus1x1Conv()
174 std::unique_ptr<GPUOperation>* gpu_op = in TryFCFCAdd() local
177 *gpu_op = absl::make_unique<FCFCAdd>(std::move(fc)); in TryFCFCAdd()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dserialization_generated.h102 const tflite::gpu::data::GPUOperation *gpu_op() const { in gpu_op() function
117 verifier.VerifyTable(gpu_op()) && in Verify()
132 void add_gpu_op(flatbuffers::Offset<tflite::gpu::data::GPUOperation> gpu_op) { in add_gpu_op()
133 fbb_.AddOffset(CLNode::VT_GPU_OP, gpu_op); in add_gpu_op()
157 flatbuffers::Offset<tflite::gpu::data::GPUOperation> gpu_op = 0,
165 builder_.add_gpu_op(gpu_op);
171 flatbuffers::Offset<tflite::gpu::data::GPUOperation> gpu_op = 0,
180 gpu_op,
Dinference_context.cc374 for (auto& gpu_op : gpu_subgraph.operations) { in ConvertOperations() local
376 cl_node.cl_operation.Init(std::move(gpu_op.operation)); in ConvertOperations()
377 cl_node.inputs.resize(gpu_op.input_ids.size()); in ConvertOperations()
378 for (int j = 0; j < gpu_op.input_ids.size(); ++j) { in ConvertOperations()
379 int id = gpu_op.input_ids[j]; in ConvertOperations()
386 cl_node.outputs.resize(gpu_op.output_ids.size()); in ConvertOperations()
387 for (int j = 0; j < gpu_op.output_ids.size(); ++j) { in ConvertOperations()
388 int id = gpu_op.output_ids[j]; in ConvertOperations()
Dserialization.fbs25 gpu_op:tflite.gpu.data.GPUOperation;
Dserialization.cc940 RETURN_IF_ERROR(Decode(fb_node->gpu_op(), &op)); in Decode()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dinference_context.cc282 for (auto& gpu_op : gpu_subgraph.operations) { in Compile() local
284 metal_node.task.Init(std::move(gpu_op.operation)); in Compile()
285 metal_node.inputs.resize(gpu_op.input_ids.size()); in Compile()
286 for (int j = 0; j < gpu_op.input_ids.size(); ++j) { in Compile()
287 int id = gpu_op.input_ids[j]; in Compile()
294 metal_node.outputs.resize(gpu_op.output_ids.size()); in Compile()
295 for (int j = 0; j < gpu_op.output_ids.size(); ++j) { in Compile()
296 int id = gpu_op.output_ids[j]; in Compile()