Searched refs:input_buffer_data (Results 1 – 3 of 3) sorted by relevance
213 T* input_buffer_data = input_buffer.template flat<T>().data(); in operator ()() local226 input_buffer_data); in operator ()()231 input_buffer_data, output + out_base, data_format); in operator ()()
912 T* input_buffer_data = input_buffer.template flat<T>().data(); in operator ()() local931 input + b * input_image_size, input_buffer_data); in operator ()()935 input_buffer_data, output_buffer); in operator ()()
1020 float* input_buffer_data = input_buffer.get(); in _vec_logsoftmax() local1047 float* input_buffer_ptr = input_buffer_data + dim_idx * CHUNK_SIZE; in _vec_logsoftmax()1074 float* input_buffer_ptr = input_buffer_data + dim_idx * CHUNK_SIZE; in _vec_logsoftmax()1101 float* input_buffer_ptr = input_buffer_data + dim_idx * CHUNK_SIZE; in _vec_logsoftmax()