Home
last modified time | relevance | path

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

1234567

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dgpu_info.cc548 GpuApi gpu_api, GpuInfo* gpu_info) { in GetGpuInfoFromDeviceDescription()
690 bool GpuInfo::IsAdreno() const { return vendor == GpuVendor::kQualcomm; } in IsAdreno()
692 bool GpuInfo::IsApple() const { return vendor == GpuVendor::kApple; } in IsApple()
694 bool GpuInfo::IsMali() const { return vendor == GpuVendor::kMali; } in IsMali()
696 bool GpuInfo::IsPowerVR() const { return vendor == GpuVendor::kPowerVR; } in IsPowerVR()
698 bool GpuInfo::IsNvidia() const { return vendor == GpuVendor::kNvidia; } in IsNvidia()
700 bool GpuInfo::IsAMD() const { return vendor == GpuVendor::kAMD; } in IsAMD()
702 bool GpuInfo::IsIntel() const { return vendor == GpuVendor::kIntel; } in IsIntel()
704 bool GpuInfo::IsRoundToNearestSupported() const { in IsRoundToNearestSupported()
723 bool GpuInfo::SupportsFP16() const { in SupportsFP16()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_generic.h88 TuningType tuning_type, const GpuInfo& gpu_info,
110 const Convolution2DAttributes& attr, const GpuInfo& gpu_info,
114 const GpuInfo& gpu_info, const BHWC* dst_shape = nullptr);
116 const FullyConnectedAttributes& attr, const GpuInfo& gpu_info,
120 const Convolution3DAttributes& attr, const GpuInfo& gpu_info,
123 void GenerateCode(const GpuInfo& gpu_info);
141 friend ConvGeneric CreateConvGeneric(const GpuInfo& gpu_info,
146 friend ConvGeneric CreateConvGeneric(const GpuInfo& gpu_info,
152 const GpuInfo& gpu_info, const OperationDef& definition,
156 const GpuInfo& gpu_info, const OperationDef& definition,
[all …]
Dwinograd.h36 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
55 const GpuInfo& gpu_info);
62 const GpuInfo& gpu_info);
68 const Padding2D& padding, const GpuInfo& gpu_info);
72 TuningType tuning_type, const GpuInfo& gpu_info,
84 const GpuInfo& gpu_info, const OperationDef& definition,
90 const GpuInfo& gpu_info);
99 const GpuInfo& gpu_info, const OperationDef& definition,
106 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
135 const GpuInfo& gpu_info);
[all …]
Dconvolution_transposed_4x4.h41 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
73 const GpuInfo& gpu_info);
76 const GpuInfo& gpu_info, const OperationDef& definition,
79 const GpuInfo& gpu_info, const OperationDef& definition,
89 const GpuInfo& gpu_info, const OperationDef& op_def,
100 const GpuInfo& gpu_info, const OperationDef& definition,
104 const GpuInfo& gpu_info, const OperationDef& definition,
Dconvolution_transposed_3x3.h41 TuningType tuning_type, const GpuInfo& gpu_info,
71 const GpuInfo& gpu_info, int2 padding);
73 const GpuInfo& gpu_info, const OperationDef& definition,
76 const GpuInfo& gpu_info, const OperationDef& definition,
85 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.h45 TuningType tuning_type, const GpuInfo& gpu_info,
67 const GpuInfo& gpu_info, const OperationDef& definition,
70 const GpuInfo& gpu_info, const OperationDef& definition,
73 const GpuInfo& gpu_info, const OperationDef& definition,
78 const GpuInfo& gpu_info);
81 const GpuInfo& gpu_info);
92 const GpuInfo& gpu_info,
162 const GpuInfo& gpu_info, const OperationDef& definition,
166 const GpuInfo& gpu_info, const OperationDef& definition,
170 const GpuInfo& gpu_info, const OperationDef& definition,
Dconvolution_transposed_3x3_thin.h57 ConvolutionTransposed3x3Thin(const GpuInfo& gpu_info,
62 const GpuInfo& gpu_info, const OperationDef& definition,
66 const GpuInfo& gpu_info, const OperationDef& definition,
75 const GpuInfo& gpu_info,
85 const GpuInfo& gpu_info, const OperationDef& definition,
89 const GpuInfo& gpu_info, const OperationDef& definition,
Dmean_stddev_normalization.h37 const GpuInfo& gpu_info, const BHWC& shape,
41 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
56 std::string GetNormalizationCode(const GpuInfo& gpu_info, bool channels_x4,
63 const OperationDef& definition, const GpuInfo& gpu_info, const BHWC& shape,
87 const GpuInfo& gpu_info, CalculationsPrecision precision,
Dreduce.h35 const OperationDef& definition, const GpuInfo& gpu_info);
38 TuningType tuning_type, const GpuInfo& gpu_info,
52 const GpuInfo& gpu_info,
62 const GpuInfo& gpu_info);
66 const OperationDef& definition, const GpuInfo& gpu_info);
Dconv_metal_simd.h33 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
88 const Convolution2DAttributes& attr, const GpuInfo& gpu_info);
93 const Convolution2DAttributes& attr, const GpuInfo& gpu_info);
95 bool IsConvolutionMetalSimdSupported(const GpuInfo& gpu_info,
100 const GpuInfo& gpu_info);
Ddepthwise_conv.h41 TuningType tuning_type, const GpuInfo& gpu_info,
52 const GpuInfo& gpu_info, const OperationDef& definition,
56 const GpuInfo& gpu_info, const OperationDef& definition,
60 const GpuInfo& gpu_info, const OperationDef& definition,
101 std::string GenerateSrcUpload(const GpuInfo& gpu_info);
102 std::string GenerateWeightsUpload(const GpuInfo& gpu_info);
103 std::string GenerateCode(const GpuInfo& gpu_info);
258 const GpuInfo& gpu_info, const OperationDef& definition,
262 const GpuInfo& gpu_info, const OperationDef& definition,
266 const GpuInfo& gpu_info, const OperationDef& definition,
Delementwise.h30 GPUOperation CreateElementwiseOneInput(const GpuInfo& gpu_info,
38 const GpuInfo& gpu_info, const OperationDef& definition,
44 GPUOperation CreateElementwise(const GpuInfo& gpu_info,
52 GPUOperation CreateElementwiseWithBroadcast(const GpuInfo& gpu_info,
Dfully_connected.h121 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
135 FullyConnected(const OperationDef& definition, const GpuInfo& gpu_info);
137 const GpuInfo& gpu_info, const OperationDef& definition,
140 const GpuInfo& gpu_info, const OperationDef& definition,
151 const GpuInfo& gpu_info,
201 FullyConnected CreateFullyConnected(const GpuInfo& gpu_info,
205 FullyConnected CreateFullyConnected(const GpuInfo& gpu_info,
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Dsimple_selectors.h31 const GpuInfo& gpu_info);
37 const GpuInfo& gpu_info,
41 const GpuInfo& gpu_info,
45 const MaxUnpooling2DAttributes& attr, const GpuInfo& gpu_info,
60 const GpuInfo& gpu_info);
64 const OperationDef& op_def, const GpuInfo& gpu_info,
68 const DepthwiseConvolution2DAttributes& attr, const GpuInfo& gpu_info,
85 const GpuInfo& gpu_info);
98 void SelectSplit(const SplitAttributes& attr, const GpuInfo& gpu_info,
109 std::unique_ptr<GPUOperation> SelectWinograd4x4To36(const GpuInfo& gpu_info,
[all …]
Dsimple_selectors.cc58 const GpuInfo& gpu_info) { in SelectLSTM()
68 const GpuInfo& gpu_info, in SelectPReLU()
74 const GpuInfo& gpu_info, in SelectPooling()
80 const MaxUnpooling2DAttributes& attr, const GpuInfo& gpu_info, in SelectMaxUnpooling()
105 const GpuInfo& gpu_info) { in SelectResampler()
120 const OperationDef& op_def, const GpuInfo& gpu_info, in SelectConcat()
142 const DepthwiseConvolution2DAttributes& attr, const GpuInfo& gpu_info, in SelectDWConvolutionDynamicWeights()
174 void SelectSplit(const SplitAttributes& attr, const GpuInfo& gpu_info, in SelectSplit()
197 const GpuInfo& gpu_info) { in SelectReduce()
226 const GpuInfo& gpu_info, const Padding2D& padding, in SelectWinograd4x4To36()
[all …]
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,
45 const OHWI& weights_shape, const BHWC& dst_shape, const GpuInfo& gpu_info,
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dgpu_operation.h96 absl::Status AddOperation(const GpuInfo& gpu_info, GPUOperation* operation);
105 absl::Status FuseSimpleElemWithSimpleElem(const GpuInfo& gpu_info,
116 const GpuInfo& gpu_info, GPUOperation* operation);
126 const GpuInfo& gpu_info, GPUOperation* operation);
135 void GetPossibleDispatches(TuningType tuning_type, const GpuInfo& gpu_info,
149 absl::Status AssembleCode(const GpuInfo& gpu_info);
151 virtual absl::Status PostCompileCheck(const GpuInfo& gpu_info, in PostCompileCheck()
199 friend absl::Status Fuse2InputElemWith2SimpleElem(const GpuInfo& gpu_info,
207 TuningType tuning_type, const GpuInfo& gpu_info,
248 absl::Status Fuse2InputElemWith2SimpleElem(const GpuInfo& gpu_info,
Dutil.h58 int GetRecommendedBlockSizeForConv(const GpuInfo& gpu_info,
64 std::string GetTypeDeclaration(const GpuInfo& gpu_info, DataType data_type,
67 std::string GetZeroValue(const GpuInfo& gpu_info, DataType data_type,
70 std::string GetOneValue(const GpuInfo& gpu_info, DataType data_type,
78 std::string GetTypeConversion(const GpuInfo& gpu_info, DataType src_type,
Darguments.h74 int GetReadTexturesCount(const GpuInfo& gpu_info) const;
75 int GetWriteTexturesCount(const GpuInfo& gpu_info) const;
126 absl::Status Compile(const GpuInfo& gpu_info,
130 absl::Status ResolveConstExprPass(const GpuInfo& gpu_info,
133 absl::Status ResolveConstExpr(const GpuInfo& gpu_info,
139 const GpuInfo& gpu_info,
144 const GpuInfo& gpu_info,
153 absl::Status AddObjectsScalarArgs(const GpuInfo& gpu_info);
162 absl::Status ResolveKernelGlobalSpaceBuffers(const GpuInfo& gpu_info,
Dtensor_desc.h68 absl::Status PerformConstExpr(const GpuInfo& gpu_info,
72 absl::Status PerformSelector(const GpuInfo& gpu_info,
78 GPUResources GetGPUResources(const GpuInfo& gpu_info) const override;
105 bool SupportsZeroClamp(const Axis& axis, const GpuInfo& gpu_info) const;
116 bool ReturnsZeroForNegOneRead(const GpuInfo& gpu_info) const;
118 absl::Status CanCreateTensorWithShape(const GpuInfo& gpu_info,
121 absl::Status CanCreateTensorWithShape(const GpuInfo& gpu_info,
129 absl::Status UpdateToSupportedStorageType(const GpuInfo& gpu_info,
181 const GpuInfo& gpu_info, const std::vector<std::string>& args,
183 absl::Status PerformReadNearestSelector(const GpuInfo& gpu_info,
[all …]
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,
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/
Dconvolution_selector.cc37 const GpuInfo& gpu_info, const OperationDef& op_def, in SelectConvolutionAdreno()
50 const GpuInfo& gpu_info, const OperationDef& op_def) { in SelectConvolutionNVidia()
62 const GpuInfo& gpu_info, const OperationDef& op_def) { in SelectConvolutionApple()
80 const GpuInfo& gpu_info, const OperationDef& op_def, in SelectConvolution()
99 const GpuInfo& gpu_info, const OperationDef& op_def, in SelectConvolutionForWinograd()
108 const BHWC& dst_shape, const GpuInfo& gpu_info, in SelectConvolutionWithDynamicWeights()
118 const OHWI& weights_shape, const BHWC& dst_shape, const GpuInfo& gpu_info, in SelectConvolutionBatchedMatMul()
Ddw_convolution_selector.cc29 const DepthwiseConvolution2DAttributes& attr, const GpuInfo& gpu_info, in SelectDWConvolutionAdreno()
41 const DepthwiseConvolution2DAttributes& attr, const GpuInfo& gpu_info, in SelectDWConvolutionPowerVR()
53 const DepthwiseConvolution2DAttributes& attr, const GpuInfo& gpu_info, in SelectDWConvolutionMali()
71 const DepthwiseConvolution2DAttributes& attr, const GpuInfo& gpu_info, in SelectDWConvolutionApple()
87 const DepthwiseConvolution2DAttributes& attr, const GpuInfo& gpu_info, in SelectDWConvolution()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Dfc_fc_add.h100 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
114 FCFCAdd(const OperationDef& definition, const GpuInfo& gpu_info);
115 friend FCFCAdd CreateFCFCAdd(const GpuInfo& gpu_info,
119 friend FCFCAdd CreateFCFCAdd(const GpuInfo& gpu_info,
133 const GpuInfo& gpu_info,
183 FCFCAdd CreateFCFCAdd(const GpuInfo& gpu_info, const OperationDef& definition,
187 FCFCAdd CreateFCFCAdd(const GpuInfo& gpu_info, const OperationDef& definition,
192 const GpuInfo& gpu_info, CalculationsPrecision precision,
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dinference_context.h122 const GpuInfo& gpu_info,
133 const GpuInfo& gpu_info,
138 const GpuModel& gpu_model, const GpuInfo& gpu_info,
143 absl::Status Tune(TuningType tuning_type, const GpuInfo& gpu_info,
166 void Init(const GpuInfo& gpu_info);
200 GpuInfo gpu_info_;
209 const GpuInfo& gpu_info,

1234567