Searched refs:input_chip (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scatter_nd_op_cpu_impl.h | 127 auto input_chip = Toutput.template chip<0>(i);
128 auto output_chip = input_chip.device(d);
131 decltype(input_chip), decltype(update_chip), decltype(output_chip),
132 OP>::Execute(input_chip, update_chip, output_chip);
211 auto input_chip = Toutput.template chip<0>(i);
212 auto output_chip = input_chip.device(d);
215 decltype(input_chip), decltype(update_chip), decltype(output_chip),
216 OP>::Execute(input_chip, update_chip, output_chip);
|
| D | ctc_decoder_ops.cc | 304 Tensor input_chip(DT_FLOAT, TensorShape({num_classes})); in Compute() local
305 auto input_chip_t = input_chip.flat<float>(); in Compute()
|