Searched defs:input_chip (Results 1 – 2 of 2) sorted by relevance
256 const auto input_chip = input.template chip<1>(i); in Compute() local310 const auto input_chip = input.template chip<1>(i); in Compute() local
325 Tensor input_chip(DataTypeToEnum<T>::v(), TensorShape({num_classes})); in Compute() local