Home
last modified time | relevance | path

Searched refs:GpuInfo (Results 1 – 25 of 138) sorted by relevance

123456

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dgpu_info.cc349 GpuApi gpu_api, GpuInfo* gpu_info) { in GetGpuInfoFromDeviceDescription()
385 bool GpuInfo::IsAdreno() const { return vendor == GpuVendor::kQualcomm; } in IsAdreno()
387 bool GpuInfo::IsApple() const { return vendor == GpuVendor::kApple; } in IsApple()
389 bool GpuInfo::IsMali() const { return vendor == GpuVendor::kMali; } in IsMali()
391 bool GpuInfo::IsPowerVR() const { return vendor == GpuVendor::kPowerVR; } in IsPowerVR()
393 bool GpuInfo::IsNvidia() const { return vendor == GpuVendor::kNvidia; } in IsNvidia()
395 bool GpuInfo::IsAMD() const { return vendor == GpuVendor::kAMD; } in IsAMD()
397 bool GpuInfo::IsIntel() const { return vendor == GpuVendor::kIntel; } in IsIntel()
399 bool GpuInfo::IsRoundToNearestSupported() const { in IsRoundToNearestSupported()
409 bool GpuInfo::SupportsFP16() const { in SupportsFP16()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_powervr.h44 TuningType tuning_type, const GpuInfo& gpu_info,
111 const Convolution2DAttributes& attr, const GpuInfo& gpu_info,
115 const GpuInfo& gpu_info, const BHWC* dst_shape = nullptr);
117 const FullyConnectedAttributes& attr, const GpuInfo& gpu_info,
121 const Convolution3DAttributes& attr, const GpuInfo& gpu_info,
124 void GenerateCode(const GpuInfo& gpu_info);
142 friend ConvPowerVR CreateConvPowerVR(const GpuInfo& gpu_info,
147 friend ConvPowerVR CreateConvPowerVR(const GpuInfo& gpu_info,
153 const GpuInfo& gpu_info, const OperationDef& definition,
158 const GpuInfo& gpu_info, const OperationDef& definition,
[all …]
Dwinograd.h35 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
65 const Padding2D& padding, const GpuInfo& gpu_info);
69 TuningType tuning_type, const GpuInfo& gpu_info,
81 const GpuInfo& gpu_info, const OperationDef& definition,
95 const GpuInfo& gpu_info, const OperationDef& definition,
102 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
131 const GpuInfo& gpu_info);
135 TuningType tuning_type, const GpuInfo& gpu_info,
148 const GpuInfo& gpu_info, const OperationDef& definition,
160 const GpuInfo& gpu_info, const OperationDef& definition,
Dconvolution_transposed_3x3.h41 TuningType tuning_type, const GpuInfo& gpu_info,
70 const GpuInfo& gpu_info, int2 padding);
72 const GpuInfo& gpu_info, const OperationDef& definition,
75 const GpuInfo& gpu_info, const OperationDef& definition,
84 const GpuInfo& gpu_info, const OperationDef& op_def,
98 const GpuInfo& gpu_info, const OperationDef& definition,
102 const GpuInfo& gpu_info, const OperationDef& definition,
Dconvolution_transposed_4x4.h41 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
72 const GpuInfo& gpu_info);
75 const GpuInfo& gpu_info, const OperationDef& definition,
78 const GpuInfo& gpu_info, const OperationDef& definition,
88 const GpuInfo& gpu_info, const OperationDef& op_def,
99 const GpuInfo& gpu_info, const OperationDef& definition,
103 const GpuInfo& gpu_info, const OperationDef& definition,
Dconvolution_transposed_3x3_thin.h59 ConvolutionTransposed3x3Thin(const GpuInfo& gpu_info,
64 const GpuInfo& gpu_info, const OperationDef& definition,
68 const GpuInfo& gpu_info, const OperationDef& definition,
86 const GpuInfo& gpu_info, const OperationDef& definition,
90 const GpuInfo& gpu_info, const OperationDef& definition,
Dconvolution_transposed.h43 TuningType tuning_type, const GpuInfo& gpu_info,
64 const GpuInfo& gpu_info, const OperationDef& definition,
67 const GpuInfo& gpu_info, const OperationDef& definition,
70 const GpuInfo& gpu_info, const OperationDef& definition,
75 const GpuInfo& gpu_info, bool weights_are_buffer);
78 const GpuInfo& gpu_info, bool weights_are_buffer);
89 const GpuInfo& gpu_info,
203 const GpuInfo& gpu_info, const OperationDef& definition,
207 const GpuInfo& gpu_info, const OperationDef& definition,
211 const GpuInfo& gpu_info, const OperationDef& definition,
Dmean_stddev_normalization.h31 const GpuInfo& gpu_info,
35 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
50 std::string GetNormalizationCode(const GpuInfo& gpu_info);
54 const OperationDef& definition, const GpuInfo& gpu_info,
Dconv_buffer_1x1.h46 TuningType tuning_type, const GpuInfo& gpu_info,
70 friend ConvBuffer1x1 CreateConvBuffer1x1(const GpuInfo& gpu_info,
74 friend ConvBuffer1x1 CreateConvBuffer1x1(const GpuInfo& gpu_info,
79 const GpuInfo& gpu_info, const OperationDef& definition,
82 const GpuInfo& gpu_info, const OperationDef& definition,
176 ConvBuffer1x1 CreateConvBuffer1x1(const GpuInfo& gpu_info,
181 ConvBuffer1x1 CreateConvBuffer1x1(const GpuInfo& gpu_info,
187 const GpuInfo& gpu_info, const OperationDef& definition,
192 const GpuInfo& gpu_info, const OperationDef& definition,
Dconv_metal.h58 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
84 const Convolution2DAttributes& attr, const GpuInfo& gpu_info);
88 const Convolution2DAttributes& attr, const GpuInfo& gpu_info);
98 const GpuInfo& gpu_info);
102 const Convolution2DAttributes& attr, const GpuInfo& gpu_info);
Dreduce.h34 const OperationDef& definition, const GpuInfo& gpu_info);
37 TuningType tuning_type, const GpuInfo& gpu_info,
60 const GpuInfo& gpu_info);
64 const OperationDef& definition, const GpuInfo& gpu_info);
Dmean_stddev_normalization.cc33 std::string GetReduceCode(const GpuInfo& gpu_info, int reduction_size) { in GetReduceCode()
91 const GpuInfo& gpu_info, in MeanStdDevNormalization()
152 const GpuInfo& gpu_info) { in GetNormalizationCode()
218 const OperationDef& definition, const GpuInfo& gpu_info, in CreateMeanStdDevNormalization()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Dsimple_selectors.h31 const GpuInfo& gpu_info);
37 const GpuInfo& gpu_info,
55 const OperationDef& op_def, const GpuInfo& gpu_info,
59 const DepthwiseConvolution2DAttributes& attr, const GpuInfo& gpu_info,
76 const GpuInfo& gpu_info);
93 std::unique_ptr<GPUOperation> SelectWinograd4x4To36(const GpuInfo& gpu_info,
98 const GpuInfo& gpu_info, const OperationDef& op_def,
Dconvolution_selector.h33 const GpuInfo& gpu_info, const OperationDef& op_def, ModelHints hints);
37 const GpuInfo& gpu_info, const OperationDef& op_def, ModelHints hints);
41 const BHWC& dst_shape, const GpuInfo& gpu_info, const OperationDef& op_def,
Dsimple_selectors.cc50 const GpuInfo& gpu_info) { in SelectLSTM()
60 const GpuInfo& gpu_info, in SelectPReLU()
91 const OperationDef& op_def, const GpuInfo& gpu_info, in SelectConcat()
113 const DepthwiseConvolution2DAttributes& attr, const GpuInfo& gpu_info, in SelectDWConvolutionDynamicWeights()
165 const GpuInfo& gpu_info) { in SelectReduce()
189 const GpuInfo& gpu_info, const Padding2D& padding, in SelectWinograd4x4To36()
207 const GpuInfo& gpu_info, const OperationDef& op_def, in SelectWinograd36To4x4()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/
Dconvolution_selector.cc38 const GpuInfo& gpu_info, const OperationDef& op_def, in SelectConvolutionAdreno()
51 const GpuInfo& gpu_info, const OperationDef& op_def, in SelectConvolutionWinogradAdreno()
60 const BHWC& dst_shape, const GpuInfo& gpu_info, in SelectConvolutionDynamicWeightsAdreno()
71 const GpuInfo& gpu_info, const OperationDef& op_def) { in SelectConvolutionNVidia()
82 const Convolution2DAttributes& attr, const GpuInfo& gpu_info, in SelectConvolutionPowerVR()
90 const GpuInfo& gpu_info, const OperationDef& op_def) { in SelectConvolutionMali()
104 const GpuInfo& gpu_info, const OperationDef& op_def) { in SelectConvolutionWinogradMali()
118 const BHWC& dst_shape, const GpuInfo& gpu_info, in SelectConvolutionDynamicWeightsMali()
139 const GpuInfo& gpu_info, const OperationDef& op_def, in SelectConvolution()
161 const GpuInfo& gpu_info, const OperationDef& op_def, in SelectConvolutionForWinograd()
[all …]
Ddw_convolution_selector.cc27 const DepthwiseConvolution2DAttributes& attr, const GpuInfo& gpu_info, in SelectDWConvolutionAdreno()
39 const DepthwiseConvolution2DAttributes& attr, const GpuInfo& gpu_info, in SelectDWConvolutionPowerVR()
51 const DepthwiseConvolution2DAttributes& attr, const GpuInfo& gpu_info, in SelectDWConvolutionMali()
68 const DepthwiseConvolution2DAttributes& attr, const GpuInfo& gpu_info, in SelectDWConvolutionApple()
84 const DepthwiseConvolution2DAttributes& attr, const GpuInfo& gpu_info, in SelectDWConvolution()
Dfully_connected_selector.cc28 const FullyConnectedAttributes& attr, const GpuInfo& gpu_info, in SelectFullyConnectedGeneric()
41 const FullyConnectedAttributes& attr, const GpuInfo& gpu_info, in SelectFullyConnectedAdreno()
54 const FullyConnectedAttributes& attr, const GpuInfo& gpu_info, in SelectFullyConnectedPowerVR()
66 const FullyConnectedAttributes& attr, const GpuInfo& gpu_info, in SelectFullyConnectedMali()
85 const FullyConnectedAttributes& attr, const GpuInfo& gpu_info, in SelectFullyConnected()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dwork_group_picking.h31 void GetPossibleWorkGroupsXYMultipleOf(int multiplier, const GpuInfo& gpu_info,
37 void GetPossibleWorkGroupsXMultipleOf(int multiplier, const GpuInfo& gpu_info,
50 void GetPossibleWorkGroups(TuningType tuning_type, const GpuInfo& gpu_info,
54 void GetPossibleWorkGroupsConv(TuningType tuning_type, const GpuInfo& gpu_info,
Dstorage_type_util.h28 bool CanCreateTensorWithShape(const GpuInfo& gpu_info, const BHWDC& shape,
31 bool CanCreateTensorWithShape(const GpuInfo& gpu_info, const BHWC& shape,
34 TensorStorageType SelectBestStorageType(const GpuInfo& gpu_info,
Dtensor_desc.h62 absl::Status PerformSelector(const GpuInfo& gpu_info,
108 const GpuInfo& gpu_info, const std::vector<std::string>& args,
128 absl::Status PerformWriteSelector(const GpuInfo& gpu_info,
132 absl::Status PerformWriteLinearSelector(const GpuInfo& gpu_info,
136 absl::Status PerformWrite2DSelector(const GpuInfo& gpu_info,
140 std::string Read(const GpuInfo& gpu_info, DataType read_as_type,
142 std::string Write(const GpuInfo& gpu_info, const std::string& var_name,
/external/tensorflow/tensorflow/python/framework/
Dgpu_util.py32 GpuInfo = collections.namedtuple('gpu_info', ['name', 'compute_capability']) variable
55 return GpuInfo(None, None)
57 return GpuInfo(match.group(1), cc)
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/workgroups/
Dcalculator_from_metadata.cc43 const GpuInfo& gpu_info) in WorkgroupsCalculatorFromMetadata()
70 const data::CustomWorkgroups& workgroups, const GpuInfo& gpu_info) { in FindWorkgroups()
82 const uint8_t* metadata, const GpuInfo& gpu_info) { in NewWorkgroupsCalculatorFromMetadata()
97 const uint8_t* metadata, const GpuInfo& gpu_info) {
Ddefault_calculator.cc29 explicit DefaultWorkgroupsCalculator(const GpuInfo& gpu_info) in DefaultWorkgroupsCalculator()
57 explicit WorkgroupsCalculatorForMali(const GpuInfo& gpu_info) in WorkgroupsCalculatorForMali()
83 const GpuInfo& gpu_info) { in NewDefaultWorkgroupsCalculator()
/external/tensorflow/tensorflow/lite/experimental/acceleration/compatibility/
Dgpu_compatibility.h64 const ::tflite::gpu::GpuInfo& gpu_info) const;
71 const ::tflite::gpu::GpuInfo& gpu_info) const;
78 const ::tflite::gpu::GpuInfo& gpu_info) const;

123456