Searched defs:input_max_data (Results 1 – 2 of 2) sorted by relevance
76 const float* input_max_data = input_max.flat<float>().data(); in Compute() local
910 scalar_t* input_max_data = buffer.get(); in _vec_logsoftmax() local1015 float* input_max_data = buffer.get(); in _vec_logsoftmax() local