Searched refs:input_chip (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantize_and_dequantize_op.h | 256 const auto input_chip = input.template chip<1>(i); in Compute() local
264 ClampScaleAndRound(d, input_chip, min_range[i], max_range[i], scale, in Compute()
267 ScaleAndRound(d, input_chip, scale, inverse_scale, round_mode, in Compute()
310 const auto input_chip = input.template chip<1>(i); in Compute() local
314 (input_chip >= min_val && input_chip <= max_val) in Compute()
315 .select(input_chip.constant(1.0f), input_chip.constant(0.0f)); in Compute()
|
| D | scatter_nd_op_cpu_impl.h | 143 auto input_chip = Toutput.template chip<0>(i);
144 auto output_chip = input_chip;
147 CPUDevice, decltype(input_chip), decltype(update_chip),
148 decltype(output_chip), OP>::Execute(d, input_chip, update_chip,
|
| D | ctc_decoder_ops.cc | 325 Tensor input_chip(DataTypeToEnum<T>::v(), TensorShape({num_classes})); in Compute() local
326 auto input_chip_t = input_chip.flat<T>(); in Compute()
|