Searched defs:flat_input (Results 1 – 6 of 6) sorted by relevance
218 auto flat_input = input.flat<qint8>(); in Operate() local
129 auto flat_input = input.flat<tstring>(); in Compute() local
739 auto flat_input = input.flat<float>(); in FloatTensorToQuantizedInPlaceUsingEigen() local751 auto flat_input = input.flat<float>(); in FloatTensorToQuantizedInPlace() local773 auto flat_input = input.flat<T>(); in QuantizedTensorToFloatInPlaceUsingEigen() local787 auto flat_input = input.flat<T>(); in QuantizedTensorToFloatInPlace() local
127 auto flat_input = input_data.flat<float>(); in TEST() local
66 def _best_effort_input_batch_size(flat_input): argument
386 auto flat_input = input.flat<float>().data(); in Compute() local