Searched refs:approx_output_shapes (Results 1 – 1 of 1) sorted by relevance
223 std::vector<const Shape*> approx_output_shapes; in ApproxTopK() local224 approx_output_shapes.reserve(operands_shapes.size()); in ApproxTopK()227 approx_output_shapes.push_back(&op_shape); in ApproxTopK()230 ShapeUtil::MakeTupleShapeWithPtrs(approx_output_shapes); in ApproxTopK()