Searched refs:addScale (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval_test.cc | 798 static TfLiteTensor* addScale(TfLiteTensor* t, float scale) { in addScale() function in tflite::__anonb312a86f0111::HybridLstmParam
910 HybridLstmParam::addScale(one_parameter.Geti2i(), kDefaultScale), nullptr, in TestOneHybridAsymmLSTM()
911 HybridLstmParam::addScale(one_parameter.Geti2f(), kDefaultScale), nullptr, in TestOneHybridAsymmLSTM()
912 HybridLstmParam::addScale(one_parameter.Geti2c(), kDefaultScale), nullptr, in TestOneHybridAsymmLSTM()
913 HybridLstmParam::addScale(one_parameter.Geti2o(), kDefaultScale), nullptr, in TestOneHybridAsymmLSTM()
914 HybridLstmParam::addScale(one_parameter.Getr2i(), kDefaultScale), nullptr, in TestOneHybridAsymmLSTM()
915 HybridLstmParam::addScale(one_parameter.Getr2f(), kDefaultScale), nullptr, in TestOneHybridAsymmLSTM()
916 HybridLstmParam::addScale(one_parameter.Getr2c(), kDefaultScale), nullptr, in TestOneHybridAsymmLSTM()
917 HybridLstmParam::addScale(one_parameter.Getr2o(), kDefaultScale), nullptr, in TestOneHybridAsymmLSTM()
930 HybridLstmParam::addScale(one_parameter.GetProjection(), 1.0), nullptr, in TestOneHybridAsymmLSTM()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 4134 nvinfer1::IScaleLayer* layer = params->converter->network()->addScale( in ConvertBiasAddInt8WithoutCalibration()
5297 nvinfer1::IScaleLayer* layer = params->converter->network()->addScale( in ConvertFusedBatchNorm()
|