Searched refs:input_condition_shape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 1484 void Select(const RuntimeShape& input_condition_shape, in Select() argument
1490 input_condition_shape, input_x_shape, input_y_shape, output_shape); in Select()
1498 void RankOneSelect(const RuntimeShape& input_condition_shape, in RankOneSelect() argument
1503 const int64_t outer_size = input_condition_shape.FlatSize(); in RankOneSelect()
1505 if (input_condition_shape.DimensionsCount() == 0) { in RankOneSelect()
1524 void BroadcastSelect4DSlow(const RuntimeShape& input_condition_shape, in BroadcastSelect4DSlow() argument
1531 TFLITE_DCHECK_LE(input_condition_shape.DimensionsCount(), 4); in BroadcastSelect4DSlow()
1542 NdArrayDescsForElementwiseBroadcast(input_condition_shape, input_x_shape, in BroadcastSelect4DSlow()
1575 void SelectTrueCoords(const RuntimeShape& input_condition_shape, in SelectTrueCoords() argument
1577 const size_t size = input_condition_shape.FlatSize(); in SelectTrueCoords()
[all …]
|