Searched refs:input_y_data (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 1487 const T* input_y_data, const RuntimeShape& output_shape, in Select() argument
1493 input_condition_data[i] ? input_x_data[i] : input_y_data[i]; in Select()
1501 const RuntimeShape& input_y_shape, const T* input_y_data, in RankOneSelect() argument
1517 const T* input_data = input_condition_data[i] ? input_x_data : input_y_data; in RankOneSelect()
1529 const T* input_y_data, in BroadcastSelect4DSlow() argument
1567 : input_y_data[y_index]; in BroadcastSelect4DSlow()
|
| D | legacy_reference_ops.h | 1110 const Dims<4>& input_x_dims, const T* input_y_data, in Select() argument
1115 input_y_data, DimsToShape(output_dims), output_data); in Select()
1122 const T* input_y_data, const Dims<4>& input_y_dims, in RankOneSelect() argument
1126 DimsToShape(input_y_dims), input_y_data, in RankOneSelect()
|