Searched refs:input_condition_data (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 1485 const D* input_condition_data, const RuntimeShape& input_x_shape, in Select() argument
1493 input_condition_data[i] ? input_x_data[i] : input_y_data[i]; in Select()
1499 const D* input_condition_data, in RankOneSelect() argument
1517 const T* input_data = input_condition_data[i] ? input_x_data : input_y_data; in RankOneSelect()
1525 const D* input_condition_data, in BroadcastSelect4DSlow() argument
1566 input_condition_data[condition_index] ? input_x_data[x_index] in BroadcastSelect4DSlow()
1576 const D* input_condition_data, T* output_data) { in SelectTrueCoords() argument
1593 if (input_condition_data[i]) { in SelectTrueCoords()
|
| D | legacy_reference_ops.h | 1108 inline void Select(const D* input_condition_data, in Select() argument
1113 Select(DimsToShape(input_condition_dims), input_condition_data, in Select()
1119 inline void RankOneSelect(const D* input_condition_data, in RankOneSelect() argument
1124 RankOneSelect(DimsToShape(input_condition_dims), input_condition_data, in RankOneSelect()
|