Searched defs:flat_input (Results 1 – 4 of 4) sorted by relevance
198 auto flat_input = input.flat<qint8>(); in Operate() local
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
113 auto flat_input = input_data.flat<float>(); in TEST() local
72 def _best_effort_input_batch_size(flat_input): argument