Searched defs:flat_input (Results 1 – 6 of 6) sorted by relevance
196 auto flat_input = input.flat<qint8>(); in Operate() local
120 auto flat_input = input.flat<tstring>(); in Compute() local
741 auto flat_input = input.flat<float>(); in FloatTensorToQuantizedInPlaceUsingEigen() local753 auto flat_input = input.flat<float>(); in FloatTensorToQuantizedInPlace() local775 auto flat_input = input.flat<T>(); in QuantizedTensorToFloatInPlaceUsingEigen() local789 auto flat_input = input.flat<T>(); in QuantizedTensorToFloatInPlace() local
127 auto flat_input = input_data.flat<float>(); in TEST() local
70 def _best_effort_input_batch_size(flat_input): argument
377 auto flat_input = input.flat<float>().data(); in Compute() local