Searched refs:scale_parameter (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbatch_normalization_test.cc800 auto scale_parameter = builder.Parameter(1, scale_literal->shape(), "scale"); local
817 auto t = builder.BatchNormGrad(input_parameter, scale_parameter,

Completed in 233 milliseconds