Home
last modified time | relevance | path

Searched defs:scale_layer (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
H A Dlayer_utils.h483 nvinfer1::IScaleLayer* scale_layer = in AddUniformScale() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes.cc4311 auto scale_layer = in ConvertFusedBatchNorm() local