Searched refs:gpu_subgraph (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | subgraph.cc | 29 GPUOperationsSubgraph* gpu_subgraph) { in InitSingleOpSubgraph() argument
30 gpu_subgraph->operations.clear(); in InitSingleOpSubgraph()
31 gpu_subgraph->new_tensors.clear(); in InitSingleOpSubgraph()
32 gpu_subgraph->operations.push_back({}); in InitSingleOpSubgraph()
34 gpu_subgraph->operations[0].input_ids.push_back(inputs[i]->id); in InitSingleOpSubgraph()
37 gpu_subgraph->operations[0].output_ids.push_back(outputs[i]->id); in InitSingleOpSubgraph()
40 return &gpu_subgraph->operations[0].operation; in InitSingleOpSubgraph()
|
| D | operation_selector.cc | 85 GPUOperationsSubgraph* gpu_subgraph) { in WinogradFromNode() argument
109 gpu_subgraph->new_tensors = {{shape_0, td_0}, {shape_1, td_1}}; in WinogradFromNode()
110 gpu_subgraph->operations.clear(); in WinogradFromNode()
111 gpu_subgraph->operations.resize(3); in WinogradFromNode()
117 auto& winograd_up = gpu_subgraph->operations[0]; in WinogradFromNode()
127 auto& conv = gpu_subgraph->operations[1]; in WinogradFromNode()
137 auto& winograd_down = gpu_subgraph->operations[2]; in WinogradFromNode()
157 GPUOperationsSubgraph* gpu_subgraph) { in GPUOperationFromNode() argument
159 InitSingleOpSubgraph(inputs, outputs, gpu_subgraph); in GPUOperationFromNode()
218 gpu_subgraph->operations.clear(); in GPUOperationFromNode()
[all …]
|
| D | special_selector.cc | 35 std::set<NodeId>* consumed_nodes, GPUOperationsSubgraph* gpu_subgraph) { in TryDepthwiseConvPlus1x1Conv() argument
86 InitSingleOpSubgraph(dw_inputs, conv_outputs, gpu_subgraph); in TryDepthwiseConvPlus1x1Conv()
99 std::set<NodeId>* consumed_nodes, GPUOperationsSubgraph* gpu_subgraph) { in TryFCFCAdd() argument
175 InitSingleOpSubgraph(fc0_inputs, add_outputs, gpu_subgraph); in TryFCFCAdd()
189 std::set<NodeId>* consumed_nodes, GPUOperationsSubgraph* gpu_subgraph, in GPUSubgraphFromGraph() argument
194 gpu_subgraph) in GPUSubgraphFromGraph()
201 consumed_nodes, gpu_subgraph) in GPUSubgraphFromGraph()
|
| D | operation_selector.h | 36 GPUOperationsSubgraph* gpu_subgraph);
|
| D | default_selector.h | 34 GPUOperationsSubgraph* gpu_subgraph);
|
| D | special_selector.h | 37 std::set<NodeId>* consumed_nodes, GPUOperationsSubgraph* gpu_subgraph,
|
| D | subgraph.h | 46 GPUOperationsSubgraph* gpu_subgraph);
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/ |
| D | default_selector.cc | 32 GPUOperationsSubgraph* gpu_subgraph) { in SelectDefault() argument
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | inference_context.cc | 231 GPUOperationsSubgraph gpu_subgraph; in Compile() local
234 tensor_descriptors, &consumed_nodes, &gpu_subgraph, in Compile()
274 outputs, node, &gpu_subgraph)); in Compile()
277 for (int j = 0; j < gpu_subgraph.new_tensors.size(); ++j) { in Compile()
278 const auto& t = gpu_subgraph.new_tensors[j]; in Compile()
282 for (auto& gpu_op : gpu_subgraph.operations) { in Compile()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | inference_context.cc | 323 GPUOperationsSubgraph gpu_subgraph; in ConvertOperations() local
326 tensor_descriptors, &consumed_nodes, &gpu_subgraph, in ConvertOperations()
366 outputs, node, &gpu_subgraph)); in ConvertOperations()
369 for (int j = 0; j < gpu_subgraph.new_tensors.size(); ++j) { in ConvertOperations()
370 const auto& t = gpu_subgraph.new_tensors[j]; in ConvertOperations()
374 for (auto& gpu_op : gpu_subgraph.operations) { in ConvertOperations()
|