Home
last modified time | relevance | path

Searched refs:GenerationContext (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dconcat.cc36 static bool IsSupported(const GenerationContext& ctx) { in IsSupported()
61 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
94 static bool IsSupported(const GenerationContext& ctx) { in IsSupported()
114 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
307 static bool IsSupported(const GenerationContext& ctx) { in IsSupported()
327 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
366 static bool IsSupported(const GenerationContext& ctx) { in IsSupported()
386 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
425 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
Dmul.cc36 bool IsApplyMaskSupported(const NodeShader::GenerationContext& ctx) { in IsApplyMaskSupported()
54 absl::Status GenerateApplyMaskCode(const NodeShader::GenerationContext& ctx, in GenerateApplyMaskCode()
83 const NodeShader::GenerationContext& ctx, GeneratedCode* generated_code) { in GenerateMultiplyScalarCode()
151 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
Dsoftmax.cc44 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
61 absl::Status GenerateCodeFor1x1(const GenerationContext& ctx, in GenerateCodeFor1x1()
165 absl::Status GenerateCodeGeneral(const GenerationContext& ctx, in GenerateCodeGeneral()
Delementwise.cc34 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
131 inline bool IsElementwiseSupported(const GenerationContext& ctx) const { in IsElementwiseSupported()
136 inline bool IsBroadcastSupported(const GenerationContext& ctx) const { in IsBroadcastSupported()
142 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
Dprelu.cc38 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
87 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
154 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
Dmean.cc46 void GenerateSubgroupBasedMean(const NodeShader::GenerationContext& ctx, in GenerateSubgroupBasedMean()
140 void GenerateTrivialMean(const NodeShader::GenerationContext& ctx, in GenerateTrivialMean()
176 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
Dpooling.cc35 const NodeShader::GenerationContext& ctx, in GenerateMaxPoolingCode()
99 const Pooling2DAttributes& attr, const NodeShader::GenerationContext& ctx, in GenerateAveragePoolingCode()
176 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
Dconv.cc40 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
150 const NodeShader::GenerationContext& ctx) { in SelectMultiplier()
165 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
Dlstm.cc46 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
Dquantize_and_dequantize.cc34 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
Dspace_to_depth.cc34 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
Dmax_unpooling.cc36 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
Drelu.cc36 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
Dreshape.cc35 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
Dtranspose_conv.cc38 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
Dfully_connected.cc37 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
Dslice.cc36 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
Dpad.cc37 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
Dadd.cc37 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
Dresize.cc36 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
Ddepthwise_conv.cc39 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
Dregistry.cc127 absl::Status GenerateCode(const GenerationContext& ctx, in GenerateCode()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
Dnode_shader.h98 struct GenerationContext { struct
112 virtual absl::Status GenerateCode(const GenerationContext& ctx, argument
Dcompiler.cc125 NodeShader::GenerationContext ctx = {&gpu_info_, options_, in Compile()