Home
last modified time | relevance | path

Searched refs:FullyConnectedAttributes (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/
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/transformations/
Dfuse_mul_to_conv.h61 FullyConnectedAttributes* attr);
87 FullyConnectedAttributes* attr);
Dfuse_add_to_conv.cc93 FullyConnectedAttributes* conv_attr = in ApplyToNodesSequence()
94 absl::any_cast<FullyConnectedAttributes>( in ApplyToNodesSequence()
134 FullyConnectedAttributes* attr) { in FuseFullyConnectedWithAdd()
Dfuse_mul_to_conv.cc84 FullyConnectedAttributes* conv_attr = in ApplyToNodesSequence()
85 absl::any_cast<FullyConnectedAttributes>( in ApplyToNodesSequence()
148 FullyConnectedAttributes* conv_attr = in ApplyToNodesSequence()
149 absl::any_cast<FullyConnectedAttributes>( in ApplyToNodesSequence()
240 FullyConnectedAttributes* attr) { in FuseFullyConnectedWithMultiply()
309 FullyConnectedAttributes* attr) { in FuseMultiplyWithFullyConnected()
Dfuse_add_to_conv.h54 FullyConnectedAttributes* attr);
Dmake_fully_connected.cc65 FullyConnectedAttributes fc_attr; in ApplyToNode()
Dadd_bias.cc85 absl::any_cast<FullyConnectedAttributes&>(node->operation.attributes); in ApplyToNode()
Dfuse_mul_to_conv_test.cc207 FullyConnectedAttributes attr; in TEST()
291 FullyConnectedAttributes attr; in TEST()
Dmake_fully_connected_test.cc105 auto fc_attr = absl::any_cast<FullyConnectedAttributes>( in TEST()
Dfuse_add_to_conv_test.cc156 FullyConnectedAttributes attr; in TEST()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Dfc_fc_add.h113 const FullyConnectedAttributes& attr0,
114 const FullyConnectedAttributes& attr1);
175 const FullyConnectedAttributes& attr0,
176 const FullyConnectedAttributes& attr1);
Dfc_fc_add.cc174 const FullyConnectedAttributes& attr0, in CreateFCFCAdd()
175 const FullyConnectedAttributes& attr1) { in CreateFCFCAdd()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dfully_connected_test_util.cc33 FullyConnectedAttributes attr; in FullyConnectedTest()
66 FullyConnectedAttributes attr; in FullyConnectedLargeTest()
117 FullyConnectedAttributes attr; in FullyConnectedExtraLargeTest()
Dfully_connected.h137 const FullyConnectedAttributes& attr);
196 const FullyConnectedAttributes& attr);
Dconv_powervr.h117 const FullyConnectedAttributes& attr, const GpuInfo& gpu_info,
149 const FullyConnectedAttributes& attr,
177 const FullyConnectedAttributes& attr,
366 const FullyConnectedAttributes& attr,
Dconv_buffer_1x1.h76 const FullyConnectedAttributes& attr,
183 const FullyConnectedAttributes& attr,
Dfully_connected.cc159 const FullyConnectedAttributes& attr) { in CreateFullyConnected()
Dconv_powervr.cc193 const FullyConnectedAttributes& attr, in ConvPowerVR()
1409 const FullyConnectedAttributes& attr, const BHWC* dst_shape) { in GuessBestParams()
1445 const FullyConnectedAttributes& attr, in CreateConvPowerVR()
Dconv_buffer_1x1.cc364 const FullyConnectedAttributes& attr, in CreateConvBuffer1x1()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Dfully_connected_selector.h29 const FullyConnectedAttributes& attr, const GpuInfo& gpu_info,
Dspecial_selector.cc148 absl::any_cast<FullyConnectedAttributes>(fc0_node->operation.attributes); in TryFCFCAdd()
150 absl::any_cast<FullyConnectedAttributes>(fc1_node->operation.attributes); in TryFCFCAdd()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dfully_connected_test.cc39 FullyConnectedAttributes attr; in TEST()
Dfully_connected.cc40 absl::any_cast<const FullyConnectedAttributes&>(ctx.op_attr); in GenerateCode()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dlstm_parser.cc57 FullyConnectedAttributes* attr) { in SetFullyConnectedWeights()
129 FullyConnectedAttributes fc_attr; in BuildLstmGate()
146 FullyConnectedAttributes fc_attr; in BuildLstmGate()
380 FullyConnectedAttributes fc_attr; in BuildOutputStateUpdate()
Doperations.h496 struct FullyConnectedAttributes { struct
504 const FullyConnectedAttributes& attr); argument

12