Searched refs:input_x_dims (Results 1 – 1 of 1) sorted by relevance
1019 const Dims<4>& input_x_dims, const T* input_y_data, in Select() argument1023 DimsToShape(input_x_dims), input_x_data, DimsToShape(input_y_dims), in Select()1030 const T* input_x_data, const Dims<4>& input_x_dims, in RankOneSelect() argument1034 DimsToShape(input_x_dims), input_x_data, in RankOneSelect()