Searched defs:input_y_data (Results 1 – 2 of 2) sorted by relevance
1178 const T* input_y_data, const RuntimeShape& output_shape, in Select()1200 const RuntimeShape& input_y_shape, const T* input_y_data, in RankOneSelect()1228 const T* input_y_data, in BroadcastSelect4DSlow()
1111 const Dims<4>& input_x_dims, const T* input_y_data, in Select()1123 const T* input_y_data, const Dims<4>& input_y_dims, in RankOneSelect()