Searched refs:nudged_input_max (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | fake_quantize_ops.cc | 72 const xla::XlaOp& nudged_input_max, in Quantize() argument
78 xla::XlaOp clamped = xla::Clamp(nudged_input_min, input, nudged_input_max); in Quantize()
114 xla::XlaOp nudged_input_max = in Compile() local
119 nudged_input_max, input_scale); in Compile()
164 xla::XlaOp nudged_input_max = in Compile() local
168 xla::Le(nudged_input_min, input), xla::Le(input, nudged_input_max)); in Compile()
206 xla::XlaOp nudged_input_min, nudged_input_max, input_scale; in Compile() local
208 &nudged_input_min, &nudged_input_max, &input_scale); in Compile()
211 nudged_input_max, input_scale); in Compile()
249 xla::XlaOp nudged_input_min, nudged_input_max, input_scale; in Compile() local
[all …]
|