Searched refs:nudged_input_max (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | fake_quantize_ops.cc | 75 const xla::ComputationDataHandle& nudged_input_max, in Quantize() argument
83 b->Clamp(nudged_input_min, input, nudged_input_max); in Quantize()
120 xla::ComputationDataHandle nudged_input_max = in Compile() local
125 b, input, data_type, nudged_input_min, nudged_input_max, input_scale); in Compile()
170 xla::ComputationDataHandle nudged_input_max = in Compile() local
174 b->And(b->Le(nudged_input_min, input), b->Le(input, nudged_input_max)); in Compile()
213 xla::ComputationDataHandle nudged_input_min, nudged_input_max, input_scale; in Compile() local
215 &nudged_input_min, &nudged_input_max, &input_scale); in Compile()
218 b, input, data_type, nudged_input_min, nudged_input_max, input_scale); in Compile()
254 xla::ComputationDataHandle nudged_input_min, nudged_input_max, input_scale; in Compile() local
[all …]
|