Searched defs:flat_input (Results 1 – 2 of 2) sorted by relevance
732 auto flat_input = input.flat<float>(); in FloatTensorToQuantizedInPlaceUsingEigen() local744 auto flat_input = input.flat<float>(); in FloatTensorToQuantizedInPlace() local766 auto flat_input = input.flat<T>(); in QuantizedTensorToFloatInPlaceUsingEigen() local780 auto flat_input = input.flat<T>(); in QuantizedTensorToFloatInPlace() local
82 def _best_effort_input_batch_size(flat_input): argument