Home
last modified time | relevance | path

Searched defs:input_chip (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dquantize_and_dequantize_op.h252 const auto input_chip = input.template chip<1>(i); in Compute() local
306 const auto input_chip = input.template chip<1>(i); in Compute() local
H A Dctc_decoder_ops.cc336 Tensor input_chip(DataTypeToEnum<T>::v(), TensorShape({num_classes})); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Druntime_conv_impl.h177 auto input_chip = input.reshape(input_reshaped_dims).chip(i, 4); in EigenConv3DImpl() local