Searched defs:input_y_data (Results 1 – 2 of 2) sorted by relevance
31 const T* input_y_data, const RuntimeShape& output_shape, in Select()54 const RuntimeShape& input_y_shape, const T* input_y_data, in RankOneSelect()83 const T* input_y_data, in BroadcastSelect5DSlow()
1113 const Dims<4>& input_x_dims, const T* input_y_data, in Select()1125 const T* input_y_data, const Dims<4>& input_y_dims, in RankOneSelect()