Home
last modified time | relevance | path

Searched refs:tensor_descriptors (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Dspecial_selector.cc34 const std::map<ValueId, TensorDescriptor>& tensor_descriptors, in TryDepthwiseConvPlus1x1Conv() argument
74 auto it = tensor_descriptors.find(dw_inputs[0]->id); in TryDepthwiseConvPlus1x1Conv()
75 if (it != tensor_descriptors.end()) { in TryDepthwiseConvPlus1x1Conv()
78 it = tensor_descriptors.find(conv_outputs[0]->id); in TryDepthwiseConvPlus1x1Conv()
79 if (it != tensor_descriptors.end()) { in TryDepthwiseConvPlus1x1Conv()
98 const std::map<ValueId, TensorDescriptor>& tensor_descriptors, in TryFCFCAdd() argument
158 auto it = tensor_descriptors.find(fc0_inputs[0]->id); in TryFCFCAdd()
159 if (it != tensor_descriptors.end()) { in TryFCFCAdd()
162 it = tensor_descriptors.find(fc1_inputs[0]->id); in TryFCFCAdd()
163 if (it != tensor_descriptors.end()) { in TryFCFCAdd()
[all …]
Dspecial_selector.h36 const std::map<ValueId, TensorDescriptor>& tensor_descriptors,
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dinference_context.cc205 std::map<ValueId, TensorDescriptor> tensor_descriptors; in Compile() local
208 tensor_descriptors[value->id] = tensor_reserver_.Get(value->id).descriptor; in Compile()
234 tensor_descriptors, &consumed_nodes, &gpu_subgraph, in Compile()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dinference_context.cc294 std::map<ValueId, TensorDescriptor> tensor_descriptors; in ConvertOperations() local
297 tensor_descriptors[value->id] = tensor_reserver_.Get(value->id).descriptor; in ConvertOperations()
326 tensor_descriptors, &consumed_nodes, &gpu_subgraph, in ConvertOperations()