Home
last modified time | relevance | path

Searched refs:input_x_shape (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h4301 const D* input_condition_data, const RuntimeShape& input_x_shape, in Select() argument
4306 input_condition_shape, input_x_shape, input_y_shape, output_shape); in Select()
4316 const RuntimeShape& input_x_shape, const T* input_x_data, in RankOneSelect() argument
4321 MatchingDim(input_x_shape, 0, input_y_shape, 0, output_shape, 0), in RankOneSelect()
4324 MatchingFlatSizeSkipDim(input_x_shape, 0, input_y_shape, output_shape); in RankOneSelect()