Searched defs:input_y_data (Results 1 – 2 of 2) sorted by relevance
1487 const T* input_y_data, const RuntimeShape& output_shape, in Select()1501 const RuntimeShape& input_y_shape, const T* input_y_data, in RankOneSelect()1529 const T* input_y_data, in BroadcastSelect4DSlow()
1110 const Dims<4>& input_x_dims, const T* input_y_data, in Select()1122 const T* input_y_data, const Dims<4>& input_y_dims, in RankOneSelect()