Searched defs:flat_input (Results 1 – 7 of 7) 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
1389 std::vector<at::Tensor> flat_input(inputTensors.size()); in reduce_scatter() local