Searched refs:ModelHints (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | convolution_selector.h | 33 const GpuInfo& gpu_info, const OperationDef& op_def, ModelHints hints);
37 const GpuInfo& gpu_info, const OperationDef& op_def, ModelHints hints);
42 ModelHints hints, WeightsDescription* weights_desc);
46 ModelHints hints);
|
| D | operation_selector.h | 32 const OperationDef& op_def, ModelHints hints,
|
| D | default_selector.h | 32 ModelHints hints, const std::vector<Value*>& inputs,
|
| D | operation_selector.cc | 82 const OperationDef& op_def, ModelHints hints, in WinogradFromNode()
153 const OperationDef& op_def, ModelHints hints, in GPUOperationFromNode()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/ |
| D | convolution_selector.cc | 39 ModelHints hints) { in SelectConvolutionAdreno()
52 ModelHints hints) { in SelectConvolutionWinogradAdreno()
61 const OperationDef& op_def, ModelHints hints, in SelectConvolutionDynamicWeightsAdreno()
119 const OperationDef& op_def, ModelHints hints, in SelectConvolutionDynamicWeightsMali()
140 ModelHints hints) { in SelectConvolution()
162 ModelHints hints) { in SelectConvolutionForWinograd()
186 const OperationDef& op_def, ModelHints hints, in SelectConvolutionWithDynamicWeights()
213 ModelHints hints) { in SelectConverterToConvWeights()
|
| D | default_selector.cc | 30 ModelHints hints, const std::vector<Value*>& inputs, in SelectDefault()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_hints.h | 24 struct ModelHints { struct
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | inference_context.h | 62 ModelHints hints;
129 ModelHints hints);
|
| D | inference_context.cc | 200 ModelHints hints) { in Compile()
232 if (hints.Check(ModelHints::kAllowSpecialKernels) && in Compile()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | inference_context.h | 66 ModelHints hints;
115 const GraphFloat32& graph, ModelHints hints);
|
| D | api.cc | 661 create_info.hints.Add(ModelHints::kReduceKernelsCount); in Initialize()
662 create_info.hints.Add(ModelHints::kFastTuning); in Initialize()
664 create_info.hints.Add(ModelHints::kAllowSpecialKernels); in Initialize()
924 create_info.hints.Add(ModelHints::kReduceKernelsCount); in BuildSerializedModel()
925 create_info.hints.Add(ModelHints::kFastTuning); in BuildSerializedModel()
927 create_info.hints.Add(ModelHints::kAllowSpecialKernels); in BuildSerializedModel()
|
| D | inference_context.cc | 170 if (create_info.hints.Check(ModelHints::kFastTuning)) { in InitFromGraph()
293 ModelHints hints) { in ConvertOperations()
324 if (hints.Check(ModelHints::kAllowSpecialKernels) && in ConvertOperations()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/testing/ |
| D | performance_profiling.cc | 47 create_info.hints.Add(ModelHints::kAllowSpecialKernels); in RunModelSample()
|