Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dadd.cc67 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
68 /*output=*/IOStructure::AUTO, in GenerateCode()
88 /*input=*/IOStructure::AUTO, in GenerateCode()
89 /*output=*/IOStructure::AUTO, in GenerateCode()
102 /*input=*/IOStructure::AUTO, in GenerateCode()
103 /*output=*/IOStructure::AUTO, in GenerateCode()
116 /*input=*/IOStructure::AUTO, in GenerateCode()
117 /*output=*/IOStructure::AUTO, in GenerateCode()
Dmul.cc76 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateApplyMaskCode()
77 /*output=*/IOStructure::AUTO, in GenerateApplyMaskCode()
94 /*input=*/IOStructure::AUTO, in GenerateMultiplyScalarCode()
95 /*output=*/IOStructure::AUTO, in GenerateMultiplyScalarCode()
115 /*input=*/IOStructure::AUTO, in GenerateMultiplyScalarCode()
116 /*output=*/IOStructure::AUTO, in GenerateMultiplyScalarCode()
140 /*input=*/IOStructure::AUTO, in GenerateMultiplyScalarCode()
141 /*output=*/IOStructure::AUTO, in GenerateMultiplyScalarCode()
Dprelu.cc60 /*input=*/IOStructure::AUTO, in GenerateCode()
61 /*output=*/IOStructure::AUTO, in GenerateCode()
78 /*input=*/IOStructure::AUTO, in GenerateCode()
79 /*output=*/IOStructure::AUTO, in GenerateCode()
125 /*input=*/IOStructure::AUTO, in GenerateCode()
126 /*output=*/IOStructure::AUTO, in GenerateCode()
145 /*input=*/IOStructure::AUTO, in GenerateCode()
146 /*output=*/IOStructure::AUTO, in GenerateCode()
Dresize.cc61 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
62 /*output=*/IOStructure::AUTO, in GenerateCode()
133 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
134 /*output=*/IOStructure::AUTO, in GenerateCode()
Dsoftmax.cc159 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCodeFor1x1()
160 /*output=*/IOStructure::ONLY_DEFINITIONS, in GenerateCodeFor1x1()
207 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCodeGeneral()
208 /*output=*/IOStructure::ONLY_DEFINITIONS, in GenerateCodeGeneral()
Dconcat.cc85 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
86 /*output=*/IOStructure::AUTO, in GenerateCode()
165 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
166 /*output=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
357 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
358 /*output=*/IOStructure::AUTO, in GenerateCode()
416 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
417 /*output=*/IOStructure::AUTO, in GenerateCode()
Dmean.cc135 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateSubgroupBasedMean()
136 /*output=*/IOStructure::ONLY_DEFINITIONS, in GenerateSubgroupBasedMean()
169 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateTrivialMean()
170 /*output=*/IOStructure::AUTO, in GenerateTrivialMean()
Dpooling.cc92 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateMaxPoolingCode()
93 /*output=*/IOStructure::AUTO, in GenerateMaxPoolingCode()
168 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateAveragePoolingCode()
169 /*output=*/IOStructure::AUTO, in GenerateAveragePoolingCode()
Delementwise.cc116 /*input=*/IOStructure::AUTO, in GenerateCode()
117 /*output=*/IOStructure::AUTO, in GenerateCode()
212 /*input=*/IOStructure::AUTO, in GenerateCode()
213 /*output=*/IOStructure::AUTO, in GenerateCode()
Dlstm.cc80 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
81 /*output=*/IOStructure::AUTO, in GenerateCode()
Dquantize_and_dequantize.cc54 /*input=*/IOStructure::AUTO, in GenerateCode()
55 /*output=*/IOStructure::AUTO, in GenerateCode()
Dspace_to_depth.cc59 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
60 /*output=*/IOStructure::AUTO, in GenerateCode()
Dmax_unpooling.cc66 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
67 /*output=*/IOStructure::AUTO, in GenerateCode()
Drelu.cc62 /*input=*/IOStructure::AUTO, in GenerateCode()
63 /*output=*/IOStructure::AUTO, in GenerateCode()
Dconv.cc142 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
143 /*output=*/IOStructure::AUTO, in GenerateCode()
284 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
285 /*output=*/multiplier == 1 ? IOStructure::AUTO in GenerateCode()
286 : IOStructure::ONLY_DEFINITIONS, in GenerateCode()
Dreshape.cc81 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
82 /*output=*/IOStructure::AUTO, in GenerateCode()
Dtranspose_conv.cc97 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
98 /*output=*/IOStructure::AUTO, in GenerateCode()
Dfully_connected.cc113 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
114 /*output=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
Dslice.cc104 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
105 /*output=*/IOStructure::AUTO, in GenerateCode()
Dpad.cc130 /*input=*/IOStructure::ONLY_DEFINITIONS, in GenerateCode()
131 /*output=*/IOStructure::AUTO, in GenerateCode()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/compiler/
Dfuse_auto_input_test.cc43 a1.code.output = IOStructure::AUTO; in TEST()
49 a2.code.output = IOStructure::AUTO; in TEST()
55 a3.code.input = IOStructure::AUTO; in TEST()
80 a1.code.output = IOStructure::AUTO; in TEST()
86 a2.code.input = IOStructure::AUTO; in TEST()
Dfuse_inline.cc45 if (attr1.code.output != IOStructure::AUTO || in ApplyToNodesSequence()
48 attr2.code.output != IOStructure::AUTO || in ApplyToNodesSequence()
49 attr2.code.input != IOStructure::AUTO || in ApplyToNodesSequence()
Dfuse_auto_input.cc51 if (node_code.input != IOStructure::AUTO) { in ApplyToNode()
79 if (input_producer_attr.code.output != IOStructure::AUTO) { in ApplyToNode()
193 if (attr.code.input == IOStructure::AUTO) { in ApplyToNode()
230 if (node_code.input == IOStructure::AUTO) { in ApplyToNode()
241 node_code.input = IOStructure::ONLY_DEFINITIONS; in ApplyToNode()
Dshader_codegen.cc115 case IOStructure::ONLY_DEFINITIONS: in Build()
121 case IOStructure::AUTO: { in Build()
132 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