Home
last modified time | relevance | path

Searched refs:IOStructure (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dadd.cc81 /*input=*/IOStructure::AUTO, in GenerateCode()
82 /*output=*/IOStructure::AUTO, in GenerateCode()
103 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
104 /*output=*/IOStructure::AUTO, in GenerateCode()
124 /*input=*/IOStructure::AUTO, in GenerateCode()
125 /*output=*/IOStructure::AUTO, in GenerateCode()
138 /*input=*/IOStructure::AUTO, in GenerateCode()
139 /*output=*/IOStructure::AUTO, in GenerateCode()
154 /*input=*/IOStructure::AUTO, in GenerateCode()
155 /*output=*/IOStructure::AUTO, in GenerateCode()
Dmul.cc80 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateApplyMaskCode()
81 /*output=*/IOStructure::AUTO, in GenerateApplyMaskCode()
98 /*input=*/IOStructure::AUTO, in GenerateMultiplyScalarCode()
99 /*output=*/IOStructure::AUTO, in GenerateMultiplyScalarCode()
119 /*input=*/IOStructure::AUTO, in GenerateMultiplyScalarCode()
120 /*output=*/IOStructure::AUTO, in GenerateMultiplyScalarCode()
144 /*input=*/IOStructure::AUTO, in GenerateMultiplyScalarCode()
145 /*output=*/IOStructure::AUTO, in GenerateMultiplyScalarCode()
Dspace_to_depth.cc61 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
62 /*output=*/IOStructure::AUTO, in GenerateCode()
97 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
98 /*output=*/IOStructure::AUTO, in GenerateCode()
Dmean.cc138 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateSubgroupBasedMean()
139 /*output=*/IOStructure::ONLY_DEFINITIONS, in GenerateSubgroupBasedMean()
171 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateTrivialMean()
172 /*output=*/IOStructure::AUTO, in GenerateTrivialMean()
236 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateTiledMean()
237 /*output=*/IOStructure::ONLY_DEFINITIONS, in GenerateTiledMean()
Dresize.cc64 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
65 /*output=*/IOStructure::AUTO, in GenerateCode()
136 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
137 /*output=*/IOStructure::AUTO, in GenerateCode()
Dsoftmax.cc160 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCodeFor1x1()
161 /*output=*/IOStructure::ONLY_DEFINITIONS, in GenerateCodeFor1x1()
208 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCodeGeneral()
209 /*output=*/IOStructure::ONLY_DEFINITIONS, in GenerateCodeGeneral()
Dconcat.cc88 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
89 /*output=*/IOStructure::AUTO, in GenerateCode()
168 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
169 /*output=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
360 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
361 /*output=*/IOStructure::AUTO, in GenerateCode()
419 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
420 /*output=*/IOStructure::AUTO, in GenerateCode()
Dprelu.cc67 /*input=*/IOStructure::AUTO, in GenerateCode()
68 /*output=*/IOStructure::AUTO, in GenerateCode()
110 /*input=*/IOStructure::AUTO, in GenerateCode()
111 /*output=*/IOStructure::AUTO, in GenerateCode()
Dpooling.cc95 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateMaxPoolingCode()
96 /*output=*/IOStructure::AUTO, in GenerateMaxPoolingCode()
171 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateAveragePoolingCode()
172 /*output=*/IOStructure::AUTO, in GenerateAveragePoolingCode()
Delementwise.cc123 /*input=*/IOStructure::AUTO, in GenerateCode()
124 /*output=*/IOStructure::AUTO, in GenerateCode()
225 /*input=*/IOStructure::AUTO, in GenerateCode()
226 /*output=*/IOStructure::AUTO, in GenerateCode()
Dlstm.cc81 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
82 /*output=*/IOStructure::AUTO, in GenerateCode()
Dtile.cc58 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
59 /*output=*/IOStructure::AUTO, in GenerateCode()
Dresampler.cc72 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
73 /*output=*/IOStructure::AUTO, in GenerateCode()
Dquantize_and_dequantize.cc55 /*input=*/IOStructure::AUTO, in GenerateCode()
56 /*output=*/IOStructure::AUTO, in GenerateCode()
Dmax_unpooling.cc69 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
70 /*output=*/IOStructure::AUTO, in GenerateCode()
Drelu.cc65 /*input=*/IOStructure::AUTO, in GenerateCode()
66 /*output=*/IOStructure::AUTO, in GenerateCode()
Dconv.cc149 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
150 /*output=*/IOStructure::AUTO, in GenerateCode()
294 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
295 /*output=*/multiplier == 1 ? IOStructure::AUTO in GenerateCode()
296 : IOStructure::ONLY_DEFINITIONS, in GenerateCode()
Dreshape.cc84 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
85 /*output=*/IOStructure::AUTO, in GenerateCode()
Dtranspose_conv.cc100 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
101 /*output=*/IOStructure::AUTO, in GenerateCode()
Dfully_connected.cc115 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
116 /*output=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/compiler/
Dfuse_auto_input_test.cc45 a1.code.output = IOStructure::AUTO; in TEST()
51 a2.code.output = IOStructure::AUTO; in TEST()
57 a3.code.input = IOStructure::AUTO; in TEST()
82 a1.code.output = IOStructure::AUTO; in TEST()
88 a2.code.input = IOStructure::AUTO; in TEST()
Dfuse_inline.cc47 if (attr1.code.output != IOStructure::AUTO || in ApplyToNodesSequence()
50 attr2.code.output != IOStructure::AUTO || in ApplyToNodesSequence()
51 attr2.code.input != IOStructure::AUTO || in ApplyToNodesSequence()
Dfuse_auto_input.cc52 if (node_code.input != IOStructure::AUTO) { in ApplyToNode()
80 if (input_producer_attr.code.output != IOStructure::AUTO) { in ApplyToNode()
194 if (attr.code.input == IOStructure::AUTO) { in ApplyToNode()
231 if (node_code.input == IOStructure::AUTO) { in ApplyToNode()
242 node_code.input = IOStructure::ONLY_DEFINITIONS; in ApplyToNode()
Dshader_codegen.cc117 case IOStructure::ONLY_DEFINITIONS: in Build()
123 case IOStructure::AUTO: { in Build()
134 if (attr.code.output == IOStructure::AUTO) { in Build()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
Dnode_shader.h38 enum class IOStructure { enum
88 IOStructure input;
89 IOStructure output;

12