Searched refs:input_x_shape (Results 1 – 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
D | reference_ops.h | 4301 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()
|