Searched refs:ReduceAll (Results 1 - 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
H A D | quantize_and_dequantize_op.cc | 61 b->ReduceAll(input, XlaHelpers::MaxValue(b, data_type), *fmin); 63 b->ReduceAll(input, XlaHelpers::MinValue(b, data_type), *fmax);
|
H A D | fake_quantize_ops.cc | 269 b->ReduceAll(b->Select(below_min, gradient, zeroes), zero, 275 b->ReduceAll(b->Select(above_max, gradient, zeroes), zero,
|
/external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
H A D | scatter.cc | 140 body_builder->ReduceAll(body_builder->Le(zero_index, index),
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
H A D | computation_builder.h | 537 ComputationDataHandle ReduceAll(const ComputationDataHandle& operand,
|
H A D | computation_builder.cc | 1284 ComputationDataHandle ComputationBuilder::ReduceAll( function in class:xla::ComputationBuilder
|
Completed in 164 milliseconds