Home
last modified time | relevance | path

Searched refs:channel_scales_node (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dtranspose_conv_2d_builder.cc150 if (per_channel_quant_.channel_scales_node != nullptr) { in PopulateSubGraph()
175 if (per_channel_quant_.channel_scales_node != nullptr) { in PopulateSubGraph()
176 AddInput(TensorID(per_channel_quant_.channel_scales_node->GetID(), 0)); in PopulateSubGraph()
Dconv_2d_builder.cc284 if (per_channel_quant_.channel_scales_node != nullptr) { in PopulateSubGraph()
286 TensorID(per_channel_quant_.channel_scales_node->GetID(), 0)); in PopulateSubGraph()
348 if (per_channel_quant_.channel_scales_node != nullptr) { in PopulateSubGraph()
349 AddInput(TensorID(per_channel_quant_.channel_scales_node->GetID(), 0)); in PopulateSubGraph()
Dconv_2d_builder.h32 OpBuilder* channel_scales_node = nullptr; member
Dconv_2d_helpers.cc76 per_channel_quant->channel_scales_node = graph_builder->AddConstNodeWithData( in ProcessPerChannelQuantizedWeights()
259 if (per_channel_quant_.channel_scales_node != nullptr) { in InitializeBiasNodes()